def fake_categories(cls, count=9):
"""
生成Blog分类虚拟数据
"""
while Category.query.count() < count + 1:
category_name = FakeData.FAKER.word()
# 分类不能同名，为了避免这个情况，这里要做一定的判断
if Category.query.filter_by(name=category_name).first():
continue
with db.auto_commit():
category = Category()
category.name = category_name
# 添加这一行代码即可生成中文分类的拼音
category.alias = ''.join(lazy_pinyin(category_name))
db.session.add(category)
def fake_categories(cls, count: int = 9):
"""
生成博客分类虚拟数据
:param count: default=9，包括我们初始数据库生成的默认分类，默认一共有 10 个分类
:return: None
"""
while Category.query.count() < count + 1:
category_name = FakeData.FAKER.word()
# 分类不能同名，为了避免这个情况，这里要做一定的判断
if Category.query.filter_by(name=category_name).first():
continue
with db.auto_commit():
category = Category()
category.name = category_name
category.alias = ''.join(lazy_pinyin(category_name))
db.session.add(category)
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('数据表已成功创建')
def initdb(drop, _init):
"""初始化数据库"""
if drop:
click.confirm('确定要删除所有数据表？', abort=True)
db.drop_all()
click.echo('数据表删除成功')
if _init:
db.drop_all()
click.echo('数据表删除成功')
db.create_all()
# 初始化数据后在分类表中添加一条记录作为默认默认分类
with db.auto_commit():
category = Category()
category.name = '未分类'
category.alias = 'default'
category.show = False
db.session.add(category)
click.echo('数据表已成功创建')
