#!/usr/bin/env python # -*- coding: utf-8 -*- import argparse import os import shutil import sys from django.conf import ENVIRONMENT_VARIABLE from django.core.management import execute_from_command_line from openslides import get_version from openslides.utils.main import ( detect_openslides_type, ensure_settings, filesystem2unicode, get_browser_url, get_database_path_from_settings, get_default_settings_path, get_default_user_data_path, get_port, setup_django_settings_module, start_browser, write_settings) def main(): """ Main entrance to OpenSlides. """ # Parse all command line args. args = parse_args() # Setup settings path: Take it either from command line or get default path if hasattr(args, 'settings') and args.settings: settings = args.settings setup_django_settings_module(settings) else: if ENVIRONMENT_VARIABLE not in os.environ: openslides_type = detect_openslides_type() settings = get_default_settings_path(openslides_type) setup_django_settings_module(settings) else: # The environment variable is set, so we do not need to process # anything more here. settings = None # Process the subcommand's callback return args.callback(settings, args) def parse_args(): """ Parses all command line arguments. The subcommand 'django' links to Django's command-line utility. """ if len(sys.argv) == 1: sys.argv.append('start') # Init parser description = 'Start script for OpenSlides.' if 'manage.py' not in sys.argv[0]: description += (' If it is called without any argument, this will be ' 'treated as if it is called with the "start" subcommand. ' 'That means OpenSlides will setup default settings and ' 'database, start the tornado webserver, launch the ' 'default web browser and open the webinterface.') parser = argparse.ArgumentParser(description=description) # Add version argument parser.add_argument( '--version', action='version', version=get_version(), help='Show version number and exit.') # Init subparsers subparsers = parser.add_subparsers( dest='subcommand', title='Available subcommands', description="Type '%s --help' for help on a " "specific subcommand." % parser.prog, help='You can choose only one subcommand at once.') # Subcommand start subcommand_start = subparsers.add_parser( 'start', help='Setup settings and database, start tornado webserver, launch the ' 'default web browser and open the webinterface.') add_general_arguments(subcommand_start, ('settings', 'user_data_path', 'address', 'port')) subcommand_start.add_argument( '--no-browser', action='store_true', help='Do not launch the default web browser.') subcommand_start.set_defaults(callback=start) # Subcommand runserver subcommand_runserver = subparsers.add_parser( 'runserver', help='Run OpenSlides using tornado webserver.') add_general_arguments(subcommand_runserver, ('settings', 'user_data_path', 'address', 'port')) subcommand_runserver.add_argument( '--start-browser', action='store_true', help='Launch the default web browser and open the webinterface.') subcommand_runserver.add_argument( '--no-reload', action='store_true', help='Do not reload the webserver if source code changes.') subcommand_runserver.set_defaults(callback=runserver) # Subcommand syncdb subcommand_syncdb = subparsers.add_parser( 'syncdb', help='Create or update database tables.') add_general_arguments(subcommand_syncdb, ('settings', 'user_data_path')) subcommand_syncdb.set_defaults(callback=syncdb) # Subcommand createsuperuser subcommand_createsuperuser = subparsers.add_parser( 'createsuperuser', help="Make sure the user 'admin' exists and uses 'admin' as password.") add_general_arguments(subcommand_createsuperuser, ('settings', 'user_data_path')) subcommand_createsuperuser.set_defaults(callback=createsuperuser) # Subcommand backupdb subcommand_backupdb = subparsers.add_parser( 'backupdb', help='Store a backup copy of the SQLite3 database at the given path.') add_general_arguments(subcommand_backupdb, ('settings', 'user_data_path')) subcommand_backupdb.add_argument( 'path', help='Path to the backup file. An existing file will be overwritten.') subcommand_backupdb.set_defaults(callback=backupdb) # Subcommand deletedb subcommand_deletedb = subparsers.add_parser( 'deletedb', help='Delete the SQLite3 database.') add_general_arguments(subcommand_deletedb, ('settings', 'user_data_path')) subcommand_deletedb.set_defaults(callback=deletedb) # Subcommand create-dev-settings subcommand_create_dev_settings = subparsers.add_parser( 'create-dev-settings', help='Create a settings file at current working directory for development use.') subcommand_create_dev_settings.set_defaults(callback=create_dev_settings) # Subcommand django subcommand_django_command_line_utility = subparsers.add_parser( 'django', description="Link to Django's command-line utility. Type " "'%s django help' for more help on this." % parser.prog, help="Call Django's command-line utility.") subcommand_django_command_line_utility.set_defaults( callback=django_command_line_utility, django_args=['%s' % subcommand_django_command_line_utility.prog]) known_args, unknown_args = parser.parse_known_args() if known_args.subcommand == 'django': if not unknown_args: unknown_args.append('help') known_args.django_args.extend(unknown_args) else: if unknown_args: parser.error('Unknown arguments %s found.' % ' '.join(unknown_args)) return known_args def add_general_arguments(subcommand, arguments): """ Adds the named arguments to the subcommand. """ general_arguments = {} openslides_type = detect_openslides_type() general_arguments['settings'] = ( ('-s', '--settings'), dict(help="Path to settings file. If this isn't provided, the " "%s environment variable will be used. " "If this isn't provided too, a default path according to the " "OpenSlides type will be used. At the moment it is %s" % ( ENVIRONMENT_VARIABLE, get_default_settings_path(openslides_type)))) general_arguments['user_data_path'] = ( ('-d', '--user-data-path'), dict(help='Path to the directory for user specific data files like SQLite3 ' 'database, uploaded media and search index. This is only used, ' 'when a new settings file is created. The given path is only ' 'written into the new settings file. Default according to the ' 'OpenSlides is at the moment %s' % get_default_user_data_path(openslides_type))) general_arguments['address'] = ( ('-a', '--address',), dict(default='0.0.0.0', help='IP address to listen on. Default is %(default)s.')) general_arguments['port'] = ( ('-p', '--port'), dict(type=int, default=80, help='Port to listen on. Default as admin or root is %(default)d, else 8000.')) for argument in arguments: try: args, kwargs = general_arguments[argument] except KeyError: raise TypeError('The argument %s is not a valid general argument.' % argument) subcommand.add_argument(*args, **kwargs) def start(settings, args): """ Starts OpenSlides: Runs syncdb and runs runserver (tornado webserver). """ ensure_settings(settings, args) syncdb(settings, args) args.start_browser = not args.no_browser args.no_reload = False runserver(settings, args) def runserver(settings, args): """ Runs tornado webserver. Runs the function start_browser if the respective argument is given. """ ensure_settings(settings, args) port = get_port(address=args.address, port=args.port) if args.start_browser: browser_url = get_browser_url(address=args.address, port=port) start_browser(browser_url) # Now the settings is available and the function can be imported. from openslides.utils.tornado_webserver import run_tornado run_tornado(args.address, port, not args.no_reload) def syncdb(settings, args): """ Run syncdb to create or update the database. """ ensure_settings(settings, args) # TODO: Check use of filesystem2unicode here. db_file = get_database_path_from_settings() if db_file is not None: db_dir = filesystem2unicode(os.path.dirname(db_file)) if not os.path.exists(db_dir): os.makedirs(db_dir) if not os.path.exists(db_file): print('Clearing old search index...') execute_from_command_line(["", "clear_index", "--noinput"]) execute_from_command_line(["", "syncdb", "--noinput"]) return 0 def createsuperuser(settings, args): """ Creates or resets the admin user. Returns 0 to show success. """ ensure_settings(settings, args) # can't be imported in global scope as it already requires # the settings module during import from openslides.participant.api import create_or_reset_admin_user if create_or_reset_admin_user(): print('Admin user successfully created.') else: print('Admin user successfully reset.') return 0 def backupdb(settings, args): """ Stores a backup copy of the SQlite3 database. Returns 0 on success, else 1. """ ensure_settings(settings, args) from django.db import connection, transaction @transaction.commit_manually def do_backup(src_path, dest_path): # perform a simple file-copy backup of the database # first we need a shared lock on the database, issuing a select() # will do this for us cursor = connection.cursor() cursor.execute("SELECT count(*) from sqlite_master") # now copy the file try: shutil.copy(src_path, dest_path) except IOError: raise Exception("Database backup failed.") # and release the lock again transaction.commit() database_path = get_database_path_from_settings() if database_path: do_backup(database_path, args.path) print('Database %s successfully stored at %s.' % (database_path, args.path)) return_value = 0 else: print('Error: Default database is not SQLite3. Only SQLite3 databases ' 'can currently be backuped.') return_value = 1 return return_value def deletedb(settings, args): """ Deletes the sqlite3 database. Returns 0 on success, else 1. """ ensure_settings(settings, args) database_path = get_database_path_from_settings() if database_path and os.path.exists(database_path): os.remove(database_path) print('SQLite3 database file %s successfully deleted.' % database_path) execute_from_command_line(["", "clear_index", "--noinput"]) print('Whoosh search index successfully cleared.') return_value = 0 else: print('SQLite3 database file %s does not exist.' % database_path) return_value = 1 return return_value def create_dev_settings(settings, args): """ Creates a settings file at the currect working directory for development use. """ settings = os.path.join(os.getcwd(), 'settings.py') if not os.path.exists(settings): context = {} context['openslides_user_data_path'] = repr(os.getcwd()) context['import_function'] = '' context['debug'] = 'True' write_settings(settings, **context) print('Settings file at %s successfully created.' % settings) return_value = 0 else: print('Error: Settings file %s already exists.' % settings) return_value = 1 return return_value def django_command_line_utility(settings, args): """ Runs Django's command line utility. Returns 0 on success, else 1. """ if 'runserver' in args.django_args: command = 'runserver' elif 'syncdb' in args.django_args: command = 'syncdb' elif 'createsuperuser' in args.django_args: command = 'createsuperuser' else: command = None if command: print("Error: The command '%s' is disabled in OpenSlides for use via Django's " "command line utility." % command) return_value = 1 else: ensure_settings(settings, args) execute_from_command_line(args.django_args) return_value = 0 return return_value if __name__ == "__main__": exit(main())