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()
