from app.models import KnowledgeModel, group_knowledge_table
|
from app.models.user_model import UserModel
|
from Log import logger
|
|
|
async def get_knowledge_list(db, user_id, keyword, page_size, page_index):
|
user = db.query(UserModel).filter(UserModel.id == user_id).first()
|
if user is None:
|
return {"rows": []}
|
if user.permission == "admin":
|
query = db.query(KnowledgeModel)
|
else:
|
group_list = [i.id for i in user.groups]
|
query = db.query(KnowledgeModel)
|
query = query.filter(KnowledgeModel.tenant_id == user_id)
|
|
query = query.union(
|
db.query(KnowledgeModel).join(
|
group_knowledge_table,
|
KnowledgeModel.id == group_knowledge_table.c.knowledge_id
|
).filter(
|
group_knowledge_table.c.group_id.in_(group_list)
|
)
|
)
|
if keyword:
|
query = query.filter(KnowledgeModel.name.like('%{}%'.format(keyword)))
|
total = query.count()
|
if page_size:
|
query = query.limit(page_size).offset((page_index - 1) * page_size)
|
return {"total": total, "rows": [kld.to_json() for kld in query.all()]}
|
