from datetime import timedelta, datetime
from pathlib import Path

from common import common_config
from utils.conf import var

FLOW_DIR = Path(__file__).parent
FLOW_NAME = 'sme_latam'

DATA_THRESHOLD=timedelta(days=7)
DAG_START_DATE=datetime(2022, 9, 1)

DROP_S3_BUCKET = common_config.COMMON_DROP_S3_BUCKET
DROP_S3_KEY_PATH_TEMPLATE = var('SME_LATAM_DROP_S3_KEY_PATH', 'ftp/sme_latam/{week_year}/')
DROP_FILENAME_TEMPLATE = 'BMAT - Sony - Sony Latam Unified Weekly Chart - Week {week_number}.xlsx'

ARCHIVE_S3_BUCKET = common_config.COMMON_ARCHIVE_S3_BUCKET
ARCHIVE_S3_KEY_PATH_TEMPLATE = var('SME_LATAM_ARCHIVE_S3_KEY_PATH', 'sme_latam/{week_year}/{week_number}/')


TEMP_TABLE_NAME='temp_staging_raw_sme_latam_weekly_unified_chart'
STAGING_RAW_TABLE_NAME='staging_raw_sme_latam_weekly_unified_chart'

AWS_CONN_ID=var('SME_LATAM_AWS_CONN_ID', common_config.COMMON_AWS_CONN_ID)