# coding=utf-8 # # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT import logging import os import syslog from lvestats.lib import dbengine from lvestats.lib.commons.logsetup import setup_logging from lvestats.lib.server_id import UpdateOrCreateServerID def configure_db(configuration, opts): log = setup_logging(configuration, 'create_db', console_level=logging.INFO) log.info('Configuring connection to db') try: engine = dbengine.make_db_engine(configuration, just_print=opts.print_sql) except dbengine.MakeDbException as e: log.error(str(e)) else: _configure_db(configuration, engine, log, opts) def _configure_db(configuration, engine, log, opts): if opts.validate: validation = dbengine.validate_database(engine) if not (validation['column_error'] or validation['table_error']): log.info("Database structure is correct") elif opts.fix: log.info('Try fix errors in database') dbengine.fix_db(engine, configuration, log_=log) # Re-create DB elif opts.recreate: log.info('Recreate DB') dbengine.recreate_db(engine) syslog.openlog('lve-stats', syslog.LOG_PID, syslog.LOG_AUTHPRIV) syslog.syslog(syslog.LOG_WARNING, f'Database recreated (all tables dropped and recreated) by uid={os.getuid()}') syslog.closelog() # Create DB if no params else: log.info('Creating tables') dbengine.setup_db(engine, cfg=configuration) log.info('Done') if opts.create_missing_tables: dbengine.setup_db(engine, create_missing_tables=True, cfg=configuration) # Update server_id manually if opts.update_serverid_prompt: log.info('Updating server ID') UpdateOrCreateServerID(engine).prompt() log.info('Done') # Update server_id (if server_id in restricted list) automatically elif opts.update_serverid_auto: log.info('Updating server ID with uuid') UpdateOrCreateServerID(engine).auto() log.info('Done')