OwlCyberSecurity - MANAGER

Edit File: lve_create_db.py

# 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
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)
    # 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')