Commit 96b6a383 authored by legoktm's avatar legoktm

Only read .xml files

parent 97623c33
......@@ -25,7 +25,7 @@ import xml.etree.cElementTree as etree
def find_files(path: str):
return sorted(os.path.join(path, fname) for fname in os.listdir(path))
return sorted(os.path.join(path, fname) for fname in os.listdir(path) if fname.endswith('.xml'))
def process_file(fname):
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment