mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-09 12:37:22 +00:00
101 lines
2.4 KiB
Python
101 lines
2.4 KiB
Python
from flask import Blueprint, request
|
|
from app import settings
|
|
|
|
from app.db.sqlite.settings import SettingsSQLMethods as sdb
|
|
from app.lib import populate
|
|
from app.logger import log
|
|
|
|
from app.db.store import Store
|
|
from app.utils import background
|
|
|
|
api = Blueprint("settings", __name__, url_prefix="/")
|
|
|
|
|
|
def get_child_dirs(parent: str, children: list[str]):
|
|
"""Returns child directories in a list, given a parent directory"""
|
|
|
|
return [dir for dir in children if dir.startswith(parent) and dir != parent]
|
|
|
|
|
|
@background
|
|
def rebuild_store(db_dirs: list[str]):
|
|
log.info("Rebuilding library...")
|
|
Store.remove_tracks_by_dir_except(db_dirs)
|
|
|
|
Store.load_all_tracks()
|
|
Store.process_folders()
|
|
Store.load_albums()
|
|
Store.load_artists()
|
|
|
|
populate.Populate()
|
|
|
|
log.info("Rebuilding library... ✅")
|
|
|
|
|
|
@api.route("/settings/add-root-dirs", methods=["POST"])
|
|
def add_root_dirs():
|
|
"""
|
|
Add custom root directories to the database.
|
|
"""
|
|
msg = {"msg": "Failed! No directories were given."}
|
|
|
|
data = request.get_json()
|
|
|
|
if data is None:
|
|
return msg, 400
|
|
|
|
try:
|
|
new_dirs: list[str] = data["new_dirs"]
|
|
removed_dirs: list[str] = data["removed"]
|
|
except KeyError:
|
|
return msg, 400
|
|
|
|
def finalize(new_dirs: list[str], removed_dirs: list[str], db_dirs: list[str]):
|
|
sdb.remove_root_dirs(removed_dirs)
|
|
sdb.add_root_dirs(new_dirs)
|
|
rebuild_store(db_dirs)
|
|
|
|
# ---
|
|
db_dirs = sdb.get_root_dirs()
|
|
|
|
if db_dirs[0] == "$home" and new_dirs[0] == "$home".strip():
|
|
return {"msg": "Not changed!"}
|
|
|
|
if db_dirs[0] == "$home":
|
|
sdb.remove_root_dirs(db_dirs)
|
|
|
|
try:
|
|
if new_dirs[0] == "$home":
|
|
finalize(["$home"], db_dirs, [settings.USER_HOME_DIR])
|
|
|
|
return {"msg": "Updated!"}
|
|
except IndexError:
|
|
pass
|
|
|
|
for _dir in new_dirs:
|
|
children = get_child_dirs(_dir, db_dirs)
|
|
removed_dirs.extend(children)
|
|
# ---
|
|
|
|
for _dir in removed_dirs:
|
|
try:
|
|
db_dirs.remove(_dir)
|
|
except ValueError:
|
|
pass
|
|
|
|
db_dirs.extend(new_dirs)
|
|
|
|
finalize(new_dirs, removed_dirs, db_dirs)
|
|
|
|
return {"msg": "Updated!"}
|
|
|
|
|
|
@api.route("/settings/get-root-dirs", methods=["GET"])
|
|
def get_root_dirs():
|
|
"""
|
|
Get custom root directories from the database.
|
|
"""
|
|
dirs = sdb.get_root_dirs()
|
|
|
|
return {"dirs": dirs}
|