mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-21 10:13:39 +00:00
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
import ast
|
|
import base64
|
|
import re
|
|
|
|
import sickgear
|
|
from .helpers import try_int
|
|
|
|
from _23 import decode_bytes, decode_str, list_items
|
|
from six import iteritems, text_type
|
|
|
|
|
|
def read_config_string(data):
|
|
|
|
return data and dict([(try_int(x[0]), x[1]) for x in ast.literal_eval(data)]) or {}
|
|
|
|
|
|
def build_config(**kwargs):
|
|
"""
|
|
kwargs is filtered for settings that enable updates to Trakt
|
|
|
|
:param kwargs: kwargs to be filtered for settings that enable updates to Trakt
|
|
:return: dict of parsed config kwargs where k is Trakt account id, v is a parent location
|
|
"""
|
|
|
|
config = {}
|
|
|
|
root_dirs = []
|
|
if sickgear.ROOT_DIRS:
|
|
root_pieces = sickgear.ROOT_DIRS.split('|')
|
|
root_dirs = root_pieces[1:]
|
|
|
|
for item in [re.findall(r'update-trakt-(\d+)-(.*)', k)
|
|
for k, v in iteritems(kwargs) if k.startswith('update-trakt-')]:
|
|
for account_id, location in item:
|
|
account_id = try_int(account_id, None)
|
|
if None is account_id:
|
|
continue
|
|
for cur_dir in root_dirs:
|
|
account_id = try_int(account_id, None)
|
|
if account_id and decode_str(base64.urlsafe_b64encode(decode_bytes(cur_dir))) == location:
|
|
if isinstance(config.get(account_id), list):
|
|
config[account_id] += [cur_dir]
|
|
else:
|
|
config[account_id] = [cur_dir]
|
|
|
|
return config
|
|
|
|
|
|
def build_config_string(config):
|
|
"""
|
|
:param config: dicts of Trakt account id, parent location
|
|
:return: string csv of parsed config kwargs for config file
|
|
"""
|
|
return text_type(list_items(config))
|
|
|
|
|
|
def trakt_collection_remove_account(account_id):
|
|
if account_id in sickgear.TRAKT_UPDATE_COLLECTION:
|
|
sickgear.TRAKT_UPDATE_COLLECTION.pop(account_id)
|
|
sickgear.save_config()
|
|
return True
|
|
return False
|