summaryrefslogtreecommitdiff
path: root/ATRI/plugins/saucenao/data_source.py
blob: efe8fe1aab25e383ae45f9df16367ff88c538e53 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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