def setUp(self):
db.app = test_app
db.create_all()
user = User()
user.username = self.username
user.set_password(self.password)
db.session.add(user)
comment = Comment()
comment.name = self.comment_name
comment.text = self.comment_text
tag = Tag()
tag.title = self.tag_title
post = Post()
post.title = self.post_title
post.text = self.post_text
post.publish_date = self.post_publish_date
# add relationships to other tables
post.user = user
post.tags = [tag]
post.comments = [comment]
db.session.add(user)
db.session.add(comment)
db.session.add(tag)
db.session.add(post)
db.session.commit()
def populate_default_data(db, app):
db.app = app
db.create_all()
user = User()
user.username = "******"
user.set_password("jim")
db.session.add(user)
db.session.commit()
tag_one = Tag(title="Python")
tag_two = Tag(title="Flask")
tag_three = Tag(title="SQLAlchemy")
tag_four = Tag(title="Jinja")
tag_list = [tag_one, tag_two, tag_three, tag_four]
s = "Example text"
for i in range(100):
new_post = Post(title="Post " + str(i))
new_post.user = user
new_post.publish_date = datetime.datetime.now()
new_post.text = s
new_post.tags = random.sample(tag_list, random.randint(1, 3))
db.session.add(new_post)
db.session.commit()
def setup_db():
db.create_all()
admin_role = Role()
admin_role.name = "admin"
admin_role.description = "admin"
db.session.add(admin_role)
default_role = Role()
default_role.name = "default"
default_role.description = "default"
db.session.add(default_role)
admin = User()
admin.username = "******"
admin.set_password("password")
admin.roles.append(admin_role)
admin.roles.append(default_role)
db.session.add(admin)
tag_one = Tag('Python')
tag_two = Tag('Flask')
tag_three = Tag('SQLAlechemy')
tag_four = Tag('Jinja')
tag_list = [tag_one, tag_two, tag_three, tag_four]
s = "Body text"
for i in xrange(100):
new_post = Post("Post " + str(i))
new_post.user = admin
new_post.publish_date = datetime.datetime.now()
new_post.text = s
new_post.tags = random.sample(tag_list, random.randint(1, 3))
db.session.add(new_post)
db.session.commit()
def setup_db():
db.create_all()
admin_role = Role()
admin_role.name = "admin"
admin_role.description = "admin"
db.session.add(admin_role)
default_role = Role()
default_role.name = "default"
default_role.description = "default"
db.session.add(default_role)
admin = User()
admin.username = "******"
admin.set_password("password")
admin.roles.append(admin_role)
admin.roles.append(default_role)
db.session.add(admin)
tag_one = Tag('Python')
tag_two = Tag('Flask')
tag_three = Tag('SQLAlechemy')
tag_four = Tag('Jinja')
tag_list = [tag_one, tag_two, tag_three, tag_four]
s = "Body text"
for i in xrange(100):
new_post = Post("Post " + str(i))
new_post.user = admin
new_post.publish_date = datetime.datetime.now()
new_post.text = s
new_post.tags = random.sample(tag_list, random.randint(1, 3))
db.session.add(new_post)
db.session.commit()
def setup_db():
db.create_all()
# 创建管理员角色
admin_role = Role('admin')
admin_role.description = 'admin'
db.session.add(admin_role)
# 创建默认角色
default_role = Role('default')
default_role.description = 'default'
db.session.add(default_role)
# 创建管理员用户信息
admin = User('admin')
admin.set_password('password')
admin.roles.append(admin_role)
admin.roles.append(default_role)
db.session.add(admin)
tag_one = Tag('Python')
tag_two = Tag('Flask')
tag_three = Tag('SQLAlchemy')
tag_four = Tag('Jinja')
tag_list = [tag_one, tag_two, tag_three, tag_four]
s = 'Body text'
for i in range(100):
new_post = Post('Post %s' % i, s)
new_post.user = admin
new_post.publish_date = datetime.datetime.now()
new_post.tags = random.sample(tag_list, random.randint(1, 3))
db.session.add(new_post)
db.session.commit()
def insert_data():
with app.app_context():
# 不需要在这里创建库，应该使用数据库升级命令`db upgrade`来创建库
# db.create_all()
# 这里设定了3种角色
role_admin = Role(name='admin')
role_admin.description = "administrator role"
role_poster = Role(name='poster')
role_poster.description = "the registered user role"
role_default = Role(name='default')
role_default.description = 'the unregistered user role'
db.session.add(role_admin)
db.session.add(role_poster)
db.session.add(role_default)
# add User
admin = User(username='******')
admin.email = '*****@*****.**'
admin.password = '******'
admin.confirmed = True
admin.roles.append(role_admin)
admin.roles.append(role_poster)
admin.roles.append(role_default)
db.session.add(admin)
user01 = User(username='******')
user01.email = '*****@*****.**'
user01.password = '******'
user01.confirmed = True
user01.roles.append(role_poster)
user01.roles.append(role_default)
db.session.add(user01)
user02 = User(username='******')
user02.email = '*****@*****.**'
user02.password = '******'
user02.confirmed = True
user02.roles.append(role_poster)
user02.roles.append(role_default)
db.session.add(user02)
# add Tag and Post
tag_one = Tag('Python')
tag_two = Tag('Flask')
tag_three = Tag('SQLAlechemy')
tag_four = Tag('Jinja')
tag_list = [tag_one, tag_two, tag_three, tag_four]
s = "Example Text"
for i in xrange(1, 101):
new_post = Post("Post {}".format(i))
if i % 2:
new_post.user = user01
else:
new_post.user = user02
new_post.publish_date = datetime.datetime.utcnow()
new_post.text = s
new_post.tags = random.sample(tag_list, random.randint(1, 3))
db.session.add(new_post)
# add comment
comment01 = Comment()
comment01.name = 'comment01'
comment01.text = 'comment text'
comment01.post_id = 99
comment01.date = datetime.datetime.utcnow()
db.session.add(comment01)
comment02 = Comment()
comment02.name = 'comment02'
comment02.text = 'comment text'
comment02.post_id = 100
comment02.date = datetime.datetime.utcnow()
db.session.add(comment02)
db.session.commit()
import random
import datetime
from webapp.models import db, User, Post, tags, Tag, Comment
user = User.query.get(1)
tag_one = Tag('Python')
tag_two = Tag('Flask')
tag_three = Tag('SQLAlechemy')
tag_four = Tag('Jinja')
tag_list = [tag_one, tag_two, tag_three, tag_four]
s = "Example text"
for i in range(100):
new_post = Post("Post " + str(i))
new_post.user = user
new_post.publish_date = datetime.datetime.now()
new_post.text = s
new_post.tags = random.sample(tag_list, random.randint(1, 3))
db.session.add(new_post)
db.session.commit()
