automata-labs/TaskB04/solution.py

import sys
import re

def solve(lines):
    p = re.compile(r"\d+")
    return [
        ' '.join(re.findall(p, l)) + '\n'
        for l in lines
        if re.search(p, l)
    ]

if __name__ == "__main__":
    lines = []
    
    fp = sys.argv[1]
    assert("polish_wiki_excerpt" in fp)

    print(fp)
    with open(fp, encoding="utf-8") as f:
        lines = f.readlines()
    sol = solve(lines)
    with open("./polish_wiki_excerpt.out", 'w', encoding="utf-8") as f:
        f.writelines(sol)