def server_setup(): """Setup server. Args: None Returns: None """ # Initialize key variables use_mysql = True pool_size = 25 max_overflow = 25 # Get configuration config = jm_configuration.Config() # Create DB connection pool if use_mysql is True: # Add MySQL to the pool engine = create_engine( DBURL, echo=True, encoding='utf8', max_overflow=max_overflow, pool_size=pool_size, pool_recycle=3600) # Try to create the database print('Attempting to create database tables') try: sql_string = ( 'ALTER DATABASE %s CHARACTER SET utf8mb4 ' 'COLLATE utf8mb4_general_ci') % (config.db_name()) engine.execute(sql_string) except: log_message = ( 'Cannot connect to database %s. ' 'Verify database server is started. ' 'Verify database is created. ' 'Verify that the configured database authentication ' 'is correct.') % (config.db_name()) log.log2die(1036, log_message) # Apply schemas print('Applying Schemas') BASE.metadata.create_all(engine) # Insert database entries insert_agent_host() insert_billtype() insert_department() insert_config() # Try some additional statements metadata.insert_oids()
def main(): """Process agent data. Args: None Returns: None """ # Get additional directory to process cli_args = cli() directory = cli_args.directory # Process directory metadata.insert_oids(directory)