def admin(): try: db.drop_all() db.create_all() Role.init_role() States.init_states() ThirdParty.init_tp() username = input('请输入超级管理员用户名:') email = input('请输入超级管理员邮箱:') pwd = input('请输入超级管理员密码:') confirm = input('请确认密码:') if pwd != confirm: click.echo('两次密码不一致') click.echo('退出当前操作') return super_user = User(username=username, email=email, password=pwd, confirm=1, avatar='/static/img/admin/admin.jpg') super_user.set_password(pwd) db.session.add(super_user) db.session.commit() click.echo('超级管理员创建成功!') click.echo('应用初始化成功!') click.echo('程序退出...') except: import traceback traceback.print_exc() db.session.rollback() click.echo('操作出现异常,退出...')
def setUp(self): app = create_app('testing') self.context = app.test_request_context() self.context.push() self.client = app.test_client() self.runner = app.test_cli_runner() # 数据库初始化操作 db.drop_all() db.create_all() Role.init_role() States.init_states() ThirdParty.init_tp() bt = BlogType(name='Test', description='test type') blog1 = Blog(title='test1', content='Blog test1 contents.', introduce='blog1 introduce', type_id=1, pre_img='test.jpg', delete_flag=1) blog2 = Blog(title='test2', content='Blog test2 contents.', introduce='blog2 introduce', type_id=1, pre_img='test.jpg', delete_flag=1) blog3 = Blog(title='test3', content='Blog test3 contents.', introduce='blog3 introduce', type_id=1, pre_img='test.jpg', delete_flag=1) ph1 = Photo(title='test photo1', description='test photo1 description', save_path='/test/save/', save_path_s='/test/save/', level=1) super_user = User(username='******', email='*****@*****.**', password='******', confirm=1, avatar='/static/img/admin/admin.jpg') super_user.set_password('12345678') user = User(username='******', email='*****@*****.**', password='******', confirm=1) user.set_password('12345678') unconfirm_user = User(username='******', email='*****@*****.**', password='******', confirm=1) unconfirm_user.set_password('12345678') db.session.add_all( [super_user, user, unconfirm_user, bt, blog1, blog2, blog3, ph1]) db.session.commit()
def initdb(drop): """Initialize the database.""" if drop: click.confirm('This operation will delete the database, do you want to continue?', abort=True) db.drop_all() click.echo('Drop tables.') db.create_all() click.echo('Initialized database.')
def init(): """Initialized Blogin""" click.echo('Initializing the database...') db.drop_all() db.create_all() click.echo('Initializing the roles and permissions...') Role.init_role() click.echo('Done.')
def tearDown(self): # 结束清除数据,结束上下文环境 db.drop_all() self.context.pop()