config.py

#!/usr/bin/env python3


import json
import os
import sys

from plexapi.myplex import MyPlexAccount
from getpass import getpass

config_path = os.path.join(os.path.dirname(os.path.realpath(sys.argv[0])), 'config.json')
base_config = {
    'PLEX_SERVER': 'https://plex.your-server.com',
    'PLEX_TOKEN': '',
    'PLEX_LIBRARIES': {},
    'AUDIO_CODEC_SCORES': {'Unknown': 0, 'wmapro': 200, 'mp2': 500, 'mp3': 1000, 'ac3': 1000, 'dca': 2000, 'pcm': 2500,
                           'flac': 2500, 'dca-ma': 4000, 'truehd': 4500, 'aac': 1000, 'eac3': 1250},
    'VIDEO_CODEC_SCORES': {'Unknown': 0, 'h264': 10000, 'h265': 5000, 'hevc': 5000, 'mpeg4': 500, 'vc1': 3000,
                           'vp9': 1000, 'mpeg1video': 250, 'mpeg2video': 250, 'wmv2': 250, 'wmv3': 250, 'msmpeg4': 100,
                           'msmpeg4v2': 100, 'msmpeg4v3': 100},
    'VIDEO_RESOLUTION_SCORES': {'Unknown': 0, '4k': 20000, '1080': 10000, '720': 5000, '480': 3000, 'sd': 1000},
    'FILENAME_SCORES': {},
    'SKIP_LIST': [],
    'SCORE_FILESIZE': True,
    'AUTO_DELETE': False,
    'FIND_DUPLICATE_FILEPATHS_ONLY': False
}
cfg = None


def prefilled_default_config(configs):
    default_config = base_config.copy()

    # Set the token and server url
    default_config['PLEX_SERVER'] = configs['url']
    default_config['PLEX_TOKEN'] = configs['token']

    # Set AUTO_DELETE config option
    default_config['AUTO_DELETE'] = configs['auto_delete']

    # sections
    default_config['PLEX_LIBRARIES'] = [
        'Movies',
        'TV'
    ]

    # filename scores
    default_config['FILENAME_SCORES'] = {
        '*Remux*': 20000,
        '*1080p*BluRay*': 15000,
        '*720p*BluRay*': 10000,
        '*WEB*NTB*': 5000,
        '*WEB*VISUM*': 5000,
        '*WEB*KINGS*': 5000,
        '*WEB*CasStudio*': 5000,
        '*WEB*SiGMA*': 5000,
        '*WEB*QOQ*': 5000,
        '*WEB*TROLLHD*': 2500,
        '*REPACK*': 1500,
        '*PROPER*': 1500,
        '*WEB*TBS*': -1000,
        '*HDTV*': -1000,
        '*dvd*': -1000,
        '*.avi': -1000,
        '*.ts': -1000,
        '*.vob': -5000
    }

    return default_config


def build_config():
    if os.path.exists(config_path):
        return False
    print(f"Dumping default config to: {config_path}")

    configs = dict(url='', token='', auto_delete=False)

    # Get URL
    configs['url'] = input("Plex Server URL: ")

    # Get Credentials for plex.tv
    user = input("Plex Username: ")
    password = getpass('Plex Password: ')

    # Get choice for Auto Deletion
    auto_del = input("Auto Delete duplicates? [y/n]: ")
    while auto_del.strip().lower() not in ['y', 'n']:
        auto_del = input("Auto Delete duplicates? [y/n]: ")
        if auto_del.strip().lower() == 'y':
            configs['auto_delete'] = True
        elif auto_del.strip().lower() == 'n':
            configs['auto_delete'] = False

    account = MyPlexAccount(user, password)
    configs['token'] = account.authenticationToken

    with open(config_path, 'w') as fp:
        json.dump(prefilled_default_config(configs), fp, sort_keys=True, indent=2)

    return True


def dump_config():
    if not os.path.exists(config_path):
        return False
    with open(config_path, 'w') as fp:
        json.dump(cfg, fp, sort_keys=True, indent=2)
    return True


def load_config():
    with open(config_path, 'r') as fp:
        return json.load(fp)


def upgrade_settings(defaults, currents):
    upgraded = False

    def inner_upgrade(default, current, key=None):
        sub_upgraded = False
        merged = current.copy()
        if isinstance(default, dict):
            for k, v in default.items():
                # missing k
                if k not in current:
                    merged[k] = v
                    sub_upgraded = True
                    if not key:
                        print("Added %r config option: %s" % (str(k), str(v)))
                    else:
                        print("Added %r to config option %r: %s" % (str(k), str(key), str(v)))
                    continue
                # iterate children
                if isinstance(v, (dict, list)):
                    did_upgrade, merged[k] = inner_upgrade(default[k], current[k], key=k)
                    sub_upgraded = did_upgrade or sub_upgraded

        elif isinstance(default, list) and key:
            for v in default:
                if v not in current:
                    merged.append(v)
                    sub_upgraded = True
                    print("Added to config option %r: %s" % (str(key), str(v)))
                    continue
        return sub_upgraded, merged

    upgraded, upgraded_settings = inner_upgrade(defaults, currents)
    return upgraded, upgraded_settings


############################################################
# LOAD CFG
############################################################

# dump/load config
if build_config():
    print("Please edit the default configuration before running again!")
    sys.exit(0)
else:
    tmp = load_config()
    upgraded, cfg = upgrade_settings(base_config, tmp)
    if upgraded:
        dump_config()
        print("New config options were added, adjust and restart!")
        sys.exit(0)