from ATRI.utils.request import post_bytes


URL = "https://saucenao.com/search.php"


class SauceNao:
    def __init__(self,
                 api_key: str,
                 output_type=2,
                 testmode=1,
                 dbmaski=32768,
                 db=5,
                 numres=5) -> None:
        params = dict()
        params['api_key'] = api_key
        params['output_type'] = output_type
        params['testmode'] = testmode
        params['dbmaski'] = dbmaski
        params['db'] = db
        params['numres'] = numres
        self.params = params
    
    async def search(self, url: str):
        self.params['url'] = url
        res = await post_bytes(url=URL, params=self.params)
        return res