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.
88 lines
2.0 KiB
88 lines
2.0 KiB
import hashlib
|
|
import os.path
|
|
import platform
|
|
import re
|
|
|
|
def getDataDir():
|
|
|
|
# Windows
|
|
try:
|
|
from win32com.shell import shellcon, shell
|
|
return os.path.join(shell.SHGetFolderPath(0, shellcon.CSIDL_APPDATA, 0, 0), 'CouchPotato')
|
|
|
|
except ImportError:
|
|
dir = os.path.expanduser("~")
|
|
|
|
# OSX
|
|
if 'darwin' in platform.platform().lower():
|
|
return os.path.join(dir, 'Library', 'Application Support', 'CouchPotato')
|
|
|
|
# Linux
|
|
return os.path.join(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]))
|
|
else:
|
|
current_dst[key] = current_src[key]
|
|
return dst
|
|
|
|
def flattenList(l):
|
|
if isinstance(l, list):
|
|
return sum(map(flattenList, l))
|
|
else:
|
|
return l
|
|
|
|
def md5(text):
|
|
return hashlib.md5(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):
|
|
|
|
if os.path.isfile(txt):
|
|
output = open(txt, 'r')
|
|
txt = output.read()
|
|
output.close()
|
|
|
|
try:
|
|
m = re.search('(?P<id>tt[0-9{7}]+)', txt)
|
|
id = m.group('id')
|
|
if id: return id
|
|
except AttributeError:
|
|
pass
|
|
|
|
return False
|
|
|
|
def tryInt(s):
|
|
try: return int(s)
|
|
except: return s
|
|
|
|
def natsortKey(s):
|
|
return map(tryInt, re.findall(r'(\d+|\D+)', s))
|
|
|
|
def natcmp(a, b):
|
|
return cmp(natsortKey(a), natsortKey(b))
|
|
|