在第一次写入时覆盖文本文件，然后附加到它 - Python

Question

我对文件的第一次写入需要覆盖它，然后我的下一个需要附加到它。但是没有办法知道先写什么。我的写作是在条件语句中。这是我所拥有的：

class MyHTMLParser(HTMLParser):
    def __init__(self):
        HTMLParser.__init__(self)
        self.strict = False
        self.indent = " "
        self.pos = 0
        self.output_file = 'output_sass.txt'

    def handle_starttag(self, tag, attrs):
        if attrs != []:
            for attr in attrs:
                if ('id' in attr):
                    id = attr.index('id')
                    with open(self.output_file, 'w') as the_file:
                        the_file.writelines(self.indent * self.getpos()[1] + '#' + attr[id+1] + ' {' +'\n')
##                    print (self.indent * self.getpos()[1] + "#" + attr[id+1] + " {")
                    self.pos = self.getpos()[1]
                    break
                elif ('class' in attr):
                    clas = attr.index('class')
                    with open(self.output_file, 'w') as the_file:
                        the_file.writelines(self.indent * self.getpos()[1] + "." + attr[clas+1] + " {"+'\n')
##                    print (self.indent * self.getpos()[1] + "." + attr[clas+1] + " {")
                    self.pos = self.getpos()[1]
                    break
                else:
                    with open(self.output_file, 'w') as the_file:
                        the_file.writelines(self.indent * self.getpos()[1] + tag + " {"+'\n')
##                    print (self.indent * self.getpos()[1] + tag + " {")
                    self.pos = self.getpos()[1]
                    break
        else:
           with open(self.output_file, 'w') as the_file:
                the_file.writelines(self.indent * self.getpos()[1] + tag + " {"+'\n')
##            print (self.indent * self.getpos()[1] + tag + " {")
                self.pos = self.getpos()[1]

    def handle_endtag(self, tag):
        with open(self.output_file, 'w') as the_file:
            the_file.writelines(self.indent * self.pos + "}"+'\n')
##        print(self.indent * self.pos + "}")

Answer 1

添加一个包含更改标志的类属性：

import itertools

class MyHTMLParser(HTMLParser):
    def __init__(self, ...):
        ...
        self.modes = itertools.chain('w', itertools.cycle('a'))

    @property
    def mode(self):
        return next(self.modes)

    def handle_starttag:
        ...
        with open(filepath, self.mode) as the_file:  # self.mode is 'w' the first time and 'a' every time thereafter
            # write stuff

Answer 2

使用一些变量first_write = True 并在所有地方检查它。然后改成False。

或者将数据放在某个列表中，最后只写一次。

当然你可以在开头打开文件写入一次（删除之前的文件）并在最后关闭它。

在__init__ 做self.the_file = open(self.output_file, 'w') 并且你有打开的文件，你可以在所有班级访问它。我不只知道程序何时结束以关闭文件self.the_file.close()。也许 HTMLParser 在数据末尾调用了一些函数。

见HTMLParser.close() - 这似乎是关闭文件的好地方。您将不得不覆盖它并可能从原始类HTMLParser 调用close()。