def rebuild(): if config == 'development': with app.app_context(): db.drop_all() db.create_all() Role.insert_roles() admin_me = User(email=app.config['FLASKR_ADMIN'], username=app.config['MY_USERNAME'], password=app.config['MY_PASSWORD'], confirmed=True, name=forgery_py.name.full_name(), location=forgery_py.address.city(), about_me=forgery_py.lorem_ipsum.sentences(10), member_since=forgery_py.date.date(True, min_delta=10)) db.session.add(admin_me) ordinary_me = User(email=forgery_py.internet.email_address(), username='******', password='******', confirmed=True, name=forgery_py.name.full_name(), location=forgery_py.address.city(), about_me=forgery_py.lorem_ipsum.sentences(10), member_since=forgery_py.date.date(True, min_delta=10)) db.session.add(ordinary_me) db.session.commit() User.generate_fake(30) Post.generate_fake(500) Follow.generate_fake(500) Comment.generate_fake(500) else: print('Permission denied.')
def setUpClass(cls): try: cls.client = webdriver.Chrome('/usr/lib/chromium-browser/chromedriver') except: pass if cls.client: cls.app = create_app('testing') cls.app_context = cls.app.app_context() cls.app_context.push() import logging logger = logging.getLogger('werkzeug') logger.setLevel("ERROR") db.create_all() Role.insert_roles() User.generate_fake(10) Post.generate_fake(10) Comment.generate_fake(3, 5) Follow.generate_fake(3, 5) admin_role = Role.query.filter_by(name='Administrator').first() admin = User(role=admin_role, email='*****@*****.**', username='******', password='******', confirmed=True) db.session.add(admin) db.session.commit() threading.Thread(target=cls.app.run).start()
def reset(): """重置数据库""" print('开始重置数据库...') print('清空数据库...') db.drop_all() print('创建数据库...') db.create_all() print('生成角色...') Role.insert_roles() print('生成我...') u = User(wow_faction='联盟', wow_race='暗夜精灵', wow_class='德鲁伊', username='******', email='*****@*****.**', password='******', confirmed=True, wow_title='野蛮角斗士', location='试炼之环', about_me='非著名猫德') u2 = User(wow_faction='部落', wow_race='兽人', wow_class='萨满祭司', username='******', email='*****@*****.**', password='******', role=Role.query.filter_by(name='官员').first(), confirmed=True, wow_title='神出鬼没的', location='奥格瑞玛', about_me='部落精神领袖,世界萨') u3 = User(username='******', email='*****@*****.**', password='******', role=Role.query.filter_by(name='官员').first(), confirmed=True, location='冬泉谷', about_me='这封信上写的什么?') db.session.add_all((u, u2, u3)) db.session.commit() print('生成小弟...') User.generate_fake(100) u = User(wow_faction='部落', wow_race='血精灵', wow_class='法师', username='******', email='*****@*****.**', avatar='../static/wow/task/kaelthas0.png', password='******', location='奎尔萨拉斯', wow_title="迷失的", about_me="魔法,能量。我的人民陷入其中不能自拔,自从太阳之井被摧毁之后就是如此。欢迎来到未来!真遗憾,你们无法阻止什么,没有人可以阻止我了!Selama ashal'anore......") db.session.add(u) db.session.commit() User.generate_fake(100) print('生成文章...') Post.generate_fake(200) print('生成评论...') Comment.generate_fake(5, 15) print('生成关注...') Follow.generate_fake(5, 20) print('生成自关注...') User.add_self_follows() def generate_likes_and_collections(): from random import randint for i in range(1000): u = User.query.get(randint(1, User.query.count())) u2 = User.query.get(randint(1, User.query.count())) if u.username == '凯尔萨斯之魂' or u2.username == '凯尔萨斯之魂': continue c = Comment.query.get(randint(1, Comment.query.count())) p = Post.query.get(randint(1, Post.query.count())) if c not in u.comments_like: u.comments_like.append(c) c.likes += 1 db.session.add(u) if p not in u2.posts_collected: u2.posts_collected.append(p) p.collects += 1 db.session.add(u2) db.session.add(p) db.session.commit() print('生成点赞和收藏...') generate_likes_and_collections() print('重置数据库完成,谢谢使用!') quit()
def fake(): from flask_migrate import upgrade from app.models import Role, User #User.generate_fake(30) #Post.generate_fake(200) #Comment.generate_fake(3000) Follow.generate_fake(1000)
def generate_fake(): """Generate fake data""" Role.insert_roles() admin = User(email='*****@*****.**', password='******', confirmed=True, username='******') db.session.add(admin) db.session.commit() print('Inserting admin user: [email protected]') User.generate_fake() Post.generate_fake() Comment.generate_fake() Follow.generate_fake()
def init_db(): print 'drop all tables...', db.drop_all() print 'done\ncreate all tables...', db.create_all() print 'done\ninsert roles...', Role.insert_roles() print 'done\ninsert me...', generate_me() print 'done\ninsert users...', User.generate_fake() print 'done\ninsert follows...', Follow.generate_fake() print 'done\ninsert posts...', Post.generate_fake() print 'done\ninsert comments...', Comment.generate_fake() print 'done'
def db_rebuild(): """ Destroy and rebuild database with fake data. """ # destroy and rebuild tables db.reflect() db.drop_all() db.create_all() # insert roles as defined in model Role.insert_roles() # insert geos and usertypes as defined in model Geo.insert_geos() UserType.insert_user_types() # insert firm types/tiers as defined in model FirmType.insert_firm_types() FirmTier.insert_firm_tiers() # insert fake admin/test users from random import seed import forgery_py seed() test_user_1 = User(email='*****@*****.**', username='******', password='******', confirmed=True, name='Salim Hamed', location='Seattle, WA', about_me=forgery_py.lorem_ipsum.sentence(), member_since=forgery_py.date.date(True)) test_user_2 = User(email='*****@*****.**', username='******', password='******', confirmed=True, name='Bryan Davis', location='Seattle, WA', about_me=forgery_py.lorem_ipsum.sentence(), member_since=forgery_py.date.date(True)) test_user_3 = User(email='*****@*****.**', username='******', password='******', confirmed=True, name='Joe Smith', location='San Francisco, CA', about_me=forgery_py.lorem_ipsum.sentence(), member_since=forgery_py.date.date(True)) test_user_4 = User(email='*****@*****.**', username='******', password='******', confirmed=True, name='Bill Gates', location='Bellevue, WA', about_me=forgery_py.lorem_ipsum.sentence(), member_since=forgery_py.date.date(True)) admin_user = User(email='*****@*****.**', username='******', password='******', confirmed=True, name='Bill Gates', location='Seattle, WA', about_me=forgery_py.lorem_ipsum.sentence(), member_since=forgery_py.date.date(True)) db.session.add_all( [test_user_1, test_user_2, test_user_3, test_user_4, admin_user]) db.session.commit() # insert fake user data User.generate_fake(60) # insert fake post data Post.generate_fake(400) # insert fake followers Follow.generate_fake(2000) # insert fake firms Firm.generate_fake(5000) # insert fake companies Company.generate_fake(10000) # insert fake relationships Relationship.generate_fake(60000) # print results inspector = db.inspect(db.engine) print('The following tables were created.') print('-' * 17) for table in inspector.get_table_names(): print(table)
def reset(): """重置数据库""" print('开始重置数据库...') print('清空数据库...') db.drop_all() print('创建数据库...') db.create_all() print('生成角色...') Role.insert_roles() print('生成我...') u = User(wow_faction='联盟', wow_race='暗夜精灵', wow_class='德鲁伊', username='******', email='*****@*****.**', password='******', confirmed=True, name='野蛮角斗士', location='试炼之环', about_me='非著名猫德') u2 = User(wow_faction='部落', wow_race='兽人', wow_class='萨满祭司', username='******', email='*****@*****.**', password='******', role=Role.query.filter_by(name='官员').first(), confirmed=True, name='神出鬼没的', location='奥格瑞玛', about_me='部落精神领袖,世界萨') u3 = User(username='******', email='*****@*****.**', password='******', role=Role.query.filter_by(name='官员').first(), confirmed=True, name='游荡的', location='冬泉谷', about_me='这封信上写的什么?') db.session.add_all((u, u2, u3)) db.session.commit() print('生成小弟...') User.generate_fake(200) print('生成文章...') Post.generate_fake(200) print('生成评论...') Comment.generate_fake(5, 15) print('生成关注...') Follow.generate_fake(5, 20) print('生成自关注...') User.add_self_follows() def generate_likes_and_collections(): from random import randint for i in range(1000): u = User.query.get(randint(1, User.query.count())) u2 = User.query.get(randint(1, User.query.count())) c = Comment.query.get(randint(1, Comment.query.count())) p = Post.query.get(randint(1, Post.query.count())) if c not in u.comments_like: u.comments_like.append(c) c.likes += 1 db.session.add(u) if p not in u2.posts_collected: u2.posts_collected.append(p) p.collects += 1 db.session.add(u2) db.session.add(p) db.session.commit() print('生成点赞和收藏...') generate_likes_and_collections() print('重置数据库完成,谢谢使用!') quit()
def db_rebuild(): """ Destroy and rebuild database with fake data. """ # destroy and rebuild tables db.reflect() db.drop_all() db.create_all() # insert roles as defined in model Role.insert_roles() # insert geos and usertypes as defined in model Geo.insert_geos() UserType.insert_user_types() # insert firm types/tiers as defined in model FirmType.insert_firm_types() FirmTier.insert_firm_tiers() # insert fake admin/test users from random import seed import forgery_py seed() test_user_1 = User( email='*****@*****.**', username='******', password='******', confirmed=True, name='Salim Hamed', location='Seattle, WA', about_me=forgery_py.lorem_ipsum.sentence(), member_since=forgery_py.date.date(True) ) test_user_2 = User( email='*****@*****.**', username='******', password='******', confirmed=True, name='Bryan Davis', location='Seattle, WA', about_me=forgery_py.lorem_ipsum.sentence(), member_since=forgery_py.date.date(True) ) test_user_3 = User( email='*****@*****.**', username='******', password='******', confirmed=True, name='Joe Smith', location='San Francisco, CA', about_me=forgery_py.lorem_ipsum.sentence(), member_since=forgery_py.date.date(True) ) test_user_4 = User( email='*****@*****.**', username='******', password='******', confirmed=True, name='Bill Gates', location='Bellevue, WA', about_me=forgery_py.lorem_ipsum.sentence(), member_since=forgery_py.date.date(True) ) admin_user = User( email='*****@*****.**', username='******', password='******', confirmed=True, name='Bill Gates', location='Seattle, WA', about_me=forgery_py.lorem_ipsum.sentence(), member_since=forgery_py.date.date(True) ) db.session.add_all([test_user_1, test_user_2, test_user_3, test_user_4, admin_user]) db.session.commit() # insert fake user data User.generate_fake(60) # insert fake post data Post.generate_fake(400) # insert fake followers Follow.generate_fake(2000) # insert fake firms Firm.generate_fake(5000) # insert fake companies Company.generate_fake(10000) # insert fake relationships Relationship.generate_fake(60000) # print results inspector = db.inspect(db.engine) print 'The following tables were created.' print '-'*17 for table in inspector.get_table_names(): print table