@ -11,8 +11,10 @@ import sys
log = CPLog ( __name__ ) log = CPLog ( __name__ )
def fnEscape ( pattern ) : def fnEscape ( pattern ) :
return pattern . replace ( ' [ ' , ' [[ ' ) . replace ( ' ] ' , ' []] ' ) . replace ( ' [[ ' , ' [[] ' ) return pattern . replace ( ' [ ' , ' [[ ' ) . replace ( ' ] ' , ' []] ' ) . replace ( ' [[ ' , ' [[] ' )
def link ( src , dst ) : def link ( src , dst ) :
if os . name == ' nt ' : if os . name == ' nt ' :
@ -21,6 +23,7 @@ def link(src, dst):
else : else :
os . link ( src , dst ) os . link ( src , dst )
def symlink ( src , dst ) : def symlink ( src , dst ) :
if os . name == ' nt ' : if os . name == ' nt ' :
import ctypes import ctypes
@ -28,6 +31,7 @@ def symlink(src, dst):
else : else :
os . symlink ( src , dst ) os . symlink ( src , dst )
def getUserDir ( ) : def getUserDir ( ) :
try : try :
import pwd import pwd
@ -37,6 +41,7 @@ def getUserDir():
return os . path . expanduser ( ' ~ ' ) return os . path . expanduser ( ' ~ ' )
def getDownloadDir ( ) : def getDownloadDir ( ) :
user_dir = getUserDir ( ) user_dir = getUserDir ( )
@ -49,6 +54,7 @@ def getDownloadDir():
return user_dir return user_dir
def getDataDir ( ) : def getDataDir ( ) :
# Windows # Windows
@ -68,8 +74,10 @@ def getDataDir():
# Linux # Linux
return os . path . join ( user_dir , ' .couchpotato ' ) return os . path . join ( user_dir , ' .couchpotato ' )
def isDict ( object ) :
return isinstance ( object , dict ) def isDict ( obj ) :
return isinstance ( obj , dict )
def mergeDicts ( a , b , prepend_list = False ) : def mergeDicts ( a , b , prepend_list = False ) :
assert isDict ( a ) , isDict ( b ) assert isDict ( a ) , isDict ( b )
@ -91,6 +99,7 @@ def mergeDicts(a, b, prepend_list = False):
current_dst [ key ] = current_src [ key ] current_dst [ key ] = current_src [ key ]
return dst return dst
def removeListDuplicates ( seq ) : def removeListDuplicates ( seq ) :
checked = [ ] checked = [ ]
for e in seq : for e in seq :
@ -98,26 +107,32 @@ def removeListDuplicates(seq):
checked . append ( e ) checked . append ( e )
return checked return checked
def flattenList ( l ) : def flattenList ( l ) :
if isinstance ( l , list ) : if isinstance ( l , list ) :
return sum ( map ( flattenList , l ) ) return sum ( map ( flattenList , l ) )
else : else :
return l return l
def md5 ( text ) : def md5 ( text ) :
return hashlib . md5 ( ss ( text ) ) . hexdigest ( ) return hashlib . md5 ( ss ( text ) ) . hexdigest ( )
def sha1 ( text ) : def sha1 ( text ) :
return hashlib . sha1 ( text ) . hexdigest ( ) return hashlib . sha1 ( text ) . hexdigest ( )
def isLocalIP ( ip ) : def isLocalIP ( ip ) :
ip = ip . lstrip ( ' htps:/ ' ) ip = ip . lstrip ( ' htps:/ ' )
regex = ' /(^127 \ .)|(^192 \ .168 \ .)|(^10 \ .)|(^172 \ .1[6-9] \ .)|(^172 \ .2[0-9] \ .)|(^172 \ .3[0-1] \ .)|(^::1)$/ ' regex = ' /(^127 \ .)|(^192 \ .168 \ .)|(^10 \ .)|(^172 \ .1[6-9] \ .)|(^172 \ .2[0-9] \ .)|(^172 \ .3[0-1] \ .)|(^::1)$/ '
return re . search ( regex , ip ) is not None or ' localhost ' in ip or ip [ : 4 ] == ' 127. ' return re . search ( regex , ip ) is not None or ' localhost ' in ip or ip [ : 4 ] == ' 127. '
def getExt ( filename ) : def getExt ( filename ) :
return os . path . splitext ( filename ) [ 1 ] [ 1 : ] return os . path . splitext ( filename ) [ 1 ] [ 1 : ]
def cleanHost ( host , protocol = True , ssl = False , username = None , password = None ) : def cleanHost ( host , protocol = True , ssl = False , username = None , password = None ) :
if not ' :// ' in host and protocol : if not ' :// ' in host and protocol :
@ -137,6 +152,7 @@ def cleanHost(host, protocol = True, ssl = False, username = None, password = No
return host return host
def getImdb ( txt , check_inside = False , multiple = False ) : def getImdb ( txt , check_inside = False , multiple = False ) :
if not check_inside : if not check_inside :
@ -161,10 +177,12 @@ def getImdb(txt, check_inside = False, multiple = False):
return False return False
def tryInt ( s , default = 0 ) : def tryInt ( s , default = 0 ) :
try : return int ( s ) try : return int ( s )
except : return default except : return default
def tryFloat ( s ) : def tryFloat ( s ) :
try : try :
if isinstance ( s , str ) : if isinstance ( s , str ) :
@ -173,17 +191,24 @@ def tryFloat(s):
return float ( s ) return float ( s )
except : return 0 except : return 0
def natsortKey ( s ) : def natsortKey ( s ) :
return map ( tryInt , re . findall ( r ' ( \ d+| \ D+) ' , s ) ) return map ( tryInt , re . findall ( r ' ( \ d+| \ D+) ' , s ) )
def natcmp ( a , b ) : def natcmp ( a , b ) :
return cmp ( natsortKey ( a ) , natsortKey ( b ) ) a2 = natsortKey ( a )
b2 = natsortKey ( b )
return ( a2 > b2 ) - ( a2 < b2 )
def toIterable ( value ) : def toIterable ( value ) :
if isinstance ( value , collections . Iterable ) : if isinstance ( value , collections . Iterable ) :
return value return value
return [ value ] return [ value ]
def getTitle ( library_dict ) : def getTitle ( library_dict ) :
try : try :
try : try :
@ -206,6 +231,7 @@ def getTitle(library_dict):
log . error ( ' Could not get title for library item: %s ' , library_dict ) log . error ( ' Could not get title for library item: %s ' , library_dict )
return None return None
def possibleTitles ( raw_title ) : def possibleTitles ( raw_title ) :
titles = [ titles = [
@ -220,16 +246,20 @@ def possibleTitles(raw_title):
return list ( set ( titles ) ) return list ( set ( titles ) )
def randomString ( size = 8 , chars = string . ascii_uppercase + string . digits ) : def randomString ( size = 8 , chars = string . ascii_uppercase + string . digits ) :
return ' ' . join ( random . choice ( chars ) for x in range ( size ) ) return ' ' . join ( random . choice ( chars ) for x in range ( size ) )
def splitString ( str , split_on = ' , ' , clean = True ) : def splitString ( str , split_on = ' , ' , clean = True ) :
list = [ x . strip ( ) for x in str . split ( split_on ) ] if str else [ ] l = [ x . strip ( ) for x in str . split ( split_on ) ] if str else [ ]
return filter ( None , list ) if clean else list return filter ( None , l ) if clean else l
def dictIsSubset ( a , b ) : def dictIsSubset ( a , b ) :
return all ( [ k in b and b [ k ] == v for k , v in a . items ( ) ] ) return all ( [ k in b and b [ k ] == v for k , v in a . items ( ) ] )
def isSubFolder ( sub_folder , base_folder ) : def isSubFolder ( sub_folder , base_folder ) :
# Returns True if sub_folder is the same as or inside base_folder # Returns True if sub_folder is the same as or inside base_folder
return base_folder and sub_folder and os . path . normpath ( base_folder ) . rstrip ( os . path . sep ) + os . path . sep in os . path . normpath ( sub_folder ) . rstrip ( os . path . sep ) + os . path . sep return base_folder and sub_folder and os . path . normpath ( base_folder ) . rstrip ( os . path . sep ) + os . path . sep in os . path . normpath ( sub_folder ) . rstrip ( os . path . sep ) + os . path . sep
Ruud
11 years ago