def create(self, data):
with db.auto_commit():
self.set_attrs(data)
self.active = 1
self.created_at = time_now()
self.updated_at = time_now()
db.session.add(self)
def create(self, data):
with db.auto_commit():
self.column = Column.query.filter_by(id=data['column_id']).api_first_or_404()
self.set_attrs(data)
self.active = 1
self.created_at = time_now()
self.updated_at = time_now()
db.session.add(self)
def update(self, data, edit_one_field=None):
with db.auto_commit():
if not edit_one_field:
if not data['password']:
data.pop('password')
self.set_attrs(data)
else:
self.set_attr(data)
self.updated_at = time_now()
def create(self, data):
attributes = request.form.getlist('attribute[]')
with db.auto_commit():
self.set_attrs(data)
self.active = 1
self.admin = current_user
self.attribute = ",".join(attributes)
self.created_at = time_now()
self.updated_at = time_now()
db.session.add(self)
def create(self, data):
permission_ids = request.form.getlist('permissions[]')
# 添加时把权限添加进去
with db.auto_commit():
self.set_attrs(data)
self.active = 1
self.created_at = time_now()
self.updated_at = time_now()
if permission_ids:
self.permissions = self._get_permissions(permission_ids)
db.session.add(self)
def admin_profile():
form = AdminProfileForm(request.form)
if request.method == "POST":
if form.validate():
user = current_user
with db.auto_commit():
user.set_attrs(form.data)
return Success(message="操作成功！")
return Fail(message=form.first_error)
else:
roles = Role.get_all()
return render_template('admin/admin/profile.html', roles=roles)
def create(self, data):
with db.auto_commit():
self.set_attrs(data)
self.active = 1
self.open_comment = 1
self.ip = get_client_ip()
self.user_id = current_user.id
self.username = current_user.name
self.user_type = 'admin'
self.published_at = time_now()
self.created_at = time_now()
self.updated_at = time_now()
db.session.add(self)
def create_group(name, label):
"""
设置用户分组
:param name:
:param label:
:return:
"""
with db.auto_commit():
group = Group()
group.name = name
group.label = label
db.session.add(group)
print("成功创建用户分组：%s " % (name, ))
def create_role(name, label):
"""
设置角色
:param name:
:param label:
:return:
"""
with db.auto_commit():
role = Role()
role.name = name
role.label = label
db.session.add(role)
print("成功创建管理员角色: %s " % (name, ))
def update(self, data, edit_one_field=None):
"""
edit_one_field 是否表内单个编辑 标识
:param data:
:param flag:
:return:
"""
with db.auto_commit():
if not edit_one_field:
self.set_attrs(data)
else:
self.set_attr(data)
self.updated_at = time_now()
def admin_edit_avatar():
if request.method == "POST":
form = AdminAvatarForm(request.form)
if form.validate():
user = current_user
with db.auto_commit():
user.set_attrs(form.data)
return Success(message="操作成功！")
return Fail(message=form.first_error)
else:
user = current_user
avatar = user.avatar
return render_template('admin/admin/avatar.html', avatar=avatar)
def update(self, data, edit_one_field=None):
"""
edit_one_field 是否表内单个编辑 标识
:param data:
:param flag:
:return:
"""
with db.auto_commit():
if not edit_one_field:
self.column = Column.query.filter_by(id=data['column_id']).api_first_or_404()
self.set_attrs(data)
else:
self.set_attr(data)
self.updated_at = time_now()
def update(self, data, edit_one_field=None):
"""
edit_one_field 是否表内单个编辑 标识
:param data:
:param flag:
:return:
"""
attributes = request.form.getlist('attribute[]')
with db.auto_commit():
if not edit_one_field:
self.set_attrs(data)
self.attribute = ",".join(attributes)
else:
self.set_attr(data)
self.updated_at = time_now()
def create_menu(name, endpoint):
"""
设置角色
:param name:
:param label:
:return:
"""
with db.auto_commit():
menu = Menu()
menu.name = name
menu.endpoint = endpoint
menu.pid = 0
menu.active = 1
menu.target = 1
db.session.add(menu)
print("成功创建后台导航: %s " % (name, ))
def update(self, data, edit_one_field=None):
"""
edit_one_field 是否表内单个编辑 标识
:param data:
:param flag:
:return:
"""
permission_ids = request.form.getlist('permissions[]')
# 添加时把权限添加进去
with db.auto_commit():
if not edit_one_field:
self.set_attrs(data)
self.permissions = self._get_permissions(permission_ids)
else:
self.set_attr(data)
def create_user(name, password, email):
"""
设置用户
:param name:
:param password:
:param email:
:return:
"""
with db.auto_commit():
user = User()
user.name = name
user.password = password
user.email = email
user.avatar = '/static/user/images/guest.png'
user.group = Group.query.get(1)
db.session.add(user)
print("成功添加用户,用户名:%s 密码:%s 邮箱：%s" % (name, password, email))
def create_admin(name, password, email):
"""
设置管理员
:param name:
:param password:
:param email:
:return:
"""
with db.auto_commit():
admin = Admin()
admin.name = name
admin.password = password
admin.email = email
admin.avatar = '/static/admin/images/guest.png'
admin.role = Role.query.get(1)
db.session.add(admin)
print("成功添加管理员,用户名:%s 密码:%s 邮箱：%s" % (name, password, email))
def admin_edit_password():
"""
修改密码
:return:
"""
form = AdminPasswordForm(request.form)
if request.method == "POST":
if form.validate():
user = current_user
if not user.check_password(form.oldpassword.data):
return Fail(message="原始密码错误！")
if form.oldpassword.data == form.password.data:
return Fail(message="新旧密码必须不一样！")
with db.auto_commit():
user.set_password(form.password.data)
return Success(message="操作成功！")
return Fail(message=form.first_error)
else:
return render_template('admin/admin/password.html')
def destroy(self):
with db.auto_commit():
self.is_delete = 1