mirror/maloja
1
0
Fork 0
mirror of https://github.com/krateng/maloja.git synced 2023-08-10 21:12:55 +03:00
Code Issues Projects Releases Wiki Activity

Added proper logging to upgrade script

Browse Source
This commit is contained in:
krateng 2022-04-08 21:17:17 +02:00
parent 1207475e4d
commit fba21b7128
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
maloja/upgrade.py
View File

@ -35,7 +35,7 @@ def upgrade_db(callback_add_scrobbles):
if os.path.exists(oldfolder): if os.path.exists(oldfolder):
scrobblefiles = [f for f in os.listdir(oldfolder) if f.endswith(".tsv")] scrobblefiles = [f for f in os.listdir(oldfolder) if f.endswith(".tsv")]
if len(scrobblefiles) > 0: if len(scrobblefiles) > 0:
print(col['yellow']("Upgrading v2 Database to v3 Database. This could take a while...")) log("Upgrading v2 Database to v3 Database. This could take a while...",color='yellow')
idx = 0 idx = 0
for sf in scrobblefiles: for sf in scrobblefiles:
idx += 1 idx += 1
@ -48,10 +48,11 @@ def upgrade_db(callback_add_scrobbles):
else: else:
origin = 'unknown' origin = 'unknown'
# TODO still tsv module here!
from doreah import tsv from doreah import tsv
scrobbles = tsv.parse(os.path.join(oldfolder,sf),"int","string","string","string","string",comments=False) scrobbles = tsv.parse(os.path.join(oldfolder,sf),"int","string","string","string","string",comments=False)
scrobblelist = [] scrobblelist = []
print(f"\tImporting from {sf} ({idx}/{len(scrobblefiles)}) - {len(scrobbles)} Scrobbles") log(f"\tImporting from {sf} ({idx}/{len(scrobblefiles)}) - {len(scrobbles)} Scrobbles")
for scrobble in scrobbles: for scrobble in scrobbles:
timestamp, artists, title, album, duration = scrobble timestamp, artists, title, album, duration = scrobble
if album in ('-',''): album = None if album in ('-',''): album = None
@ -73,4 +74,4 @@ def upgrade_db(callback_add_scrobbles):
}) })
callback_add_scrobbles(scrobblelist) callback_add_scrobbles(scrobblelist)
os.rename(os.path.join(oldfolder,sf),os.path.join(newfolder,sf)) os.rename(os.path.join(oldfolder,sf),os.path.join(newfolder,sf))
print(col['yellow']("Done!")) log("Done!",color='yellow')