def deploy(): "Run deployment tasks" from flask.ext.migrate import upgrade from models import User,Role,Post #数据库迁移到最新修订版本 # upgrade() #创建用户角色 Role.insert_roles() #创建管理员账号 User.add_admin()
def deploy():
"Run deployment tasks"
from flask.ext.migrate import upgrade
from models import User, Role, Post
#数据库迁移到最新修订版本
# upgrade()
#创建用户角色
Role.insert_roles()
#创建管理员账号
User.add_admin()
def deploy(): """Run deployment tasks.""" from app import create_app,bcrypt app = create_app() app.app_context().push() from models import Role,User from app import db from flask_migrate import upgrade,migrate,init,stamp db.create_all() # create user roles Role.insert_roles() # migrate database to latest revision stamp() migrate() upgrade()
app.jinja_env.globals['momentjs'] = momentjs
from app import views, models
# make sure admin account is setup
from models import User, Role, Permission
"""
@app.context_processor
def inject_permissions():
return dict(Permission=Permission)
"""
# will throw an error if database or the table doesnt exist yet
try:
# initialize roles
Role.insert_roles()
# add admin if not yet added
admin_email = ADMINS[0]
# this returns result object
admin_user = User.query.filter_by(email=admin_email).first()
# this returns object values
admin_role = db.session.query(Role).filter(Role.permissions==Permission.ADMINISTER).first()
if not admin_user:
defaultpassword = bcrypt.generate_password_hash('1234')
user = User(username='******', email=admin_email, role=admin_role, password = defaultpassword)
db.session.add(user)
db.session.commit()
# follow yourself
def test_roles_and_permissions(self):
Role.insert_roles()
u = User(email='*****@*****.**', password='******')
self.assertTrue(u.can(Permission.WRITE))
self.assertFalse(u.can(Permission.MODERATE_COMMENTS))
def setup_class(cls):
cls.app = create_app('testing')
cls.app_context = cls.app.app_context()
cls.app_context.push()
Role.insert_roles()
def test_insert_roles(self):
Role.insert_roles()
