from __future__ import annotations

from src.interface.python_project import (
    PythonProjectInterface,
    get_aws_client,
    get_credentials,
    get_environment,
)

SQL_DATA_DIR = "sql_master_data"
WORK_BUCKET_NAME = "senior-datalake-artifacts"
METADATA_DIR = "metadata"
PRODUCT_FILE = "product.yaml"
TABLES_FILE = "tables.yaml"
MASTER_DATAS_FILE = "master_datas.yaml"


class PythonProjectDPSIInterface(PythonProjectInterface):
    def __init__(self):
        super().__init__()
        env = get_environment()
        self.work_bucket_name = (
            WORK_BUCKET_NAME if env == "prod" else f"{WORK_BUCKET_NAME}-{env}"
        )
        self.credentials = get_credentials(env)
        self.s3_client = get_aws_client("s3", self.credentials)