from motor.motor_asyncio import AsyncIOMotorDatabase

import schemas
from crud.base import CRUDBase

__all__ = 'api_list',


class CRUDApiList(CRUDBase):
    async def add_api(self, db: AsyncIOMotorDatabase, data_in: schemas.AddApi):
        where = {'path': data_in.path}
        data = {'$set': schemas.AddApiDB(**data_in.dict()).dict(by_alias=True)}

        return await self.update_one(db, where, data, upsert=True)

    async def edit_api(self, db: AsyncIOMotorDatabase, data_in: schemas.EditApi):
        where = {'_id': data_in.id}
        data = {'$set': data_in.dict(exclude={'id'})}
        return await self.update_one(db, where, data)


    async def all_api(self, db: AsyncIOMotorDatabase):
        return await self.find_many(db)





    async def del_api(self, db: AsyncIOMotorDatabase, data_in: schemas.DelApi):
        return await self.delete_id(db, *data_in.ids)

    async def create_index(self, db: AsyncIOMotorDatabase):
        await db[self.coll_name].create_index('path', unique=True)


api_list = CRUDApiList('api_list')