From b7a0e83c68f77c19b89c7c4c02709ea58e3d3fae Mon Sep 17 00:00:00 2001 From: Oskar Hahn Date: Tue, 14 Aug 2012 14:19:59 +0200 Subject: [PATCH] changed the main script --- openslides/main.py | 221 +++++++++++++++--------------- openslides/openslides_settings.py | 2 +- openslides/participant/models.py | 1 - openslides/participant/views.py | 4 +- start.py | 4 +- 5 files changed, 112 insertions(+), 120 deletions(-) diff --git a/openslides/main.py b/openslides/main.py index 29e431681..08593d8c6 100644 --- a/openslides/main.py +++ b/openslides/main.py @@ -21,13 +21,9 @@ import time import threading import base64 import webbrowser -from contextlib import nested import django.conf from django.core.management import execute_from_command_line -from django.utils.crypto import get_random_string - -import openslides CONFIG_TEMPLATE = """ from openslides.openslides_settings import * @@ -66,6 +62,7 @@ INSTALLED_APPS += INSTALLED_PLUGINS KEY_LENGTH = 30 + def main(argv=None): if argv is None: argv = sys.argv[1:] @@ -74,58 +71,76 @@ def main(argv=None): description="Run openslides using django's builtin webserver") parser.add_option("-a", "--address", help="IP Address to listen on") parser.add_option("-p", "--port", type="int", help="Port to listen on") - parser.add_option("--nothread", action="store_true", - help="Do not use threading") - parser.add_option("--syncdb", action="store_true", + parser.add_option( + "--syncdb", action="store_true", help="Update/create database before starting the server") - parser.add_option("--reset-admin", action="store_true", + parser.add_option( + "--reset-admin", action="store_true", help="Make sure the user 'admin' exists and uses 'admin' as password") + parser.add_option("-s", "--settings", help="Path to the setting.") + parser.add_option( + "--no-reload", action="store_true", help="Do not reload the development server") opts, args = parser.parse_args(argv) - if not args: - main(argv + ['init', 'openslides']) - main(argv + ['start', 'openslides']) - return 0 + if args: + sys.stderr.write("This command does not take arguments!\n\n") + parser.print_help() + sys.exit(1) - command = args[0] + # Find the path to the settings + settings = opts.settings or \ + os.path.expanduser('~/.openslides/openslidessettings.py') + # Create settings if necessary + if not os.path.exists(settings): + create_settings(settings) + + # Set the django environment to the settings + setup_django_environment(settings) + + # Find url to openslides addr, port = detect_listen_opts(opts.address, opts.port) if port == 80: - url = "http://%s" % (addr, ) + url = "http://%s" % addr else: url = "http://%s:%d" % (addr, port) - try: - environment_name = args[1] - except IndexError: - environment_name = None + # Create Database if necessary + if not database_exists() or opts.syncdb: + run_syncdb() + set_system_url(url) + create_or_reset_admin_user() - if command == 'init': - if check_environment(environment_name): - print "'%s' is allready an environment" % environment_name - return 1 - create_environment(environment_name or 'openslides', url) + # Reset Admin + elif opts.reset_admin: + create_or_reset_admin_user() - elif command == 'start': - set_settings_environment(environment_name or os.getcwd()) - if not check_environment(): - print "'%s' is not a valid OpenSlides environment." % environment_name - sys.exit(1) - # NOTE: --insecure is needed so static files will be served if - # DEBUG is set to False - argv = ["", "runserver", "--noreload", "--insecure"] - if opts.nothread: - argv.append("--nothread") - - argv.append("%s:%d" % (addr, port)) - - start_browser(url) - execute_from_command_line(argv) + # Start OpenSlides + if opts.no_reload: + extra_args = ['--noreload'] + else: + extra_args = [] + start_openslides(addr, port, start_browser_url=url, extra_args=extra_args) -def set_settings_environment(environment): - sys.path.append(environment) - os.environ[django.conf.ENVIRONMENT_VARIABLE] = 'settings' +def create_settings(settings): + path_to_dir = os.path.dirname(settings) + + setting_content = CONFIG_TEMPLATE % dict( + default_key=base64.b64encode(os.urandom(KEY_LENGTH)), + dbpath=os.path.join(path_to_dir, 'database.db')) + + if not os.path.exists(path_to_dir): + os.makedirs(path_to_dir) + + with open(settings, 'w') as settings_file: + settings_file.write(setting_content) + + +def setup_django_environment(settings): + sys.path.append(os.path.dirname(settings)) + setting_module = os.path.basename(settings)[:-3] + os.environ[django.conf.ENVIRONMENT_VARIABLE] = setting_module def detect_listen_opts(address, port): @@ -150,91 +165,29 @@ def detect_listen_opts(address, port): return address, port -def start_browser(url): - browser = webbrowser.get() - def f(): - time.sleep(1) - browser.open(url) - - t = threading.Thread(target = f) - t.start() - - -def create_environment(environment, url=None): - setting_content = CONFIG_TEMPLATE % dict( - default_key=base64.b64encode(os.urandom(KEY_LENGTH)), - dbpath=os.path.join(os.path.abspath(environment), 'database.db')) - - dirname = environment - if dirname and not os.path.exists(dirname): - os.makedirs(dirname) - - settings = os.path.join(environment, 'settings.py') - if not os.path.exists(settings): - with open(settings, 'w') as fp: - fp.write(setting_content) - - set_settings_environment(environment) - - run_syncdb(url) - create_or_reset_admin_user() - - -def run_syncdb(url=None): - # now initialize the database - argv = ["", "syncdb", "--noinput"] - execute_from_command_line(argv) - - if url is not None: - set_system_url(url) - - -def check_environment(environment=None): - if environment is not None: - set_settings_environment(environment) - try: - import settings - except ImportError: - return False - if not check_database(): - return False - return True - - -def check_database(): - """Detect if database was deleted and recreate if necessary""" +def database_exists(): + """Detect if database exists""" # can't be imported in global scope as they already require # the settings module during import from django.db import DatabaseError - from django.contrib.auth.models import User + from django.core.exceptions import ImproperlyConfigured + from openslides.participant.models import User try: User.objects.count() except DatabaseError: return False + except ImproperlyConfigured: + print "Your settings file seems broken" + sys.exit(0) else: return True -def create_or_reset_admin_user(): - # can't be imported in global scope as it already requires - # the settings module during import - from django.contrib.auth.models import User - from openslides.config.models import config - try: - obj = User.objects.get(username = "admin") - except User.DoesNotExist: - User.objects.create_superuser( - username="admin", - password="admin", - email="admin@example.com") - config['admin_password'] = "admin" - print("Created default admin user") - return - - print("Password for user admin was reset to 'admin'") - obj.set_password("admin") - obj.save() +def run_syncdb(): + # now initialize the database + argv = ["", "syncdb", "--noinput"] + execute_from_command_line(argv) def set_system_url(url): @@ -248,5 +201,45 @@ def set_system_url(url): config[key] = url +def create_or_reset_admin_user(): + # can't be imported in global scope as it already requires + # the settings module during import + from openslides.participant.models import User + try: + admin = User.objects.get(username="admin") + print("Password for user admin was reset to 'admin'") + except User.DoesNotExist: + admin = User() + admin.username = 'admin' + admin.last_name = 'Admin User' + print("Created default admin user") + + admin.is_superuser = True + admin.default_password = 'admin' + admin.set_password(admin.default_password) + admin.save() + + +def start_openslides(addr, port, start_browser_url=None, extra_args=[]): + argv = ["", "runserver", '--noreload'] + extra_args + + argv.append("%s:%d" % (addr, port)) + + if start_browser_url: + start_browser(start_browser_url) + execute_from_command_line(argv) + + +def start_browser(url): + browser = webbrowser.get() + + def f(): + time.sleep(1) + browser.open(url) + + t = threading.Thread(target=f) + t.start() + + if __name__ == "__main__": main() diff --git a/openslides/openslides_settings.py b/openslides/openslides_settings.py index fcf068c83..1da9f28fe 100755 --- a/openslides/openslides_settings.py +++ b/openslides/openslides_settings.py @@ -47,7 +47,7 @@ USE_I18N = True USE_L10N = True LOCALE_PATHS = ( - _fs2unicode(os.path.join(SITE_ROOT, 'locale')) + _fs2unicode(os.path.join(SITE_ROOT, 'locale')), ) # Absolute path to the directory that holds media. diff --git a/openslides/participant/models.py b/openslides/participant/models.py index f9c69ee8d..b97ed119c 100644 --- a/openslides/participant/models.py +++ b/openslides/participant/models.py @@ -169,7 +169,6 @@ def default_config(sender, key, **kwargs): return { 'participant_pdf_system_url': 'http://example.com:8000', 'participant_pdf_welcometext': _('Welcome to OpenSlides!'), - 'admin_password': None, }.get(key) diff --git a/openslides/participant/views.py b/openslides/participant/views.py index cb0505cb9..528e14a89 100644 --- a/openslides/participant/views.py +++ b/openslides/participant/views.py @@ -428,7 +428,7 @@ def login(request): extra_content = {} try: admin = User.objects.get(pk=1) - if admin.check_password(config['admin_password']): + if admin.check_password(admin.default_password): extra_content['first_time_message'] = _( "Installation was successfully! Use %(user)s " "(password: %(password)s) for first login.
" @@ -436,7 +436,7 @@ def login(request): "first login! Otherwise this message still appears for " "everyone and could be a security risk.") % { 'user': html_strong(admin.username), - 'password': html_strong(config['admin_password'])} + 'password': html_strong(admin.default_password)} extra_content['next'] = reverse('password_change') except User.DoesNotExist: pass diff --git a/start.py b/start.py index 03076d6eb..693a543e4 100755 --- a/start.py +++ b/start.py @@ -7,7 +7,7 @@ :license: GNU GPL, see LICENSE for more details. """ -import openslides.main +from openslides.main import main if __name__ == "__main__": - openslides.main.main() + main()