mirror of https://github.com/veops/cmdb.git
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
# -*- coding:utf-8 -*-
|
|
|
|
|
|
from flask import abort
|
|
|
|
from api.extensions import db
|
|
from api.lib.perm.acl.cache import UserCache
|
|
from api.models.acl import User
|
|
|
|
|
|
class UserCRUD(object):
|
|
@staticmethod
|
|
def search(q, page=1, page_size=None):
|
|
query = db.session.query(User).filter(User.deleted.is_(False))
|
|
if q:
|
|
query = query.filter(User.username.ilike('%{0}%'.format(q)))
|
|
|
|
numfound = query.count()
|
|
|
|
return numfound, query.offset((page - 1) * page_size).limit(page_size)
|
|
|
|
@staticmethod
|
|
def add(**kwargs):
|
|
existed = User.get_by(username=kwargs['username'], email=kwargs['email'])
|
|
existed and abort(400, "User <{0}> is already existed".format(kwargs['username']))
|
|
|
|
kwargs['nickname'] = kwargs['username'] if not kwargs.get('nickname') else kwargs['nickname']
|
|
kwargs['block'] = 0
|
|
return User.create(**kwargs)
|
|
|
|
@staticmethod
|
|
def update(rid, **kwargs):
|
|
user = User.get_by_id(rid) or abort(404, "User <{0}> does not exist".format(rid))
|
|
|
|
UserCache.clean(rid)
|
|
|
|
return user.update(**kwargs)
|
|
|
|
@classmethod
|
|
def delete(cls, uid):
|
|
user = User.get_by_id(uid) or abort(404, "User <{0}> does not exist".format(uid))
|
|
|
|
UserCache.clean(user)
|
|
|
|
user.soft_delete()
|