检查注释是否属于特定类型

import re

def check_comment_type(comment, comment_type):
    if comment_type == "TODO":
        pattern = r"TODO:.*"
    elif comment_type == "FIXME":
        pattern = r"FIXME:.*"
    elif comment_type == "XXX":
        pattern = r"XXX:.*"
    else:
        pattern = r"{}:.*".format(comment_type)

    if re.match(pattern, comment):
        return True
    else:
        return False

comment = "TODO: 完成这个功能"
if check_comment_type(comment, "TODO"):
    print("这是一个TODO注释")
else:
    print("这不是一个TODO注释")

这是一个TODO注释

comment = "FIXME: 这个地方有问题"
if check_comment_type(comment, "FIXME"):
    print("这是一个FIXME注释")
else:
    print("这不是一个FIXME注释")

这是一个FIXME注释

comment = "HACK: 这个地方需要优化"
if check_comment_type(comment, "HACK"):
    print("这是一个HACK注释")
else:
    print("这不是一个HACK注释")

这是一个HACK注释