navidrome updater
This commit is contained in:
51
navidrome/analyzer.py
Normal file
51
navidrome/analyzer.py
Normal file
@@ -0,0 +1,51 @@
|
||||
import sqlite3
|
||||
import meilisearch
|
||||
|
||||
client = meilisearch.Client("http://127.0.0.1:7700", "aSampleMasterKey")
|
||||
index = client.index("itunes-albums")
|
||||
|
||||
con = sqlite3.connect("navidrome.db")
|
||||
cur = con.cursor()
|
||||
instertor = con.cursor()
|
||||
update_statement = """UPDATE album SET
|
||||
persistent_id = ?,
|
||||
itunes_date = ?,
|
||||
meili_nb_res = ?,
|
||||
meili_title = ?,
|
||||
meili_artist = ?,
|
||||
meili_doc = ?,
|
||||
meili_song_count = ?,
|
||||
validated = ?
|
||||
WHERE id = ?"""
|
||||
|
||||
for row in cur.execute("SELECT * FROM album;"):
|
||||
msearch = index.search(
|
||||
row[1], {"filter": [f'Artist="{row[2]}"'], "showRankingScore": True}
|
||||
)
|
||||
meili_result_len = len(msearch["hits"])
|
||||
if meili_result_len > 0:
|
||||
meili_hit = msearch["hits"][0]
|
||||
if meili_hit["_rankingScore"] > 0.5:
|
||||
print(f"Update {row[1]}")
|
||||
val_score = 0
|
||||
if meili_result_len == 1:
|
||||
val_score += 50
|
||||
if meili_hit["Track Count"] == row[3]:
|
||||
val_score += 40
|
||||
if meili_hit["Name"] == row[1]:
|
||||
val_score += 30
|
||||
if row[2] in meili_hit["Artist"]:
|
||||
val_score += 10
|
||||
data = (
|
||||
meili_hit["Persistent ID"],
|
||||
meili_hit["Date Added"],
|
||||
meili_result_len,
|
||||
meili_hit["Name"],
|
||||
str(meili_hit["Artist"]),
|
||||
str(meili_hit),
|
||||
meili_hit["Track Count"],
|
||||
val_score,
|
||||
row[0],
|
||||
)
|
||||
instertor.execute(update_statement, data)
|
||||
con.commit()
|
||||
Reference in New Issue
Block a user