Commited backup

KeyZenD/modules/YandexRS.py

@@ -0,0 +1,55 @@
+from .. import loader, utils
+import json
+import io
+import requests
+from PIL import Image
+import random
+import string
+
+@loader.tds
+class YandexReverseSearchMod(loader.Module):
+    """Reverse image search via Yandex (he is the best, imho)"""
+    strings = {"name": "YandexReverseSearch",
+               "search": "⚪⚪⚪\n⚪❓⚪\n⚪⚪⚪",
+               "no_reply": "<b>Reply to image or sticker!</b>",
+               "result": '<a href="{}"><b>🔴⚪🔴|See</b>\n<b>⚪🔴⚪|Search</b>\n<b>⚪🔴⚪|Results</b></a>',
+               "error": '<b>Something went wrong...</b>'}
+    @loader.owner
+    async def yarscmd(self, message):
+        """.yars <repy to image>"""
+        reply = await message.get_reply_message()
+        data = await check_media(message, reply)
+        if not data:
+            await utils.answer(message, self.strings("no_reply", message))
+            return
+        await utils.answer(message, self.strings("search", message))
+        searchUrl = 'https://yandex.ru/images/search'
+        files = {'upfile': ('blob', data, 'image/jpeg')}
+        params = {'rpt': 'imageview', 'format': 'json', 'request': '{"blocks":[{"block":"b-page_type_search-by-image__link"}]}'}
+        response = requests.post(searchUrl, params=params, files=files)
+        if response.ok:
+            query_string = json.loads(response.content)['blocks'][0]['params']['url']
+            link = searchUrl + '?' + query_string
+            text = self.strings("result", message).format(link)
+            await utils.answer(message, text)
+        else:
+        	await utils.answer(message, self.strings("error", message))
+        
+async def check_media(message, reply):
+    if reply and reply.media:
+        if reply.photo:
+            data = reply.photo
+        elif reply.document:
+            if reply.gif or reply.video or reply.audio or reply.voice:
+                return None
+            data = reply.media.document
+        else:
+            return None
+    else:
+        return None
+    if not data or data is None:
+        return None
+    else:
+        data = await message.client.download_file(data, bytes)
+        img = io.BytesIO(data)
+        return img