myned/modufur
1
0
Fork 0
mirror of https://github.com/myned/modufur.git synced 2024-12-24 06:27:27 +00:00
Code Issues Activity

Add back e621 preference, refactor artist logic

Browse source
This commit is contained in:
Myned 2019-09-24 22:21:10 -04:00
parent 98eb5e2cc0
commit cd27cadb52
No known key found for this signature in database
GPG key ID: BC58C09870A63E59
1 changed files with 19 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
src/utils/scraper.py
View file

@ -81,19 +81,30 @@ async def query_saucenao(url):
similarity = int(float(match['header']['similarity']))
if similarity < 55:
return False
artist = 'member_name'
elif 'creator' in match['data']:
artist = 'creator'
else:
artist = 'imdb_id'
source = match['data']['ext_urls'][0]
for e in match['data']['ext_urls']:
if 'e621' in e:
source = e
break
artist = 'Unknown'
for e in (
'author_name',
'member_name',
'creator'
):
if e in match['data']:
artist = match['data'][e]
break
result = {
'source': match['data']['ext_urls'][0],
'artist': match['data'][artist],
'source': source,
'artist': artist,
'thumbnail': match['header']['thumbnail'],
'similarity': str(similarity),
'database': 'SauceNAO'
}
}
return result