如何从字符串中删除“#”注释？

Question

问题： 实现一个名为 stripComments(code) 的 Python 函数，其中 code 是一个参数，它接受一个包含 Python 代码的字符串。函数 stripComments() 返回删除所有 cmets 的代码。

我有：

def stripComments(code):
   code = str(code)
   for line in code:
       comments = [word[1:] for word in code.split() if word[0] == '#']
       del(comments)
stripComments(code)

我不确定如何具体告诉 python 搜索字符串的每一行，并在找到主题标签时删除该行的其余部分。 请帮忙。 :(

Answer 1

你可以通过re.sub函数来实现。

import re
def stripComments(code):
    code = str(code)
    return re.sub(r'(?m)^ *#.*\n?', '', code)

print(stripComments("""#foo bar
bar foo
# buz"""))

(?m) 启用多行模式。 ^ 断言我们处于起步阶段。 <space>*# 匹配开头的字符 #，带或不带前面的空格。 .* 匹配除换行符以外的所有以下字符。用空字符串替换那些匹配的字符会给你删除注释行的字符串。

Answer 2

def remove_comments(filename1, filename2):
    """ Remove all comments beginning with # from filename1 and writes
    the result to filename2
    """

    with open(filename1, 'r') as f:
        lines = f.readlines()

    with open(filename2, 'w') as f:
        for line in lines:
            # Keep the Shebang line
            if line[0:2] == "#!":
                f.writelines(line)
            # Also keep existing empty lines
            elif not line.strip():
                f.writelines(line)
            # But remove comments from other lines
            else:
                line = line.split('#')
                stripped_string = line[0].rstrip()
                # Write the line only if the comment was after the code.
                # Discard lines that only contain comments.
                if stripped_string:
                    f.writelines(stripped_string)
                    f.writelines('\n')