diff options
Diffstat (limited to 'src/ee/tools/part_apply_souffle_post.py')
| -rw-r--r-- | src/ee/tools/part_apply_souffle_post.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/src/ee/tools/part_apply_souffle_post.py b/src/ee/tools/part_apply_souffle_post.py index 9322ccb..8630603 100644 --- a/src/ee/tools/part_apply_souffle_post.py +++ b/src/ee/tools/part_apply_souffle_post.py @@ -10,6 +10,8 @@ from ee.part import Part, load_db, save_db, PartDb def work(in_path: Path, out_path: Path, work_dir: Path): + dialect = "excel-tab" + in_parts: ObjDb[Part] = ObjDb[Part]() uri_idx = in_parts.add_unique_index("uri", lambda p: p.uri) @@ -27,12 +29,12 @@ def work(in_path: Path, out_path: Path, work_dir: Path): for file in out_dir.iterdir(): if file.name.endswith(".csv"): with file.open("r") as f: - reader = csv.reader(f, dialect="excel-tab") + reader = csv.reader(f, dialect=dialect) rows = sorted(reader) tmp = Path(str(file) + ".tmp") with tmp.open("w") as out: - writer = csv.writer(out, dialect="excel-tab") + writer = csv.writer(out, dialect=dialect) writer.writerows(rows) tmp.rename(file) @@ -41,7 +43,7 @@ def work(in_path: Path, out_path: Path, work_dir: Path): new_facts = 0 with (out_dir / "add_fact.csv").open("r") as f: - reader = csv.reader(f, dialect="excel-tab") + reader = csv.reader(f, dialect=dialect) for uri, key, value in reader: try: part = uri_idx.get_single(uri) @@ -52,7 +54,7 @@ def work(in_path: Path, out_path: Path, work_dir: Path): if mpn_csv.is_file(): with mpn_csv.open("r") as f: - reader = csv.reader(f, dialect="excel-tab") + reader = csv.reader(f, dialect=dialect) for uri, mpn in reader: try: part = uri_idx.get_single(uri) |