30 lines
796 B
Python
30 lines
796 B
Python
|
import parse_docx, pis_find
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
def start():
|
||
|
|
print("Local mode activated")
|
||
|
|
working_directory = os.getcwd()
|
||
|
|
print("Working directory: ", working_directory)
|
||
|
|
|
||
|
|
## Get all files in directory
|
||
|
|
files = [f for f in os.listdir(working_directory) if os.path.isfile(os.path.join(working_directory, f))]
|
||
|
|
docx_files = [f for f in files if f.endswith(".docx")]
|
||
|
|
|
||
|
|
results = []
|
||
|
|
|
||
|
|
if docx_files:
|
||
|
|
print(f"Found {len(docx_files)} DOCX files in directory")
|
||
|
|
for file in docx_files:
|
||
|
|
print(file)
|
||
|
|
items = parse_docx.extract_tables(file)
|
||
|
|
results.extend(items)
|
||
|
|
else:
|
||
|
|
print("No DOCX files found")
|
||
|
|
|
||
|
|
print(f"Found {len(results)} PIS Codes in documents")
|
||
|
|
pis_find.run(results)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
start()
|