1 files changed, 29 insertions, 0 deletions
diff --git a/ATRI/plugins/saucenao/data_source.py b/ATRI/plugins/saucenao/data_source.py
new file mode 100644
index 0000000..636c540
--- /dev/null
+++ b/ATRI/plugins/saucenao/data_source.py
@@ -0,0 +1,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