anime-downloader/anime_downloader/config.py

import click
import os
import errno
import json
from anime_downloader import util

APP_NAME = 'anime downloader'
APP_DIR = click.get_app_dir(APP_NAME)
DEFAULT_CONFIG = {
    'dl': {
        'url': False,
        'player': None,
        'skip_download': False,
        'download_dir': '.',
        'quality': '1080p',
        'chunk_size': '10',
        'fallback_qualities': ['720p', '480p', '360p'],
        'force_download': False,
        'file_format': '{anime_title}/{anime_title}_{ep_no}',
        'provider': 'twist.moe',
        'external_downloader': '',
        'aria2c_for_torrents': False,
        'aria2c_log_level': 'error',
        'selescrape_browser': None,
        'selescrape_browser_executable_path': None,
        'selescrape_driver_binary_path': None,
        'speed_limit': 0,
    },
    'ezdl': {
        'file_format': '{animeinfo_anime_title}/{animeinfo_anime_title}_{provider}_{ep_no}',
        'provider': 'twist.moe',
        'ratio': 50,
        'fallback_providers': ['vidstream', 'anime8'],
        'download_metadata': False,
    },
    'watch': {
        'quality': '1080p',
        'fallback_qualities': ['720p', '480p', '360p'],
        'log_level': 'INFO',
        'provider': 'twist.moe',
        'autoplay_next': True,
        'mpv_arguments': '',
        'iina_arguments': '',
        'default_player': 'mpv'
    },
    'gui': {
        'player': 'mpv'
    },
    'siteconfig': {
        'animefrenzy': {
            'version': 'subbed',
            'servers': ['trollvideo', 'mp4upload']
        },
        'egyanime': {
            'version': 'subbed',
            'servers': [
                'clipwatching',
                'streamtape'
            ]
        },
        'animebinge': {
            'version': 'subbed',
            'servers': [
                'mp4upload',
                'xstreamcdn',
                'trollvid'
            ]
        },
        'animixplay': {
            'server': 'vidstream',
            'v5-servers': ['mp4up', 'stape'],
            'version': 'subbed',
        },
        '9anime': {
            'server': 'mp4upload',
            'version': 'subbed',
            'domain_extension': 'to'
        },
        'anistream.xyz': {
            'version': 'subbed',
        },
        'animepahe': {
            'version': 'subbed',
        },
        'animeflv': {
            'version': 'subbed',
            'servers': [
                'stape',
                'natsuki',
                'gocdn',
                'yu',
                'fembed'
            ]
        },
        'anime8': {
            'version': 'subbed',
            'servers': ['fserver', 'fdserver', 'oserver'],
            'include_special_eps': False
        },
        'gogoanime': {
            'server': 'cdn',
            'version': 'subbed'
        },
        'animerush': {
            'servers': ['Mp4uploadHD Video', 'MP4Upload', 'Mp4upload Video', 'Youruploads Video']
        },
        'kickass': {
            'server': 'A-KICKASSANIME',
            'fallback_servers': ['ORIGINAL-QUALITY-V2', 'HTML5-HQ', 'HTML5', 'A-KICKASSANIME', 'BETAPLAYER', 'KICKASSANIME', 'DEVSTREAM'],
            'ext_fallback_servers': ['Mp4Upload', 'Vidcdn', 'Vidstreaming'],
        },
        'kissanimex': {
            'version': 'subbed',
        },
        'animesimple': {
            'version': 'subbed',
            'servers': ['vidstreaming', 'trollvid', 'mp4upload', 'xstreamcdn']
        },
        'darkanime': {
            'version': 'subbed',
            'servers': ['mp4upload', 'trollvid'],
        },
        'dreamanime': {
            'version': 'subbed',
            'server': 'trollvid',
        },
        'ryuanime': {
            'version': 'subbed',
            'servers': [
                'vidstream',
                'mp4upload',
                'xstreamcdn',
                'trollvid'
            ]
        },
        'animekisa': {
            'server': 'gcloud',
            'fallback_servers': ['mp4upload', 'vidstream']
        },

        'watchmovie': {
            'servers': ['vidstream', 'gcloud', 'yourupload', 'hydrax'],
            'version': 'subbed',
        },
        'wcostream': {
            'servers': ['vidstreampro', 'mcloud'],
            'version': 'subbed',
        },
        'animeflix': {
            'server': 'AUEngine',
            'fallback_servers': ['FastStream'],
            'version': 'sub',
        },
        'dubbedanime': {
            'servers': ['vidstream', 'mp4upload', 'trollvid'],
            'version': 'dubbed',
        },
        'animedaisuki': {
            'servers': ['official']
        },
        'nyaa': {
            'filter': 'Trusted only',
            'category': 'English-translated'
        },
        'vidstream': {
            'servers': ['vidstream', 'vidstream_bk', 'gcloud', 'mp4upload', 'cloud9', 'hydrax', 'mixdrop'],
            'version': 'subbed'
        },
        'justdubs': {
            'servers': ['mp4upload', 'gcloud']
        },
        'kisscartoon': {
            'servers': [
                'mpserver',
                'yuserver',
                'oserver',
                'xserver',
                'ptserver'
            ]
        },
        'animevibe': {
            'servers': [
                'vidstream',
                '3rdparty',
                'mp4upload',
                'hydrax',
                'gcloud',
                'fembed'
            ]
        },
        'yify': {
            'servers': [
                'vidstream',
                'yify'
            ]
        },
        'vostfree': {
            'server': 'sibnet'
        },
        'voiranime': {
            'servers': [
                'gounlimited'
            ]
        },
        'dbanimes': {
            'servers': ['mixdrop', 'gounlimited', 'vudeo', 'fembed', 'sendvid']
        },
        'kissanime': {
            'version': 'subbed'
        },
        'animeonline360': {
            'version': 'subbed'
        },
        'animesuge': {
            'version': 'subbed',
            'servers': [
                'mp4upload',
                'streamtape'
            ]
        },
        'animetake': {
            'version': 'subbed',
            'servers': [
                'gstore',
                'hydrax',
                'fembed',
                'vidstreaming',
                'mixdrop',
            ]
        },
        'putlockers': {
            'version': 'dubbed',
            'servers': [
                'eplay',
                'mixdrop'
            ]
        }
    }
}


