summaryrefslogtreecommitdiff
path: root/ATRI/plugins/saucenao/data_source.py
blob: 636c54061349e7e14cc17a96d66aff5e4a9b53b4 (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
28
29
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=0,
                 dbmask=None,
                 dbmaski=32768,
                 db=5,
                 numres=1) -> None:
        params = dict()
        params['api_key'] = api_key
        params['output_type'] = output_type
        params['testmode'] = testmode
        params['dbmask'] = dbmask
        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