"""Lambda shared module."""

import contextlib
import logging
import sys

import pymysql

logger = logging.getLogger()
logger.setLevel(logging.INFO)


@contextlib.contextmanager
def reportsdd_connection(
        db_host, user, passwd, db, connect_timeout=5):
    """Context manager for reportsdd connection objects."""
    try:
        conn = pymysql.connect(
            db_host, user=user, passwd=passwd, db=db,
            connect_timeout=connect_timeout)
        yield conn
        try:
            conn.close()
        except:
            pass
    except:
        logger.error(
            'ERROR: Unexpected error: Could not connect to MySql instance.')
        sys.exit()