update check_kontor.py to call Kontor API
This commit is contained in:
+5
-6
@@ -13,16 +13,17 @@ from schema import Base, KontorDB
|
||||
from config import get_logger
|
||||
|
||||
parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
|
||||
parser.add_argument('--recreate-db', action='store_true')
|
||||
parser.add_argument('--verbose', '-v', action='count', default=0)
|
||||
parser.add_argument('--config', '-c', default='kontor-docker')
|
||||
parser.add_argument('--recreate-db', action='store_true')
|
||||
parser.add_argument('--file', '-f', default='~/data.json')
|
||||
args = parser.parse_args()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
logger = get_logger(args.verbose)
|
||||
logger = get_logger(args.verbose, args.config)
|
||||
logger.info('kontor.import started')
|
||||
dirs = PlatformDirs("kontor")
|
||||
dirs = PlatformDirs(args.config)
|
||||
database_config = Path(dirs.user_config_dir, 'database-config.yaml')
|
||||
with open(database_config, 'rt') as f:
|
||||
db_config = yaml.safe_load(f.read())
|
||||
@@ -37,7 +38,5 @@ if __name__ == '__main__':
|
||||
Base.metadata.create_all(bind=engine, checkfirst=True)
|
||||
__session__ = sessionmaker(bind=engine)
|
||||
kontor_db = KontorDB(engine, logger)
|
||||
if args.recreate_db:
|
||||
kontor_db.delete_entries()
|
||||
kontor_db.import_db(args.file)
|
||||
kontor_db.export_db(args.file)
|
||||
logger.info('kontor.import finished')
|
||||
|
||||
Reference in New Issue
Block a user