1
0
Fork 0
mirror of https://github.com/myned/modufur.git synced 2024-12-24 22:27:28 +00:00

Change reverse image footer to display source hostname

This commit is contained in:
Myned 2019-11-12 16:10:25 -05:00
parent 68fc9bb541
commit ffee58656c
No known key found for this signature in database
GPG key ID: BC58C09870A63E59

View file

@ -1,6 +1,7 @@
import aiohttp import aiohttp
import ast import ast
import re import re
from urllib.parse import urlparse
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
import lxml import lxml
@ -72,7 +73,7 @@ async def query_kheina(url):
'artist': match[4], 'artist': match[4],
'thumbnail': f'https://f002.backblazeb2.com/file/kheinacom/{match[1]}.jpg', 'thumbnail': f'https://f002.backblazeb2.com/file/kheinacom/{match[1]}.jpg',
'similarity': str(similarity), 'similarity': str(similarity),
'database': 'Kheina' 'database': urlparse(match[3].replace('\\', '')).hostname.split('.')[0]
} }
return result return result
@ -124,7 +125,7 @@ async def query_saucenao(url):
'artist': artist, 'artist': artist,
'thumbnail': match['header']['thumbnail'], 'thumbnail': match['header']['thumbnail'],
'similarity': str(similarity), 'similarity': str(similarity),
'database': 'SauceNAO' 'database': urlparse(source).hostname.split('.')[0]
} }
return result return result