config.py

import envparse
from datetime import datetime, tzinfo
from pathlib import PurePath, Path
from typing import Optional

from vk_api import VkUserPermissions


def _get_local_timezone() -> tzinfo:
    tz = datetime.now().astimezone().tzinfo
    assert tz is not None
    return tz


class Config:
    def __init__(self, config_file_path: Optional[PurePath] = PurePath("env")) -> None:
        env: envparse.Env = envparse.Env()  # If config_file_path is None, read from environment
        if config_file_path is not None:
            env.read_envfile(config_file_path)  # Read everything from file (from scratch)

        self.tg = Config.Telegram(env)
        self.vk = Config.Vk(env)
        self.vk_default_raw_export_file = Path("vk_raw_history.json")
        self.vk_default_export_file = Path("vk_history.pickle")
        self.tg_default_export_file = Path("tg_history.pickle")
        self.tg_default_media_export_dir = Path("exported_media")
        self.default_contacts_mapping_file = Path("contacts_mapping.yaml")

    class Telegram:
        def __init__(self, env: envparse.Env) -> None:
            self.env = env
            self.client_name = "tg"
            self.max_simultaneously_uploaded_files = 10
            self.timezone = _get_local_timezone()
            self.allowed_video_formats = ["mp4", "flv", "ogg", "mkv", "avi"]  # I'm not sure telegram supports them all
            self.video_conversion_format = "mp4"  # All unsupported videos will be converted to this format

        @property
        def api_id(self) -> str:
            return self.env.str("TG_API_ID")  # type: ignore

        @property
        def api_hash(self) -> str:
            return self.env.str("TG_API_HASH")  # type: ignore

    class Vk:
        def __init__(self, env: envparse.Env) -> None:
            self.env = env
            self.client_name = "vk"
            self.scope: int = VkUserPermissions.MESSAGES | VkUserPermissions.VIDEO
            self.api_version = "5.131"
            self.timezone = _get_local_timezone()
            # workers which download media files
            self.max_non_video_workers = 10
            self.max_video_workers = 5

            self.max_video_download_retries = 5
            self.max_video_size_mb = 50
            # See https://github.com/ytdl-org/youtube-dl#format-selection for more information
            self.video_quality = "(bestvideo+bestaudio/best)[filesize<=?{}M]".format(self.max_video_size_mb)

        @property
        def api_id(self) -> int:
            return self.env.int("VK_API_ID")  # type: ignore