"""Configuration for 'save VO to DD DB' lambda."""

import os

import common_config

# direct_delivery database connection setup.
DD_DB_CREDENTIALS = {
    'user': os.environ.get('DD_MYSQL_USER'),
    'password': os.environ.get('DD_MYSQL_PASSWORD'),
    'host': os.environ.get('DD_MYSQL_HOST'),
    'port': os.environ.get('DD_MYSQL_PORT', '3306'),
    'database': os.environ.get('DD_MYSQL_DATABASE'),
}
DD_DB_CONNECTION_STRING = (
    'mysql+pymysql://{user}:{password}@{host}:{port}/{database}'.format(
        user=DD_DB_CREDENTIALS.get('user'),
        password=DD_DB_CREDENTIALS.get('password'),
        host=DD_DB_CREDENTIALS.get('host'),
        port=DD_DB_CREDENTIALS.get('port'),
        database=DD_DB_CREDENTIALS.get('database')))
DD_DB_CONNECT_ARGS = {'charset': 'utf8'}


# limit for batch insert not to exceed max_allowed_packet
BATCH_WRITE_COUNT_LIMIT = int(os.environ.get(
    'BATCH_WRITE_COUNT_LIMIT', 10000))

if common_config.ENVIRONMENT != common_config.TEST_ENVIRONMENT:
    for k, v in DD_DB_CREDENTIALS.items():
        assert v, '{} is not set'.format(k)