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