def runserver(host='localhost', port=5000):
from werkzeug.serving import run_simple
from bbs.orm import connect_db
from bbs import app
connect_db()
run_simple(host, int(port), app, use_reloader=True, use_debugger=True)
def make_shell_context():
from bbs import models
from bbs.orm import db_session, Base, connect_db
connect_db()
return dict(db_session=db_session, **Base._decl_class_registry)
def run(self, host, port):
from werkzeug.serving import run_simple
from bbs import app
from bbs.orm import connect_db
connect_db()
run_simple(host, int(port), app, use_reloader=True, use_debugger=True)
def create_user():
username = prompt('用户名').strip()
password = prompt_pass('密码').strip()
repassword = prompt_pass('重复密码').strip()
is_super = prompt_bool('是否超级用户')
is_staff = prompt_bool('是否后台用户')
if not all([username, password, repassword]):
print('请填入完整信息')
return
if password != repassword:
print('两次输入密码不一致')
return
connect_db()
user = User()
user.username = username
user.password = password
user.is_staff = is_staff
user.is_super = is_super
db_session.add(user)
db_session.commit()
print('创建用户成功')
def db_shell():
from bbs.orm import db_session, connect_db, Base
from bbs import models
connect_db()
return dict(db_session=db_session, **Base._decl_class_registry)
