# from airflow import DAG
from ewah.constants import EWAHConstants as EC
from ewah.dag_factories.dag_factory_atomic import dag_factory_atomic
from ewah.operators.amazon_seller_central import (
    EWAHAmazonSellerCentralReportsAPIOperator as operator,
)

from datetime import datetime, timedelta, date
import pytz

factory_kwargs = {
    "dwh_engine": EC.DWH_ENGINE_POSTGRES,
    "dwh_conn_id": "dwh",
    "start_date": datetime(2022, 1, 1, tzinfo=pytz.utc),
    "el_operator": operator,
    "schedule_interval": timedelta(days=1),  # Runs every day during the week
    "default_args": {
        "retries": 2,
        "retry_delay": timedelta(minutes=5),
        "email_on_retry": False,
        "email_on_failure": False,
        "email": [],
        "owner": "EWAH",
    },
}

dag_sp_api_fr = dag_factory_atomic(
    dag_name="EL_SP_API_FR",
    target_schema_name="raw_sp_api_reporting_new",
    operator_config={
        "general_config": {
            "source_conn_id": "seller_central",
            "marketplace_region": "DE",
            "reload_data_from": datetime(2024, 2, 1, tzinfo=pytz.utc),
        },
        "tables": {
            "listings": {
                "report_name": "listings",
                "ewah_options": {"add_bsr": True},
            },
            "fba_returns": {
                "report_name": "fba_returns",
                "reload_data_from": datetime(2022, 1, 1, tzinfo=pytz.utc),
            },
            "orders": {
                "report_name": "orders",
                # "reload_data_from": datetime(2022, 1, 1, tzinfo=pytz.utc),
                "extract_strategy": "subsequent",
            },
            "sales_and_traffic": {
                "report_name": "sales_and_traffic",
                # "reload_data_from": datetime(2022, 7, 22, tzinfo=pytz.utc),
                "extract_strategy": "subsequent",
                "source_conn_id": "seller_central_ne",
            },
        },
    },
    **factory_kwargs
)[0]