#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 3 of the License,
or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, see <http://www.gnu.org/licenses/>.
@author: RaNaN
"""
from getpass import getpass
import gettext
from hashlib import sha1
import os
from os import remove
from os import makedirs
from os.path import abspath
from os.path import dirname
from os.path import exists
from os.path import isfile
from os.path import join
import random
import re
from subprocess import PIPE
from subprocess import call
import sys
from sys import exit
class Setup():
"""
pyLoads initial setup configuration assistent
"""
def __init__(self, path, config):
self.path = path
self.config = config
def start(self):
lang = self.ask(u"Choose your Language / Wähle deine Sprache", "en", ["en", "de","it","pl"])
translation = gettext.translation("setup", join(self.path, "locale"), languages=["en", lang])
translation.install(True)
print ""
print _("Welcome to the pyLoad Configuration Assistent.")
print _("It will check your system and make a basic setup in order to run pyLoad.")
print ""
print _("The value in brackets [] always is the default value,")
print _("in case you don't want to change it or you are unsure what to choose, just hit enter.")
print _("Don't forget: You can always rerun this assistent with --setup or -s parameter, when you start pyLoadCore.")
print _("If you have any problems with this assistent hit STRG-C,")
print _("to abort and don't let him start with pyLoadCore automatically anymore.")
print ""
print _("When you are ready for system check, hit enter.")
raw_input()
basic, ssl, captcha, gui, web, js = self.system_check()
print ""
if not basic:
print _("You need pycurl, sqlite and python 2.5, 2.6 or 2.7 to run pyLoad.")
print _("Please correct this and re-run pyLoad.")
print _("Setup will now close.")
raw_input()
return False
raw_input(_("System check finished, hit enter to see your status report."))
print ""
print _("## Status ##")
print ""
avail = []
if self.check_module("Crypto"): avail.append(_("container decrypting"))
if ssl: avail.append(_("ssl connection"))
if captcha: avail.append(_("automatic captcha decryption"))
if gui: avail.append(_("GUI"))
if web: avail.append(_("Webinterface"))
if js: avail.append(_("extended Click'N'Load"))
string = ""
for av in avail:
string += ", "+av
print _("Features available:") + string[1:]
print ""
if len(avail) < 5:
print _("Featues missing: ")
print
if not self.check_module("Crypto"):
print _("no py-crypto available")
print _("You need this if you want to decrypt container files.")
print ""
if not ssl:
print _("no SSL available")
print _("This is needed if you want to establish a secure connection to core or webinterface.")
print _("If you only want to access locally to pyLoad ssl is not usefull.")
print ""
if not captcha:
print _("no Captcha Recognition available")
print _("Only needed for some hosters and as freeuser.")
print ""
if not gui:
print _("Gui not available")
print _("The Graphical User Interface.")
print ""
if not web:
print _("no Webinterface available")
print _("Gives abillity to control pyLoad with your webbrowser.")
print ""
if not js:
print _("no JavaScript engine found")
print _("You will need this for some Click'N'Load links. Install Spidermonkey or ossp-js")
print _("You can abort the setup now and fix some dependicies if you want.")
con = self.ask(_("Continue with setup?"), "y", bool=True)
if not con:
return False
print ""
print _("Do you want to change the config path? Current is %s" % abspath(""))
print _("If you use pyLoad on a server or the home partition lives on an iternal flash it may be a good idea to change it.")
path = self.ask(_("Change config path?"), "n", bool=True)
if path:
self.conf_path()
#calls exit when changed
print ""
print _("Do you want to configure basic settings?")
print _("This is recommend for first run.")
con = self.ask(_("Make basic setup?"), "y", bool=True)
if con:
self.conf_basic()
if ssl:
print ""
print _("Do you want to configure ssl?")
ssl = self.ask(_("Configure ssl?"), "n", bool=True)
if ssl:
self.conf_ssl()
if web:
print ""
print _("Do you want to configure webinterface?")
web = self.ask(_("Configure webinterface?"), "y", bool=True)
if web:
self.conf_web()
print ""
print _("Setup finished successfully.")
print _("Hit enter to exit and restart pyLoad")
raw_input()
return True
def system_check(self):
""" make a systemcheck and return the results"""
print _("## System Check ##")
python = False
if sys.version_info > (2, 7):
print _("Your python version is to new, Please use Python 2.6/2.7")
python = False
elif sys.version_info < (2, 5):
print _("Your python version is to old, Please use at least Python 2.5")
python = False
else:
print _("Python Version: OK")
python = True
curl = self.check_module("pycurl")
self.print_dep("pycurl", curl)
sqlite = self.check_module("sqlite3")
self.print_dep("sqlite3", sqlite)
basic = python and curl and sqlite
print ""
crypto = self.check_module("Crypto")
self.print_dep("pycrypto", crypto)
ssl = self.check_module("OpenSSL")
self.print_dep("OpenSSL", ssl)
print ""
pil = self.check_module("Image")
self.print_dep("py-imaging", pil)
if os.name == "nt":
tesser = self.check_prog([join(pypath, "tesseract", "tesseract.exe"), "-v"])
else:
tesser = self.check_prog(["tesseract", "-v"])
self.print_dep("tesseract", tesser)
captcha = pil and tesser
print ""
gui = self.check_module("PyQt4")
self.print_dep("PyQt4", gui)
print ""
web = self.check_module("django")
try:
import django
if django.VERSION < (1, 1):
print _("Your django version is to old, please upgrade to django 1.1")
web = False
elif django.VERSION > (1, 3):
print _("Your django version is to new, please use django 1.2")
web = False
except:
web = False
self.print_dep("django", web)
web = web and sqlite
js = self.check_prog(["js", "-v"])
self.print_dep(_("JS engine"), js)
return (basic, ssl, captcha, gui, web, js)
def conf_basic(self):
print ""
print _("## Basic Setup ##")
print ""
print _("The following logindata are only valid for CLI and GUI, but NOT for webinterface.")
self.config.username = self.ask(_("Username"), "User")
self.config.password = self.ask("", "", password=True)
print ""
self.config["general"]["language"] = self.ask(_("Language"), "en", ["en", "de", "it", "pl", "es"])
#@TODO get language from config type
self.config["general"]["download_folder"] = self.ask(_("Downloadfolder"), "Downloads")
self.config["general"]["max_downloads"] = self.ask(_("Max parallel downloads"), "3")
print _("You should disable checksum proofing, if you have low hardware requirements.")
self.config["general"]["checksum"] = self.ask(_("Proof checksum?"), "y", bool=True)
reconnect = self.ask(_("Use Reconnect?"), "n", bool=True)
self.config["reconnect"]["activated"] = reconnect
if reconnect:
self.config["reconnect"]["method"] = self.ask(_("Reconnect script location"), "./reconnect.sh")
def conf_web(self):
print ""
print _("## Webinterface Setup ##")
db_path = "pyload.db"
is_db = isfile(db_path)
db_setup = True
if is_db:
print _("You already have a database for the webinterface.")
db_setup = self.ask(_("Do you want to delete it and make a new one?"), "n", bool=True)
if db_setup:
if is_db: remove(db_path)
from django import VERSION
import sqlite3
if VERSION[:2] < (1,2):
from module.web import syncdb_django11 as syncdb
else:
from module.web import syncdb
from module.web import createsuperuser
print ""
syncdb.handle_noargs()
print _("If you see no errors, your db should be fine and we're adding an user now.")
username = self.ask(_("Username"), "User")
createsuperuser.handle(username, "email@trash-mail.com")
password = self.ask("", "", password=True)
salt = reduce(lambda x, y: x + y, [str(random.randint(0, 9)) for i in range(0, 5)])
hash = sha1(salt + password)
password = "sha1$%s$%s" % (salt, hash.hexdigest())
conn = sqlite3.connect(db_path)
c = conn.cursor()
c.execute('UPDATE "main"."auth_user" SET "password"=? WHERE "username"=?', (password, username))
conn.commit()
c.close()
print ""
self.config["webinterface"]["activated"] = self.ask(_("Activate webinterface?"), "y", bool=True)
print ""
print _("Listen address, if you use 127.0.0.1 or localhost, the webinterface will only accessible locally.")
self.config["webinterface"]["host"] = self.ask(_("Address"), "0.0.0.0")
self.config["webinterface"]["port"] = self.ask(_("Port"), "8000")
#@TODO setup for additional webservers
def conf_ssl(self):
print ""
print _("## SSL Setup ##")
print ""
print _("Execute these commands from pyLoad folder to make ssl certificates:")
print ""
print "openssl genrsa -out ssl.key 1024"
print "openssl req -new -key ssl.key -out ssl.csr"
print "openssl req -days 36500 -x509 -key ssl.key -in ssl.csr > ssl.crt "
print ""
print _("If you're done and everything went fine, you can activate ssl now.")
self.config["ssl"]["activated"] = self.ask(_("Activate SSL?"), "y", bool=True)
def set_user(self):
translation = gettext.translation("setup", join(self.path, "locale"), languages=[self.config["general"]["language"]])
translation.install(unicode=(True if sys.getfilesystemencoding().startswith("utf") else False))
print _("Setting new username and password")
print ""
self.config.username = self.ask(_("Username"), "User")
self.config.password = self.ask("", "", password=True)
self.config.save()
def conf_path(self, trans=False):
if trans:
translation = gettext.translation("setup", join(self.path, "locale"), languages=[self.config["general"]["language"]])
translation.install(unicode=(True if sys.getfilesystemencoding().startswith("utf") else False))
print _("Setting new configpath, current configuration will not be transfered!")
path = self.ask(_("Configpath"), abspath(""))
try:
path = join(pypath, path)
if not exists(path):
makedirs(path)
f = open(join(pypath, "module","config", "configdir"), "wb")
f.write(path)
f.close()
print _("Configpath changed, setup will now close, please restart to go on.")
print _("Press Enter to exit.")
raw_input()
exit()
except Exception, e:
print _("Setting config path failed: %s") % str(e)
def print_dep(self, name, value):
"""Print Status of dependency"""
if value:
print _("%s: OK") % name
else:
print _("%s: missing") % name
def check_module(self, module):
try:
__import__(module)
return True
except:
return False
def check_prog(self, command):
pipe = PIPE
try:
call(command, stdout=pipe, stderr=pipe)
return True
except:
return False
def ask(self, qst, default, answers=[], bool=False, password=False):
"""produce one line to asking for input"""
if answers:
info = "("
for i, answer in enumerate(answers):
info += (", " if i != 0 else "") + str((answer == default and "[%s]" % answer) or answer)
info += ")"
elif bool:
if default == "y":
info = "([y]/n)"
else:
info = "(y/[n])"
else:
info = "[%s]" % default
if password:
p1 = True
p2 = False
while p1 != p2:
if os.name == "nt":
qst = str("Password: ") #no unicode on windows
else:
qst = _("Password: ")
p1 = getpass(qst.encode("utf-8"))
if len(p1) < 4:
print _("Password to short. Use at least 4 symbols.")
continue
if os.name == "nt":
qst = str("Password (again): ")
else:
qst = _("Password (again): ")
p2 = getpass(qst.encode("utf-8"))
if p1 == p2:
return p1
else:
print _("Passwords did not match.")
while True:
input = raw_input(qst + " %s: " % info)
if input.strip() == "":
input = default
if bool:
if re.match(r"(y|yes|j|ja|true)", input.lower().strip()):
return True
elif re.match(r"(n|no|nein|false)", input.lower().strip()):
return False
else:
print _("Invalid Input")
continue
if not answers:
return input
else:
if input in answers:
return input
else:
print _("Invalid Input")
if __name__ == "__main__":
test = Setup(join(abspath(dirname(__file__)), ".."), None)
test.start()