mirror of
https://gist.github.com/6ba37e4d4084e858f917e271550ce5f6.git
synced 2024-09-20 00:34:20 +03:00
18 lines
573 B
Python
18 lines
573 B
Python
|
from bs4 import BeautifulSoup
|
||
|
|
import logging
|
||
|
|
import requests
|
||
|
|
|
||
|
|
class Iqdb:
|
||
|
|
def search(self, file):
|
||
|
|
logging.info('Searching %s', file)
|
||
|
|
files = {'file': open(file, 'rb')}
|
||
|
|
resp = requests.post('https://iqdb.org/', files=files, timeout=10)
|
||
|
|
doc = BeautifulSoup(resp.text, 'html.parser')
|
||
|
|
for tag in doc.select(".image a"):
|
||
|
|
url = tag.get("href")
|
||
|
|
if "danbooru.donmai.us/posts" in url:
|
||
|
|
if url.startswith("//"):
|
||
|
|
url = "https:" + url
|
||
|
|
return url
|
||
|
|
return None
|