mirror of
https://github.com/Dvorinka/swingmusic-extended.git
synced 2026-06-03 20:13:02 +00:00
cwilvx
7852be5e3f
+ add extra fields for backup in favorites and scrobble data - not yet for the playlist tracks
79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
from dataclasses import asdict
|
|
import json
|
|
from pathlib import Path
|
|
import shutil
|
|
from time import time
|
|
from flask_openapi3 import Tag
|
|
from flask_openapi3 import APIBlueprint
|
|
from app.api.auth import admin_required
|
|
|
|
from app.db.userdata import FavoritesTable, PlaylistTable, ScrobbleTable
|
|
from app.settings import Paths
|
|
|
|
bp_tag = Tag(name="Backup and Restore", description="Backup and Restore")
|
|
api = APIBlueprint("backup_and_restore", __name__, url_prefix="/", abp_tags=[bp_tag])
|
|
|
|
|
|
@api.post("/backup")
|
|
@admin_required()
|
|
def backup():
|
|
"""
|
|
Create a backup file of your favorites, playlists and scrobble data.
|
|
"""
|
|
backup_dir = Path(Paths.get_app_dir()) / "backup"
|
|
backup_dir.mkdir(parents=True, exist_ok=True)
|
|
backup_name = f"backup.{int(time())}"
|
|
backup_file = backup_dir / f"{backup_name}.json"
|
|
|
|
# INFO: Image folder for playlist images
|
|
img_folder = backup_dir / "images" / backup_name
|
|
img_folder.mkdir(parents=True, exist_ok=True)
|
|
|
|
favorites = FavoritesTable.get_all()
|
|
favorites = [asdict(entry) for entry in favorites]
|
|
|
|
scrobbles = ScrobbleTable.get_all(start=0)
|
|
scrobbles = [asdict(entry) for entry in scrobbles]
|
|
|
|
# SECTION: Playlists
|
|
playlists = PlaylistTable.get_all()
|
|
playlist_dicts = []
|
|
|
|
for entry in playlists:
|
|
playlist = asdict(entry)
|
|
for key in ["_last_updated", "has_image", "images", "duration", "count"]:
|
|
del playlist[key]
|
|
|
|
playlist_dicts.append(playlist)
|
|
|
|
# copy images
|
|
if playlist["thumb"]:
|
|
img_path = Path(Paths.get_playlist_img_path()) / playlist["thumb"]
|
|
shutil.copy(img_path, img_folder / playlist["thumb"])
|
|
|
|
# !SECTION
|
|
|
|
data = {
|
|
"favorites": favorites,
|
|
"scrobbles": scrobbles,
|
|
"playlists": playlist_dicts,
|
|
}
|
|
|
|
with open(backup_file, "w") as f:
|
|
json.dump(data, f, indent=4)
|
|
|
|
return {
|
|
"msg": "Backup created",
|
|
"data_path": str(backup_file),
|
|
"images_path": str(img_folder),
|
|
}, 200
|
|
|
|
|
|
@api.post("/restore")
|
|
@admin_required()
|
|
def restore():
|
|
"""
|
|
Restore your favorites, playlists and scrobble data from a backup file.
|
|
"""
|
|
return {"msg": "Restore"}
|