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.
158 lines
4.0 KiB
158 lines
4.0 KiB
#!/usr/bin/env python2
|
|
|
|
''' Contains youtube-dlG util functions. '''
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
|
|
|
|
def path_seperator():
|
|
''' Return path seperator for current OS. '''
|
|
return '\\' if os.name == 'nt' else '/'
|
|
|
|
|
|
def fix_path(path):
|
|
''' Add path seperator at the end of the path
|
|
if not exist and replace ~ with user $HOME '''
|
|
if path == '':
|
|
return path
|
|
|
|
if path[-1:] != path_seperator():
|
|
path += path_seperator()
|
|
|
|
path_list = path.split(path_seperator())
|
|
|
|
for index, item in enumerate(path_list):
|
|
if item == '~':
|
|
path_list[index] = get_home()
|
|
|
|
path = path_seperator().join(path_list)
|
|
|
|
return path
|
|
|
|
|
|
def get_home():
|
|
''' Return user $HOME path. '''
|
|
return os.path.expanduser("~")
|
|
|
|
|
|
def abs_path(filename):
|
|
''' Return absolute path. '''
|
|
path = os.path.realpath(os.path.abspath(filename))
|
|
return os.path.dirname(path)
|
|
|
|
|
|
def open_dir(path):
|
|
''' Open path using default file navigator. '''
|
|
if os.name == 'nt':
|
|
os.startfile(path)
|
|
else:
|
|
subprocess.call(('xdg-open', path))
|
|
|
|
|
|
def check_path(path):
|
|
''' Create path if not exist. '''
|
|
if not os.path.exists(path):
|
|
os.makedirs(path)
|
|
|
|
|
|
def get_youtubedl_filename():
|
|
''' Return youtube-dl executable name. '''
|
|
youtubedl_fl = 'youtube-dl'
|
|
if os.name == 'nt':
|
|
youtubedl_fl += '.exe'
|
|
|
|
return youtubedl_fl
|
|
|
|
|
|
def get_config_path():
|
|
''' Return user config path. Windows=AppData, Linux=~/.config. '''
|
|
if os.name == 'nt':
|
|
path = os.getenv('APPDATA')
|
|
else:
|
|
path = fix_path(get_home()) + '.config'
|
|
|
|
return path
|
|
|
|
|
|
def shutdown_sys(password=''):
|
|
''' Shutdown system. !!!On Linux you need to provide
|
|
password for sudo if you dont have admin prev.
|
|
'''
|
|
if os.name == 'nt':
|
|
subprocess.call(['shutdown', '/s', '/t', '1'])
|
|
else:
|
|
if password == '':
|
|
subprocess.call(['/sbin/shutdown', '-h', 'now'])
|
|
else:
|
|
shutdown_proc = subprocess.Popen(
|
|
['sudo', '-S', '/sbin/shutdown', '-h', 'now'],
|
|
stdin=subprocess.PIPE
|
|
)
|
|
shutdown_proc.communicate(password + '\n')
|
|
|
|
|
|
def get_time(seconds):
|
|
''' Return day, hours, minutes, seconds from given seconds. '''
|
|
dtime = {'seconds': 0, 'minutes': 0, 'hours': 0, 'days': 0}
|
|
|
|
if seconds < 60:
|
|
dtime['seconds'] = seconds
|
|
elif seconds < 3600:
|
|
dtime['minutes'] = seconds / 60
|
|
dtime['seconds'] = seconds % 60
|
|
elif seconds < 86400:
|
|
dtime['hours'] = seconds / 3600
|
|
dtime['minutes'] = seconds % 3600 / 60
|
|
dtime['seconds'] = seconds % 3600 % 60
|
|
else:
|
|
dtime['days'] = seconds / 86400
|
|
dtime['hours'] = seconds % 86400 / 3600
|
|
dtime['minutes'] = seconds % 86400 % 3600 / 60
|
|
dtime['seconds'] = seconds % 86400 % 3600 % 60
|
|
|
|
return dtime
|
|
|
|
|
|
def get_icon_path():
|
|
''' Return path to the icon file if exist else return None.
|
|
Search __main__ dir, $XDG_DATA_DIRS, /usr/share/pixmaps in that order. '''
|
|
|
|
SIZES = ('256x256', '128x128', '64x64', '48x48', '32x32', '16x16')
|
|
ICON_NAME = 'youtube-dl-gui_%s.png'
|
|
|
|
ICONS_LIST = [ICON_NAME % size for size in SIZES]
|
|
|
|
# __main__ dir
|
|
path = os.path.join(abs_path(sys.argv[0]), 'icons')
|
|
|
|
for icon in ICONS_LIST:
|
|
icon_path = os.path.join(path, icon)
|
|
|
|
if os.path.exists(icon_path):
|
|
return icon_path
|
|
|
|
# $XDG_DATA_DIRS/icons
|
|
path = os.getenv('XDG_DATA_DIRS')
|
|
|
|
if path is not None:
|
|
for temp_path in path.split(':'):
|
|
temp_path = os.path.join(temp_path, 'icons', 'hicolor')
|
|
|
|
for size in SIZES:
|
|
icon_path = os.path.join(temp_path, size, 'apps')
|
|
icon_path = fix_path(icon_path) + ICON_NAME % size
|
|
|
|
if os.path.exists(icon_path):
|
|
return icon_path
|
|
|
|
# /usr/share/pixmaps
|
|
path = '/usr/share/pixmaps/'
|
|
for icon in ICONS_LIST:
|
|
icon_path = path + icon
|
|
|
|
if os.path.exists(icon_path):
|
|
return icon_path
|
|
|
|
return None
|