99 lines
4.0 KiB
Python
99 lines
4.0 KiB
Python
import argparse
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
import configparser
|
|
|
|
"""
|
|
Dieses Programm:
|
|
- Kopiert die CSV-Datei mit allen RuleDesigner-Nummern
|
|
- Fragt Sivas ab und generiert dabei eine json-Datei, in der alle Teilenummern aufgelistet sind,
|
|
die in Verbindung mit einer angegebenen Teilenummer migriert werden müssen.
|
|
"""
|
|
def get_blacklist(outdir):
|
|
"""read out list of sivas Ids which are not allowed from network drive
|
|
'//ruledesigner//DF_Daten//onfigurator//ustomVaulting//RP_BOM.ini'
|
|
within the section 'Settings'
|
|
in the key 'PartNumbersToDrop'
|
|
"""
|
|
blacklist_net_share = os.environ.get("BLACKLIST_DATABASE_SOURCE")
|
|
blacklist_ini_filename = os.environ.get("BLACKLIST_DATABASE_NAME")
|
|
blacklist_source_path = os.path.join(blacklist_net_share, blacklist_ini_filename)
|
|
config = configparser.RawConfigParser()
|
|
try:
|
|
config.read(blacklist_source_path)
|
|
except configparser.DuplicateOptionError:
|
|
pass
|
|
content = config.get('Settings','PartNumbersToDrop')
|
|
blacklist_str = content[0] # warum das als Liste erkannt wird wissen die Götter...
|
|
if len(blacklist_str) == 0 or len(content) == 0:
|
|
print(f"import of '{blacklist_ini_filename}' from '{blacklist_net_share}' failed")
|
|
exit
|
|
|
|
blacklist = blacklist_str.split(';')
|
|
|
|
outfile = os.path.join(outdir, 'blacklist.txt')
|
|
with open(outfile, 'w', encoding='utf-8') as fh:
|
|
fh.write('\n'.join(blacklist))
|
|
|
|
def get_rd_dbase(out_dir):
|
|
|
|
rd_csv_source_dir = os.environ.get("RD_DATABASE_SOURCE")
|
|
rd_csv_filename = os.environ.get("RD_DATABASE_NAME")
|
|
csv_source_path = os.path.join(rd_csv_source_dir, rd_csv_filename)
|
|
|
|
if os.path.isfile(csv_source_path):
|
|
csv_destiny_path = os.path.join(out_dir, rd_csv_filename)
|
|
|
|
if os.path.isdir(out_dir):
|
|
print("Copying CSV-file of RuleDesigner database...")
|
|
shutil.copyfile(csv_source_path, csv_destiny_path)
|
|
print("done")
|
|
else:
|
|
print(f"Can't copy RuleDesigner-database, because output directory ${out_dir} doesn't exist. Check path of %CREMIG_DATA% in cremig_setenv.bat")
|
|
else:
|
|
print(f"no such file ${rd_csv_source_dir} of environment variable RD_DATABASE_SOURCE. Check cremig_setenv.bat")
|
|
|
|
|
|
def get_sivas_dbase(sivas_id, out_dir):
|
|
sivas_exe_path = os.environ.get('SIVAS_DATABASE_QUERY')
|
|
|
|
if os.path.isfile(sivas_exe_path):
|
|
|
|
if os.path.isdir(out_dir):
|
|
print(f"Attempting to generate a json-file of all parts to be migrated in relation with '{sivas_id}'")
|
|
subprocess.run(f"{sivas_exe_path} -o {out_dir} --format json -number {sivas_id}")
|
|
|
|
if os.path.isfile(os.path.join(out_dir, sivas_id + ".json")):
|
|
print("done")
|
|
else:
|
|
print(f"Failure - Couldn't generate {sivas_id}.json. Check for any spelling/typing errors. Otherwise given number is a part without a BOM.")
|
|
exit()
|
|
else:
|
|
print(f"'Output directory for .json-file {out_dir}' doesn't exist.")
|
|
else:
|
|
print(f"Can't execute SIVAS-database query. Check if you have established a VPN connection to the company's network and check the path of %SIVAS_DATABASE_QUERY% in cremig_setenv.bat")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description='fetches data from sivas or ruledesigner fusion', prog='update_database')
|
|
parser.add_argument('-s', '--sivas', action='store', help='fetch data of this id from sivas fo local database folder', metavar='id')
|
|
parser.add_argument('-r', '--rd', action='store_true', help='fetch list of existing parts from network drive')
|
|
parser.add_argument('-b', '--blacklist', action='store_true', help='fetch list of not allowed parts from network drive')
|
|
|
|
args = parser.parse_args()
|
|
|
|
out_dir = os.environ.get('CREMIG_DATA')
|
|
|
|
if args.rd:
|
|
get_rd_dbase(out_dir)
|
|
|
|
elif args.sivas:
|
|
get_sivas_dbase(args.sivas, out_dir)
|
|
|
|
elif args.blacklist:
|
|
get_blacklist(out_dir)
|
|
|
|
else:
|
|
parser.print_help()
|