replace flask.ext.script with custom command interface

This commit is contained in:
j 2014-08-09 19:06:53 +02:00
parent fc4ccb8083
commit 7bd6181e06
3 changed files with 197 additions and 210 deletions

View file

@ -6,10 +6,24 @@ from __future__ import division
import os
import sys
import app
import api
import commands
import server
if len(sys.argv) > 1 and sys.argv[1] == 'server':
server.run()
else:
app.run()
names = [c[8:] for c in dir(commands) if c.startswith('command_')]
command = sys.argv[1] if len(sys.argv) > 1 else None
if command and command in names:
getattr(commands, "command_%s"%command)(sys.argv[1:])
else:
print 'usage: %s [action]' % 'ctl'
indent = max([len(command) for command in names]) + 4
for command in sorted(names):
space = ' ' * (indent - len(command))
info = getattr(commands, "command_%s"%command).__doc__.split('\n')
info = [' %s%s' % (' ' * indent, i.strip()) for i in info]
info = '\n'.join(info).strip()
print " %s%s%s"%(command, space, info)
sys.exit(1)

View file

@ -1,50 +0,0 @@
# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
from __future__ import division
import logging
import settings
import db
import changelog
import item.models
import user.models
import item.person
import api
import commands
'''
app = Flask('openmedialibrary', static_folder=settings.static_path)
manager = Manager(app, with_default_commands=False)
manager.add_command('release', commands.Release)
manager.add_command('debug', commands.Debug)
manager.add_command('update', commands.Update)
manager.add_command('install_update', commands.InstallUpdate)
manager.add_command('start', commands.Start)
manager.add_command('stop', commands.Stop)
manager.add_command('setup', commands.Setup)
manager.add_command('version', commands.Version)
manager.add_command('postupdate', commands.PostUpdate)
manager.add_command('shell', Shell)
manager.add_command('update_static', commands.UpdateStatic)
@app.route('/')
@app.route('/<path:path>')
def main(path=None):
return app.send_static_file('html/oml.html')
'''
def run():
import sys
command = sys.argv[1] if len(sys.argv) > 1 else None
if command and getattr(commands, "command_%s"%command):
getattr(commands, "command_%s"%command)(sys.argv[1:])
else:
print 'usage: ... fixme'
sys.exit(1)

View file

@ -7,8 +7,6 @@ from os.path import join, exists, dirname
import os
import sys
from flask.ext.script import Command, Option
import settings
root_dir = dirname(settings.base_dir)
@ -38,64 +36,56 @@ def version(module):
version = settings.release['modules'][module]['version']
return version
class Version(Command):
def command_version(*args):
"""
Print current version
"""
def run(self):
print version('openmedialibrary')
class Debug(Command):
def command_debug(*args):
"""
Start in debug mode
"""
def run(self):
pass
class Start(Command):
def command_start(*args):
"""
Start Open Media Libary
"""
def run(self):
pass
class Stop(Command):
def command_stop(*args):
"""
Stop Open Media Libary
"""
def run(self):
pass
class InstallUpdate(Command):
def command_install_update(*args):
"""
Update to latest development version
Install available updates
"""
def run(self):
import update
if not update.install():
print "UPDATE FAILED"
sys.exit(1)
class Update(Command):
def command_update(*args):
"""
Update to latest development version
"""
def run(self):
import update
if not (update.download() and update.install()):
print "UPDATE FAILED"
class PostUpdate(Command):
def command_postupdate(*args):
"""
Called by update to fix stuff
Called after update with -o old -n new
"""
def get_options(self):
return [
Option('-o', '--old', dest='old'),
Option('-n', '--new', dest='new'),
]
def run(self, old, new):
def run(*args):
o, old, n, new = args
if o != '-o' or n != '-n':
print 'usage: -o oldversion -n newversion'
sys.exit(1)
if old <= '20140521-65-e14c686' and new > '20140521-65-e14c686':
if not os.path.exists(settings.db_path):
r('./ctl', 'setup')
@ -117,20 +107,18 @@ class PostUpdate(Command):
import item.models
item.models.Find.query.filter_by(key='list').delete()
class Setup(Command):
def command_setup(*args):
"""
Setup new node
"""
def run(self):
import setup
setup.create_db()
setup.create_default_lists()
class UpdateStatic(Command):
def command_update_static(*args):
"""
Update static files
"""
def run(self):
oxjs = os.path.join(settings.static_path, 'oxjs')
if not os.path.exists(oxjs):
r('git', 'clone', 'https://git.0x2620.org/oxjs.git', oxjs)
@ -139,11 +127,10 @@ class UpdateStatic(Command):
r('python2', os.path.join(oxjs, 'tools', 'build', 'build.py'))
r('python2', os.path.join(settings.static_path, 'py', 'build.py'))
class Release(Command):
def command_release(*args):
"""
Release new version
"""
def run(self):
print 'checking...'
import os
import json
@ -199,3 +186,39 @@ class Release(Command):
with open('updates/release.json', 'w') as fd:
json.dump(release, fd, indent=2)
print 'signed latest release in updates/release.json'
def command_shell(*args):
'''
Runs a Python shell inside the application context.
'''
context = None
banner = 'Open Media Library'
import db
with db.session():
# Try BPython
try:
from bpython import embed
embed(banner=banner, locals_=context)
return
except ImportError:
pass
# Try IPython
try:
try:
# 0.10.x
from IPython.Shell import IPShellEmbed
ipshell = IPShellEmbed(banner=banner)
ipshell(global_ns=dict(), local_ns=context)
except ImportError:
# 0.12+
from IPython import embed
embed(banner1=banner, user_ns=context)
return
except ImportError:
pass
import code
# Use basic python shell
code.interact(banner, local=context)