improve conversion for initial import

Reviewed-by: gtema <artem.goncharov@gmail.com>
Co-authored-by: Goncharov, Artem argoncha <artem.goncharov@t-systems.com>
Co-committed-by: Goncharov, Artem argoncha <artem.goncharov@t-systems.com>
This commit is contained in:
Goncharov, Artem argoncha 2022-07-25 15:30:38 +00:00 committed by zuul
parent f4a6797818
commit fa175242cc

View File

@ -8,6 +8,7 @@ import os
import pathlib import pathlib
import re import re
import shutil import shutil
import sys
class OTCDocConvertor: class OTCDocConvertor:
@ -246,17 +247,24 @@ class OTCDocConvertor:
'--dest', '--dest',
help='Directory to write resulting files') help='Directory to write resulting files')
self.args = parser.parse_args() self.args = parser.parse_args()
meta_data = json.loads(open(
pathlib.Path(self.args.path, "CLASS.TXT.json")
).read())
metadata_by_uri = dict()
metadata_by_code = dict()
self.doc_images = set()
if self.args.dest: if self.args.dest:
dest = pathlib.Path(self.args.dest) dest = pathlib.Path(self.args.dest)
else: else:
dest = pathlib.Path(self.args.path, 'result') dest = pathlib.Path(self.args.path, 'result')
dest.mkdir(parents=True, exist_ok=True)
metadata_file = pathlib.Path(
self.args.path, "CLASS.TXT.json")
if not metadata_file.exists():
logging.warning(
f"CLASS.TXT.json file is missing in {self.args.path}, "
f"assuming initial import")
sys.exit(0)
meta_data = json.loads(open(metadata_file).read())
metadata_by_uri = dict()
metadata_by_code = dict()
self.doc_images = set()
for f in meta_data: for f in meta_data:
f['new_name'] = self.get_new_name(f['title']) f['new_name'] = self.get_new_name(f['title'])
metadata_by_uri[f['uri']] = f metadata_by_uri[f['uri']] = f