import os import logging from dotenv import load_dotenv from datetime import datetime from secrets_manager.python_ext import PythonSecretsManager load_dotenv() DEV_ENVIRONMENT = 'dev' ENVIRONMENT = os.environ.get('Environment', DEV_ENVIRONMENT) SCRIPT_NAME = 'quarterly-oa-account-review' secrets_manager_client = PythonSecretsManager( environment=ENVIRONMENT, service_name=SCRIPT_NAME) if ENVIRONMENT == DEV_ENVIRONMENT: db_config = { 'host': os.environ.get('AR_MYSQL_HOST'), 'user': os.environ.get('AR_MYSQL_USER'), 'password': os.environ.get('AR_MYSQL_PASSWORD'), 'db': 'art_relations' } else: db_config = { 'host': os.environ.get('AR_MYSQL_HOST'), 'user': os.environ.get('AR_MYSQL_USER'), 'password': secrets_manager_client.get_cred('AR_MYSQL_PASSWORD'), 'db': 'art_relations' } # Get the current date and time as a string CURRENT_DATE_TIME = datetime.now().strftime('%Y-%m-%d-%H-%M-%S') S3_BUCKET_NAME = f'{ENVIRONMENT}-pde-audit-review' S3_UPLOAD_FOLDER = f'{SCRIPT_NAME}/{CURRENT_DATE_TIME}/' # create logger logger = logging.getLogger() logger.setLevel(logging.INFO) # create console handler and set level to INFO console_handler = logging.StreamHandler() console_handler.setLevel(logging.INFO) # add formatter to console handler formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s') console_handler.setFormatter(formatter) # add console handler to logger logger.addHandler(console_handler)