from gevent import wsgi from realms import config, app, cli, db from realms.lib.util import random_string import click @cli.command() @click.option('--site-title', default=config.SITE_TITLE, prompt='Enter site title.') @click.option('--base_url', default=config.BASE_URL, prompt='Enter base URL.') @click.option('--port', default=config.PORT, prompt='Enter port number.') @click.option('--secret-key', default=config.SECRET_KEY if config.SECRET_KEY != "CHANGE_ME" else random_string(64), prompt='Enter secret key.') @click.option('--wiki-path', default=config.WIKI_PATH, prompt='Where do you want to store wiki data?', help='Wiki Directory (git repo)') @click.option('--allow-anon', default=config.ALLOW_ANON, is_flag=True, prompt='Allow anonymous edits?') @click.option('--registration-enabled', default=config.REGISTRATION_ENABLED, is_flag=True, prompt='Enable registration?') @click.option('--cache-type', default=config.CACHE_TYPE, type=click.Choice([None, 'simple', 'redis', 'memcached']), prompt='Cache type?') @click.option('--db-uri', default=config.DB_URI, prompt='Database URI, Examples: http://goo.gl/RyW0cl') @click.pass_context def setup(ctx, **kw): """ Start setup wizard """ conf = {} for k, v in kw.items(): conf[k.upper()] = v config.update(conf) if conf['CACHE_TYPE'] == 'redis': ctx.invoke(setup_redis) elif conf['CACHE_TYPE'] == 'memcached': ctx.invoke(setup_memcached) click.secho('Config saved to %s/config.json' % config.APP_PATH, fg='green') click.secho('Type "realms-wiki run" to start server', fg='yellow') @click.command() @click.option('--cache-redis-host', default=getattr(config, 'CACHE_REDIS_HOST', "127.0.0.1"), prompt='Redis host') @click.option('--cache-redis-port', default=getattr(config, 'CACHE_REDIS_POST', 6379), prompt='Redis port') @click.option('--cache-redis-password', default=getattr(config, 'CACHE_REDIS_PASSWORD', None), prompt='Redis password') @click.option('--cache-redis-db', default=getattr(config, 'CACHE_REDIS_DB', 0), prompt='Redis db') def setup_redis(**kw): conf = {} for k, v in kw.items(): conf[k.upper()] = v config.update(conf) @click.command() @click.option('--cache-memcached-servers', default=getattr(config, 'CACHE_MEMCACHED_SERVERS', ["127.0.0.1:11211"]), type=click.STRING, prompt='Memcached servers, separate with a space') def setup_memcached(**kw): conf = {} for k, v in kw.items(): conf[k.upper()] = v config.update(conf) @cli.command() @click.option('--port', default=5000) def dev(port): """ Run development server """ click.secho("Starting development server", fg='green') app.run(host="0.0.0.0", port=port, debug=True) @cli.command() def run(): """ Run production server """ click.secho("Server started. Env: %s Port: %s" % (config.ENV, config.PORT), fg='green') wsgi.WSGIServer(('', int(config.PORT)), app).serve_forever() @cli.command() def create_db(): """ Creates DB tables """ click.echo("Creating all tables") db.create_all() @cli.command() @click.confirmation_option(help='Are you sure you want to drop the db?') def drop_db(): """ Drops DB tables """ click.echo("Dropping all tables") db.drop_all() if __name__ == '__main__': cli()