def add_roles():
"""添加角色"""
if request.method == 'GET':
return render_template('addroles.html')
if request.method == 'POST':
r_name = request.form.get('r_name')
role = Role(r_name=r_name)
role.save()
return redirect(url_for('user.roles_list'))
def post(self):
role_name = request.form.get('r_name')
p_ids = request.form.get('p_id').split(',')
role = Role(r_name=role_name)
permission = Permission.query.filter(Permission.p_id.in_(p_ids)).all()
role.permission = permission
db.session.add(role)
db.session.commit()
return status_code.SUCCESS
def addroles():
if request.method == 'GET':
return render_template('addroles.html')
if request.method == 'POST':
r_name = request.form.get('r_name')
role = Role(r_name)
role.r_name = r_name
db.session.add(role)
db.session.commit()
return redirect(url_for('user.roles'))
def add_role():
if request.method == 'GET':
return render_template('addroles.html')
if request.method == 'POST':
r_name = request.form.get('r_name')
if Role.query.filter(Role.r_name == r_name).first():
return render_template('addroles.html', msg='该班级已经存在')
db.session.add(Role(r_name))
db.session.commit()
return redirect(url_for('user.show_role'))
def addroles():
if request.method == 'GET':
return render_template('addroles.html')
if request.method == 'POST':
r_name = request.form['r_name']
old_role = Role.query.filter(Role.r_name == r_name).first()
if old_role:
msg = "该角色已存在"
return render_template('addroles.html', msg=msg)
elif r_name == "":
msg = "角色名称不能为空"
return render_template('addroles.html', msg=msg)
else:
new_role = Role(r_name=r_name)
db.session.add_all([new_role])
db.session.commit()
return redirect('/role/role/')
def create_app(env):
app = Flask(__name__)
# 初始化配置
app.config.from_object(envs.get(env))
# 初始化第三方
init_ext(app)
# 初始化路由
User()
Userlog()
Tag()
Video()
Preview()
Comment()
Videocol()
Auth()
Role()
Admin()
Adminlog()
Oplog()
Videoup()
init_bili(app)
init_admin(app)
return app
def addrole(request):
role = Role()
role.role_name = 'role%s' % random.randrange(100)
role.save()
return HttpResponse('add role success')
__author__ = 'responsible'
from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy
from flask.ext.security import SQLAlchemyUserDatastore, Security
from App import config
app = Flask(__name__)
app.config.from_object(config)
db = SQLAlchemy(app)
from App.models import User, Role, roles_users
from App.routes import api
# Flask-Security
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
security = Security().init_app(app, user_datastore, register_blueprint=False)
# init database data
try:
db.create_all()
db.session.add(User("test1", "test1"))
db.session.add(User("test2", "test2"))
db.session.add(Role("admin", "管理员"))
db.session.commit()
db.engine.execute(roles_users.insert(), user_id=1, role_id=1)
db.session.commit()
except:
pass
("Pinkie Pie",
"{'name': 'db01','region': 'nyc3','size': '2GB','image':'centos 7.0','ssh_keys': None,'backups': False,'ipv6': true,'user_data':None,'private_networking': None,'volumes': None,'tags': ['web']}"
),
("Alex Tema",
"{'name': 'test1','region': 'nyc4','size': '128KB','image': 'for-the-horde-x64','ssh_keys': None,'backups': False,'ipv6':False,'user_data': None,'private_networking': None,'volumes':None,'tags': ['test']}"
)
]
default_users = [
"*****@*****.**", "*****@*****.**", "*****@*****.**",
"*****@*****.**", "*****@*****.**", "*****@*****.**",
"[email protected]"
]
# init database data
try:
db.create_all()
for user in default_users:
db.session.add(User(user, 'password'))
db.session.add(Role('admin', 'Can Create/Delete Users and Services'))
db.session.add(Role('user', 'Can Create Services'))
db.session.add(Role('automation', 'Can Delete Services'))
db.session.add(Alex('Alex', 'Only in Alpha'))
db.session.add(Alex('Am reusit intr un final', 'Yay'))
for element in default_services:
db.session.add(Service(element[0], element[1]))
db.engine.execute(roles_users.insert(), user_id=1, role_id=1)
db.session.commit()
except:
pass