copywriting_production/utils/config.py

import os
import yaml
from pathlib import Path
from pydantic import BaseModel, Field
from typing import List, Dict, Union,Optional,Any
from utils.pydantic_auto_field import AutoLoadModel
from dotenv import load_dotenv
load_dotenv()

class Config(BaseModel):
    storage: str = "local"
    s3_access_key: Optional[str] = os.environ.get("S3_ACCESS_KEY", 'bh9LbfsPHRJgQ44wXIlv')
    s3_secret_key: Optional[str] = os.environ.get("S3_SECRET_KEY", 'N744RZ60T1b4zlcWG2MROCzjEE2mPTdNQCc7Pk3M')
    s3_endpoint: Optional[str] = os.environ.get("S3_ENDPOINT", 'http://vs1.lan:9002')
    chrome_config_ini: Optional[str] = r'G:\code\amazone\copywriting_production\config\dp_conf\9321.ini'
    redis_url: Optional[str] = os.environ.get("REDIS_URL", 'redis://localhost:6379/0')
    def save(self, config_path: Path = None):
        config_path = config_path or get_config_path()
        with open(config_path, "w", encoding="utf-8") as file:
            yaml.dump(self.model_dump(), file)
        return self
            
def get_config_path():
    return os.environ.get('CONFIG_PATH',CONFIG_DIR / "config.yaml") 

def read_config(config_path: Path):
    if isinstance(config_path, str):
        config_path = Path(config_path)
    if not config_path.exists():
        config = Config()
        config.save(config_path)
        return config
    with open(config_path, "r", encoding="utf-8") as file:
        config_dict = yaml.safe_load(file)
    return Config(**config_dict)

CFG = read_config(get_config_path())

def main():
    print(CFG)
    CFG.save()

if __name__ == "__main__":
    main()