"""Config for Theatrical ETL."""
from os import environ

from flows import config as base_config


# SWF parameters
SWF_WORKFLOW_NAME = 'ows_ft_etl_theatrical'
SWF_WORKFLOW_ID = 'ows_ft_etl_theatrical_execution'
SWF_WORKFLOW_TIMEOUT = 600
SWF_WORKFLOW_VERSION = '1.0'

SNS_ACTION_SUCCESS = 'etl-success'
SNS_TOPIC_ARN = environ.get('SNS_TOPIC_ARN')
SNS_REGION_NAME = 'us-east-1'
SNS_SOURCE = 'theatrical'

DATE_FORMAT = '%Y%m%d'  # for the source file names

# Bucket paths
if base_config.ENVIRONMENT == base_config.PROD_ENVIRONMENT:
    UNLOAD_BUCKET = 'prod-orcdbucket'
    UNLOAD_PREFIX = 'feed-drop/RentrakTheatrical/'
    ARCHIVE_BUCKET = 'cucumbers'
    ARCHIVE_PREFIX = 'film-transparency/theatrical/archives/'
else:
    UNLOAD_BUCKET = environ.get('DATA_BUCKET', 'dev-cucumbers')
    UNLOAD_PREFIX = 'film-transparency/theatrical/'
    ARCHIVE_BUCKET = environ.get('DATA_BUCKET', 'dev-cucumbers')
    ARCHIVE_PREFIX = 'film-transparency/theatrical/archives/'

if base_config.ENVIRONMENT == base_config.TEST_ENVIRONMENT:
    THEATRICAL_REVENUE_RAW = 'test_theatrical_revenue_raw'
    THEATRICAL_REVENUE_ETL_LOG = 'test_theatrical_revenue_etl_log'
    THEATRICAL_REVENUE = 'test_theatrical_revenue'
else:
    THEATRICAL_REVENUE_RAW = 'theatrical_revenue_raw'
    THEATRICAL_REVENUE_ETL_LOG = 'theatrical_revenue_etl_log'
    THEATRICAL_REVENUE = 'theatrical_revenue'