mirror of
https://gist.github.com/6ba37e4d4084e858f917e271550ce5f6.git
synced 2024-09-20 00:34:20 +03:00
22 lines
637 B
Python
22 lines
637 B
Python
import logging
|
|
from typing import Optional
|
|
|
|
import requests
|
|
from bs4 import BeautifulSoup
|
|
|
|
|
|
class Iqdb:
|
|
@staticmethod
|
|
def search(file: str) -> Optional[str]:
|
|
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
|