def forge(category, post, comment):
"""
Generate test data
"""
from app.libs.fakers import fake_admin, fake_category, fake_comments, \
fake_posts
db.drop_all()
db.create_all()
click.echo('Generating admin...')
fake_admin()
click.echo('Generating %d categories...' % category)
fake_category(category)
click.echo('Generating %d posts...' % post)
fake_posts(post)
click.echo('Generating %s comments...' % comment)
fake_comments(comment)
click.echo('Generating links...')
fake_links()
click.echo('Done.')
def init(username, password):
"""
Init Admin account
"""
click.echo('Initializing Database...')
db.create_all()
admin = Admin.query.first()
with db.auto_commit():
if admin:
click.echo('The administrator already exists, updating...')
admin.username = username
admin.set_password(password)
else:
click.echo('Creating the temporary administrator account...')
admin = Admin(
username=username,
blog_title='临时博客名',
blog_sub_title='临时副标题',
name='临时昵称',
about='临时关于'
)
admin.set_password(password)
db.session.add(admin)
category = Category.query.first()
if category is None:
click.echo('Create the default category...')
category = Category(name='默认')
db.session.add(category)
click.echo('Done.')
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('Tables was removed successfully.')
db.create_all()
click.echo('Initialized database.')
def initdb(drop):
"""初始化数据库"""
if drop:
click.confirm('确定要删除所有数据表?', abort=True)
db.drop_all()
click.echo('数据表删除成功')
db.create_all()
# 初始化数据后在分类表中添加一条记录作为默认默认分类
with db.auto_commit():
category = Category()
category.name = "未分类"
category.show = False
db.session.add(category)
click.echo('数据表已成功创建')
def setUp(self):
app = create_app(config='testing')
self.context = app.test_request_context()
self.context.push()
self.client = app.test_client()
self.runner = app.test_cli_runner()
db.create_all()
admin = Admin(username='******',
blog_title='Test title',
blog_sub_title='Test sub title',
name='name',
about='about')
admin.set_password('12345678')
db.session.add(admin)
db.session.commit()
def initdb(drop, _init):
"""初始化数据库"""
if drop:
click.confirm('确定要删除所有数据表？', abort=True)
db.drop_all()
click.echo('数据表删除成功')
# 如果传递的是 --_init 参数，那么会直接删除数据库
if _init:
db.drop_all()
click.echo('数据表删除成功')
db.create_all()
# 初始化数据后在分类表中添加一条记录作为默认默认分类
with db.auto_commit():
category = Category()
category.alias = 'default'
db.session.add(category)
click.echo('数据表已成功创建')
