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)
