读写具体内容

Question

我正在尝试读取 file1.txt 的特定内容并将此特定内容写入另一个文件 file2.txt。问题是我在 Bar 之后阅读了整个部分，我只想阅读以 [x] 开头的行并且只阅读 Bar 部分。

源代码

def read_write_file_content():
    data_file = open('file1.txt')
    block = ""
    found = False

    for line in data_file:
        if found:
            if line.strip() == "##### Foo":
                break
            else:
                block += line

        else:
            if line.strip() == "##### Bar:":
                    found = True
                    block = line
    print block




    data_file.close()

view_today()

输入文件 文件1.txt

##### Xyz
* [] Task 112
* [] Cl 221

##### Foo
* [] Task 1
* [x] Clone 2


##### Bar:
* [x] Email to A
* [] Email to B
* [x] Email to C
##### Bob
* [] Task 3
* [x] Clone Bob

输出文件 文件2.txt

##### Bar:
* [x] Email to A
* [x] Email to C

任何建议将不胜感激？谢谢:)

Subsequent question

Answer 1

通过检测部分来打开和关闭found。当found 为True 时，过滤带有'[x]' in line 的行。

found = False

for line in open('file1.txt'):
    line = line.strip()
    if not line:
        continue
    if line.startswith('#####'):
        if line == '##### Bar:':
            found = True
            print(line)
        else:
            if found:
                break
        continue

    if found and '[x]' in line:
        print(line)

Answer 2

您首先需要检测您是否在“Bar”块内。然后，当你在的时候，打印/累积那些以* [x] 开头的行。这是一种方法：

def get_selected_block_entries(lines, block_name,
                               block_prefix='#####', selected_entry_prefix='* [x]'):
    selected_lines = []

    block_marker = '{} {}'.format(block_prefix, block_name)
    for line in lines:
        if line.startswith(block_prefix):
            in_block = line.startswith(block_marker)
            if in_block:
                selected_lines.append(line)
        else:
            if in_block and line.startswith(selected_entry_prefix):
                selected_lines.append(line)

    return selected_lines

with open('file1.txt') as infile, open('file2.txt', 'w') as outfile:
    selected = get_selected_block_entries(infile, 'Bar:')
    print selected    # a list of selected entries within a Bar: block
    outfile.writelines(selected)

当file1.txt包含时运行上述代码：

##### 福 * [] 任务1 * [x] 克隆 2 ＃＃＃＃＃ 酒吧： * [x] 电邮给 A * [] 电邮给 B * [x] 发邮件给 C ##### 福 * [] 任务1 * [x] 克隆 2

打印：

['##### Bar:\n', '* [x] 发邮件给 A\n', '* [x] 发邮件给 C\n']

这是从get_selected_block_entries() 函数返回的列表。同样file2.txt 包含：

＃＃＃＃＃ 酒吧： * [x] 电邮给 A * [x] 发邮件给 C

此输出显示未收集“栏：”块后面的选定条目。

另请注意，如果有多个匹配块，则将从所有匹配块中收集选定条目，例如

get_selected_block_entries(infile, 'Foo') 将返回从 两个 Foo 块中选择的条目：

['##### Foo\n', '* [x] Clone 2\n', '##### Foo\n', '* [x] Clone 2\n']

而且，如果您想从所有块中选择 all 个选定条目，您可以这样做：

get_selected_block_entries(infile, '')

Answer 3

您可能想测试给定行是否以"* [x]" 开头。

import re
section = None
for line in data_file:
    sre = re.match("^#####\s*(\w):\s*",line)
    if sre:
        section = sre.group(1)
    if line.startswith("* [x]") and section == "Bar":
            block += line

查看here 了解有关在 python 中使用正则表达式的更多信息。