You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
129 lines
3.2 KiB
129 lines
3.2 KiB
from couchpotato.core.logger import CPLog
|
|
import hashlib
|
|
import os.path
|
|
import platform
|
|
import random
|
|
import re
|
|
import string
|
|
import sys
|
|
|
|
log = CPLog(__name__)
|
|
|
|
def getDataDir():
|
|
|
|
# Windows
|
|
if os.name == 'nt':
|
|
return os.path.join(os.environ['APPDATA'], 'CouchPotato')
|
|
|
|
import pwd
|
|
os.environ['HOME'] = pwd.getpwuid(os.geteuid()).pw_dir
|
|
user_dir = os.path.expanduser('~')
|
|
|
|
# OSX
|
|
if 'darwin' in platform.platform().lower():
|
|
return os.path.join(user_dir, 'Library', 'Application Support', 'CouchPotato')
|
|
|
|
# FreeBSD
|
|
if 'freebsd' in sys.platform:
|
|
return os.path.join('/usr/local/', 'couchpotato', 'data')
|
|
|
|
# Linux
|
|
return os.path.join(user_dir, '.couchpotato')
|
|
|
|
def isDict(object):
|
|
return isinstance(object, dict)
|
|
|
|
def mergeDicts(a, b):
|
|
assert isDict(a), isDict(b)
|
|
dst = a.copy()
|
|
|
|
stack = [(dst, b)]
|
|
while stack:
|
|
current_dst, current_src = stack.pop()
|
|
for key in current_src:
|
|
if key not in current_dst:
|
|
current_dst[key] = current_src[key]
|
|
else:
|
|
if isDict(current_src[key]) and isDict(current_dst[key]):
|
|
stack.append((current_dst[key], current_src[key]))
|
|
elif isinstance(current_src[key], list) and isinstance(current_dst[key], list):
|
|
current_dst[key].extend(current_src[key])
|
|
current_dst[key] = removeListDuplicates(current_dst[key])
|
|
else:
|
|
current_dst[key] = current_src[key]
|
|
return dst
|
|
|
|
def removeListDuplicates(seq):
|
|
checked = []
|
|
for e in seq:
|
|
if e not in checked:
|
|
checked.append(e)
|
|
return checked
|
|
|
|
def flattenList(l):
|
|
if isinstance(l, list):
|
|
return sum(map(flattenList, l))
|
|
else:
|
|
return l
|
|
|
|
def md5(text):
|
|
return hashlib.md5(text).hexdigest()
|
|
|
|
def sha1(text):
|
|
return hashlib.sha1(text).hexdigest()
|
|
|
|
def getExt(filename):
|
|
return os.path.splitext(filename)[1][1:]
|
|
|
|
def cleanHost(host):
|
|
if not host.startswith(('http://', 'https://')):
|
|
host = 'http://' + host
|
|
|
|
if not host.endswith('/'):
|
|
host += '/'
|
|
|
|
return host
|
|
|
|
def getImdb(txt, check_inside = True):
|
|
|
|
if check_inside and os.path.isfile(txt):
|
|
output = open(txt, 'r')
|
|
txt = output.read()
|
|
output.close()
|
|
|
|
try:
|
|
id = re.findall('(tt\d{7})', txt)[0]
|
|
return id
|
|
except IndexError:
|
|
pass
|
|
|
|
return False
|
|
|
|
def tryInt(s):
|
|
try: return int(s)
|
|
except: return 0
|
|
|
|
def tryFloat(s):
|
|
try: return float(s) if '.' in s else tryInt(s)
|
|
except: return 0
|
|
|
|
def natsortKey(s):
|
|
return map(tryInt, re.findall(r'(\d+|\D+)', s))
|
|
|
|
def natcmp(a, b):
|
|
return cmp(natsortKey(a), natsortKey(b))
|
|
|
|
def getTitle(library_dict):
|
|
try:
|
|
try:
|
|
return library_dict['titles'][0]['title']
|
|
except:
|
|
log.error('Could not get title for %s', library_dict['identifier'])
|
|
return None
|
|
except:
|
|
log.error('Could not get title for library item: %s', library_dict)
|
|
return None
|
|
|
|
def randomString(size = 8, chars = string.ascii_uppercase + string.digits):
|
|
return ''.join(random.choice(chars) for x in range(size))
|
|
|
|
|