python - 用于重构 for 循环和嵌套 if 语句组合的生成器

Question

下一个代码扫描一些日志文件并选择具有某些特定字符的行。

with open(file) as input:
    for line in input:
        if 'REW' in line or 'LOSE' in line:
            <some optional code>
            if 'REW VAL' in line or 'LOSE INV' in line:
                <some code>

我在此基础上编写了一些函数，但每个函数都包含这些重复的代码，所以我认为需要重构。我想我需要制作一台发电机。如何制作这个生成器以允许我更改括号中的代码？

Answer 1

有时候重复并没有那么糟糕!

但无论如何你可以尝试这样的事情:

#!/usr/bin/env python
# script.py

def process_file(filename, func1, func2):
    with open(filename) as f:
        for line in f:
            if '1' in line:
                func1(line)
                if '2' in line:
                    func2(line)


def main():
    counters = {1: 0, 2: 0}
    def func1(line):
        # TODO Add some logic based on line value here
        counters[1] += 1
    def func2(line):
        counters[2] += 1
    process_file('table.csv', func1, func2)
    return counters


if __name__ == '__main__':
    print(main())

如果你有一个文件:

$ cat table.csv
1 just one
1 2 one and two
1
1
0
0
2
2 1
1 0 2
0

并运行脚本:

python script.py

您将得到下一个输出:

{1: 6, 2: 3}

您还可以分解 if 语句的谓词:

def process_file(filename, func1, func2, predicate1, predicate2):
    with open(filename) as f:
        for line in f:
            if predicate1(line):
                func1(line)
                if predicate2(line):
                    func2(line)

def predicate1(line):
    return 'REW' in line or 'LOSE' in line

不要忘记选择好听的函数名称!