diff options
Diffstat (limited to 'core/module/setup.py')
| -rw-r--r-- | core/module/setup.py | 448 |
1 files changed, 448 insertions, 0 deletions
diff --git a/core/module/setup.py b/core/module/setup.py new file mode 100644 index 000000000..c1f449be4 --- /dev/null +++ b/core/module/setup.py @@ -0,0 +1,448 @@ +#!/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("Choose your Language / Wähle deine Sprache", "en", ["en", "de"]) + translation = gettext.translation("setup", join(self.path, "locale"), languages=[lang]) + translation.install(unicode=(True if sys.getfilesystemencoding().startswith("utf") else False)) + + 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 = 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")) + + 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 "" + + 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 + + return (basic, ssl, captcha, gui, web) + + 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", "fr", "nl", "pl"]) + 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 - 1024 > ssl.key" + 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: + p1 = getpass(_("Password: ")) + + if len(p1) < 4: + print _("Password to short. Use at least 4 symbols.") + continue + + p2 = getpass(_("Password (again): ")) + + 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() |