def admin():
# convert to dict
user = dict((name, convert.to_dict(y)) for name, y in API.get_all_user_data().iteritems())
perms = permlist()
for data in user.itervalues():
data['perms'] = {}
get_permission(data['perms'], data['permission'])
data['perms']['admin'] = data['role'] is 0
s = bottle.request.environ.get('beaker.session')
if bottle.request.environ.get('REQUEST_METHOD', "GET") == "POST":
for name in user:
if bottle.request.POST.get("%s|admin" % name, False):
user[name]['role'] = 0
user[name]['perms']['admin'] = True
elif name != s['name']:
user[name]['role'] = 1
user[name]['perms']['admin'] = False
# set all perms to false
for perm in perms:
user[name]['perms'][perm] = False
for perm in bottle.request.POST.getall("%s|perms" % name):
user[name]['perms'][perm] = True
user[name]['permission'] = set_permission(user[name]['perms'])
API.set_user_permission(name, user[name]['permission'], user[name]['role'])
return render_to_response("admin.html", {"users": user, "permlist": perms}, [pre_processor])
def pre_processor():
s = bottle.request.environ.get('beaker.session')
user = parse_userdata(s)
perms = parse_permissions(s)
status = {}
captcha = False
update = False
plugins = False
if user['is_authenticated']:
status = API.status_server()
info = API.get_info_by_plugin("UpdateManager")
captcha = API.is_captcha_waiting()
# check if update check is available
if info:
if info['pyload'] == "True":
update = info['version']
if info['plugins'] == "True":
plugins = True
return {"user": user,
'status': status,
'captcha': captcha,
'perms': perms,
'url': bottle.request.url,
'update': update,
'plugins': plugins}
def add(request):
package = bottle.request.POST.get('referer', None)
urls = filter(lambda x: x != "", bottle.request.POST['urls'].split("\n"))
if package:
API.add_package(package, urls, 0)
else:
API.generate_and_add_packages(urls, 0)
return ""
def config():
conf = API.get_config()
plugin = API.get_plugin_config()
conf_menu = []
plugin_menu = []
for entry in sorted(conf.keys()):
conf_menu.append((entry, conf[entry].description))
last_name = None
for entry in sorted(plugin.keys()):
desc = plugin[entry].description
name, none, type = desc.partition("_")
if type in API.pyload.pluginManager.TYPES:
if name == last_name or len([a for a, b in plugin.iteritems() if b.description.startswith(name + "_")]) > 1:
desc = name + " (" + type.title() + ")"
else:
desc = name
last_name = name
plugin_menu.append((entry, desc))
accs = API.get_accounts(False)
for data in accs:
if data.trafficleft == -1:
data.trafficleft = _("unlimited")
elif not data.trafficleft:
data.trafficleft = _("not available")
else:
data.trafficleft = format_size(data.trafficleft)
if data.validuntil == -1:
data.validuntil = _("unlimited")
elif not data.validuntil:
data.validuntil = _("not available")
else:
t = time.localtime(data.validuntil)
data.validuntil = time.strftime("%d.%m.%Y - %H:%M:%S", t)
try:
data.options['time'] = data.options['time'][0]
except Exception:
data.options['time'] = "0:00-0:00"
if "limitDL" in data.options:
data.options['limitdl'] = data.options['limitDL'][0]
else:
data.options['limitdl'] = "0"
return render_to_response('settings.html',
{'conf': {'plugin': plugin_menu, 'general': conf_menu, 'accs': accs},
'types': API.get_account_types()},
[pre_processor])
def addcrypted():
package = bottle.request.forms.get('referer', 'ClickNLoad Package')
dlc = bottle.request.forms['crypted'].replace(" ", "+")
dlc_path = os.path.join(API.get_config_value('general', 'download_folder'),
package.replace("/", "").replace("\\", "").replace(":", "") + ".dlc")
with open(dlc_path, "wb") as dlc_file:
dlc_file.write(dlc)
try:
API.add_package(package, [dlc_path], 0)
except Exception:
return bottle.HTTPError()
else:
return "success\r\n"
def info():
conf = API.get_config_dict()
extra = os.uname() if hasattr(os, "uname") else tuple()
data = {"python" : sys.version,
"os" : " ".join((os.name, sys.platform) + extra),
"version" : API.get_server_version(),
"folder" : os.path.abspath(PYLOAD_DIR), "config": os.path.abspath(""),
"download" : os.path.abspath(conf['general']['download_folder']['value']),
"freespace": format_size(API.free_space()),
"remote" : conf['remote']['port']['value'],
"webif" : conf['webui']['port']['value'],
"language" : conf['general']['language']['value']}
return render_to_response("info.html", data, [pre_processor])
def flashgot():
if bottle.request.environ['HTTP_REFERER'] not in ("http://localhost:9666/flashgot", "http://127.0.0.1:9666/flashgot"):
return bottle.HTTPError()
autostart = int(bottle.request.forms.get('autostart', 0))
package = bottle.request.forms.get('package', None)
urls = filter(lambda x: x != "", bottle.request.forms['urls'].split("\n"))
folder = bottle.request.forms.get('dir', None)
if package:
API.add_package(package, urls, autostart)
else:
API.generate_and_add_packages(urls, autostart)
return ""
def call_API(func, args=""):
bottle.response.headers.replace("Content-type", "application/json")
bottle.response.headers.append("Cache-Control", "no-cache, must-revalidate")
s = bottle.request.environ.get('beaker.session')
if 'session' in bottle.request.POST:
s = s.get_by_id(bottle.request.POST['session'])
if not s or not s.get("authenticated", False):
return bottle.HTTPError(403, json_dumps("Forbidden"))
if not API.is_authorized(func, {"role": s['role'], "permission": s['perms']}):
return bottle.HTTPError(401, json_dumps("Unauthorized"))
args = args.split("/")[1:]
kwargs = {}
for x, y in itertools.chain(bottle.request.GET.iteritems(), bottle.request.POST.iteritems()):
if x == "session":
continue
kwargs[x] = urllib.unquote(y)
try:
return call_API(func, *args, **kwargs)
except Exception, e:
traceback.print_exc()
return bottle.HTTPError(500, json_dumps({"error": e.message, "traceback": traceback.format_exc()}))
def login():
bottle.response.headers.replace("Content-type", "application/json")
bottle.response.headers.append("Cache-Control", "no-cache, must-revalidate")
user = bottle.request.forms.get("username")
password = bottle.request.forms.get("password")
remote_addr = bottle.request.environ.get("REMOTE_ADDR", "0")
info = API.check_auth(user, password)
if info:
API.pyload.log.debug(_("API login from IP address: %s") % remote_addr)
else:
API.pyload.log.warning(_("Failed API login from IP address: %s") % remote_addr)
return json_dumps(False)
s = set_session(request, info)
# get the session id by dirty way, documentations seems wrong
try:
sid = s._headers['cookie_out'].split("=")[1].split(";")[0]
return json_dumps(sid)
except Exception:
return json_dumps(True)
def downloads():
root = API.get_config_value("general", "download_folder")
if not os.path.isdir(root):
return base([_('Download directory not found.')])
data = {
'folder': [],
'files': []
}
items = os.listdir(fs_encode(root))
for item in sorted([fs_decode(x) for x in items]):
if os.path.isdir(fs_join(root, item)):
folder = {
'name': item,
'path': item,
'files': []
}
files = os.listdir(fs_join(root, item))
for file in sorted([fs_decode(x) for x in files]):
try:
if os.path.isfile(fs_join(root, item, file)):
folder['files'].append(file)
except Exception:
pass
data['folder'].append(folder)
elif os.path.isfile(os.path.join(root, item)):
data['files'].append(item)
return render_to_response('downloads.html', {'files': data}, [pre_processor])
def login():
if API.get_config_value("webui", "nolocalauth") \
and bottle.request.environ.get("REMOTE_ADDR", "0") in ("127.0.0.1", "localhost"):
set_session(request, "local")
return bottle.redirect("/")
else:
return render_to_response("login.html", proc=[pre_processor])
def get_download(path):
path = urllib.unquote(decode(path))
#@TODO some files can not be downloaded
root = API.get_config_value("general", "download_folder")
path = path.replace("..", "")
return bottle.static_file(fs_encode(path), fs_encode(root))
def addcrypted2():
package = bottle.request.forms.get("source", None)
crypted = bottle.request.forms['crypted']
jk = bottle.request.forms['jk']
crypted = base64.standard_b64decode(urllib.unquote(crypted.replace(" ", "+")))
if JS:
jk = "%s f()" % jk
jk = JS.eval(jk)
else:
try:
jk = re.findall(r"return ('|\")(.+)('|\")", jk)[0][1]
except Exception:
# Test for some known js functions to decode
if jk.find("dec") > -1 and jk.find("org") > -1:
org = re.findall(r"var org = ('|\")([^\"']+)", jk)[0][1]
jk = list(org)
jk.reverse()
jk = "".join(jk)
else:
print "Could not decrypt key, please install py-spidermonkey or ossp-js"
try:
Key = binascii.unhexlify(jk)
except Exception:
print "Could not decrypt key, please install py-spidermonkey or ossp-js"
return "failed"
IV = Key
obj = Crypto.Cipher.AES.new(Key, Crypto.Cipher.AES.MODE_CBC, IV)
result = obj.decrypt(crypted).replace("\x00", "").replace("\r", "").split("\n")
result = filter(lambda x: x != "", result)
try:
if package:
API.add_package(package, result, 0)
else:
API.generate_and_add_packages(result, 0)
except Exception:
return "failed can't add"
else:
return "success\r\n"
def home():
try:
res = [convert.to_dict(x) for x in API.status_downloads()]
except Exception:
s = bottle.request.environ.get('beaker.session')
s.delete()
return bottle.redirect("/login")
for link in res:
if link['status'] == 12:
link['information'] = "%s kB @ %s kB/s" % (link['size'] - link['bleft'], link['speed'])
return render_to_response("home.html", {"res": res}, [pre_processor])
def login_post():
user = bottle.request.forms.get("username")
password = bottle.request.forms.get("password")
remote_addr = bottle.request.environ.get("REMOTE_ADDR", "0")
info = API.check_auth(user, password)
if info:
API.pyload.log.debug(_("WebUI login from IP address: %s") % remote_addr)
else:
API.pyload.log.warning(_("Failed WebUI login from IP address: %s") % remote_addr)
return render_to_response("login.html", {"errors": True}, [pre_processor])
set_session(info)
return bottle.redirect("/")
def logs(item=-1):
s = bottle.request.environ.get('beaker.session')
perpage = s.get('perpage', 34)
reversed = s.get('reversed', False)
warning = ""
conf = API.get_config_value("log", "file_log")
if not conf:
warning = "Warning: File log is disabled, see settings page."
perpage_p = ((20, 20), (50, 50), (100, 100), (250, 250), (0, 'all'))
fro = None
if bottle.request.environ.get('REQUEST_METHOD', "GET") == "POST":
try:
fro = datetime.datetime.strptime(bottle.request.forms['from'], '%d.%m.%Y %H:%M:%S')
except Exception:
pass
try:
perpage = int(bottle.request.forms['perpage'])
s['perpage'] = perpage
reversed = bool(bottle.request.forms.get('reversed', False))
s['reversed'] = reversed
except Exception:
pass
s.save()
try:
item = int(item)
except Exception:
pass
log = API.get_log()
if not perpage:
item = 1
if item < 1 or type(item) is not int:
item = 1 if len(log) - perpage + 1 < 1 else len(log) - perpage + 1
if type(fro) is datetime.datetime: #: we will search for datetime.datetime
item = -1
data = []
counter = 0
perpagecheck = 0
for l in log:
counter += 1
if counter >= item:
try:
date, time, level, message = decode(l).split(" ", 3)
dtime = datetime.datetime.strptime(date + ' ' + time, '%Y-%m-%d %H:%M:%S')
except Exception:
dtime = None
date = '?'
time = ' '
level = '?'
message = l
if item == -1 and dtime is not None and fro <= dtime:
item = counter #: found our datetime.datetime
if item >= 0:
data.append({'line': counter, 'date': date + " " + time, 'level': level, 'message': message})
perpagecheck += 1
if fro is None and dtime is not None: #: if fro not set set it to first showed line
fro = dtime
if perpagecheck >= perpage > 0:
break
if fro is None: #: still not set, empty log?
fro = datetime.datetime.now()
if reversed:
data.reverse()
return render_to_response('logs.html', {'warning': warning, 'log': data, 'from': fro.strftime('%d.%m.%Y %H:%M:%S'),
'reversed': reversed, 'perpage': perpage, 'perpage_p': sorted(perpage_p),
'iprev': 1 if item - perpage < 1 else item - perpage,
'inext': (item + perpage) if item + perpage < len(log) else item,
'color_console': API.get_config_value("log", "color_console")},
[pre_processor])
def collector():
queue = API.get_collector()
queue.sort(key=operator.attrgetter("order"))
return render_to_response('queue.html', {'content': queue, 'target': 0}, [pre_processor])
def checksupport():
url = bottle.request.GET.get("url")
res = API.checkURLs([url])
supported = (not res[0][1] is None)
return str(supported).lower()