class _Config:
    CONFIG_FILE = os.path.join(APP_DIR, 'config.json')

    def __init__(self):
        try:
            os.makedirs(APP_DIR)
        except OSError as e:
            if e.errno != errno.EEXIST:
                raise

        if not os.path.exists(self.CONFIG_FILE):
            self._write_default_config()
            self._CONFIG = DEFAULT_CONFIG
        else:
            self._CONFIG = self._read_config()

            def update(gkey, to_be, from_dict):
                if gkey not in to_be:
                    to_be[gkey] = {}
                for key, val in from_dict[gkey].items():
                    if key not in to_be[gkey].keys():
                        to_be[gkey][key] = val
                    elif isinstance(from_dict[gkey][key], dict):
                        update(key, to_be[gkey], from_dict[gkey])

            for key in DEFAULT_CONFIG.keys():
                update(key, self._CONFIG, DEFAULT_CONFIG)
            self.write()
            # Expand environment variables in download_dir (#222)
            download_dir = self._CONFIG['dl']['download_dir']
            download_dir = os.path.expandvars(download_dir)
            self._CONFIG['dl']['download_dir'] = download_dir

    @property
    def CONTEXT_SETTINGS(self):
        return dict(
            default_map=self._CONFIG
        )

    def __getitem__(self, attr):
        return self._CONFIG[attr]

    def write(self):
        self._write_config(self._CONFIG)

    def _write_config(self, config_dict):
        with open(self.CONFIG_FILE, 'w') as configfile:
            json.dump(config_dict, configfile, indent=4, sort_keys=True)

    def _read_config(self):
        with open(self.CONFIG_FILE, 'r') as configfile:
            try:
                conf = json.load(configfile)
            except:
                raise SyntaxWarning(
                    'The config file is not correctly formatted')
        return conf

    def _write_default_config(self):
        if util.check_in_path('aria2c'):
            DEFAULT_CONFIG['dl']['external_downloader'] = '{aria2}'
        self._write_config(DEFAULT_CONFIG)


Config = _Config()