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('数据表已成功创建')