from motor.motor_asyncio import AsyncIOMotorDatabase
import schemas
from crud.base import CRUDBase

__all__ = 'api_interview',


class ApiInterview(CRUDBase):

    # 获取所有面试安排信息
    async def get_api_interview(self, db: AsyncIOMotorDatabase):
        return await self.find_many(db)

    # 获取单个面试安排数据
    async def get_one(self, db: AsyncIOMotorDatabase, data_in: schemas.Url_quanxian):
        return await self.find_one(db, {'user_id': data_in.user_id})

    # 插入一条全新的面试安排数据
    async def insert_quanxian(self, db: AsyncIOMotorDatabase, data_in: schemas.Url_module):
        return await self.insert_one(db, data_in.dict())

    # 更新一条面试安排数据
    async def updata_quanxian_module(self, db: AsyncIOMotorDatabase, data_in: schemas.Url_module):
        return await self.update_one(db, {'auth_id': data_in.auth_id, 'path_name': data_in.path_name},
                                     {'$set': {'api_list': data_in.api_list, 'api_name': data_in.api_name,
                                               'state': data_in.state}})

    # 获取一条权限模板信息
    async def get_one_module(self, db: AsyncIOMotorDatabase, data_in: schemas.Add_module):
        return await self.find_one(db, {'auth_id': data_in.auth_id})

    # 更新一条面试安排数据
    async def update_one_interview(self, db: AsyncIOMotorDatabase, res):
        return await self.update_one(db, {'_id': res['_id']}, {
            '$set': res})


api_interview = ApiInterview('api_interview')