mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-18 16:53:38 +00:00
8ddffb7882
Remove py2 part from _23.py Remove more mapped stuff. Replace filter_iter with native filter. Replace map_iter with native map. Remove unidecode from _23 (empty wrapper on py3). Remove map_list and replace with native list(map( for performance reasons. Replace filter_list with list(filter. Replace list_keys with list(. Replace list_values with list(...values()). Replace list_items with list(....items()). Replace ordered_dict with dict. Fix tvinfo base type docs. Remove py2 parts from sg_futures. Remove scandir lib ... it's a sub module of os in py3. Remove PY2 stuff. Ignore unknown ids for characters/persons. Fix tvdb image parsing. Ignore unknown id sources on person page.
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
|
|
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(config.items()))
|
|
|
|
|
|
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
|