mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-07 03:35:35 +00:00
130 lines
3.2 KiB
Python
130 lines
3.2 KiB
Python
"""
|
|
Contains all the folder routes.
|
|
"""
|
|
import os
|
|
import psutil
|
|
|
|
from pathlib import Path
|
|
from flask import Blueprint, request
|
|
|
|
from app import settings
|
|
from app.lib.folderslib import GetFilesAndDirs
|
|
from app.db.sqlite.settings import SettingsSQLMethods as db
|
|
from app.models import Folder
|
|
from app.utils import create_folder_hash, is_windows, win_replace_slash
|
|
|
|
api = Blueprint("folder", __name__, url_prefix="/")
|
|
|
|
|
|
@api.route("/folder", methods=["POST"])
|
|
def get_folder_tree():
|
|
"""
|
|
Returns a list of all the folders and tracks in the given folder.
|
|
"""
|
|
data = request.get_json()
|
|
req_dir = "$home"
|
|
|
|
if data is not None:
|
|
try:
|
|
req_dir: str = data["folder"]
|
|
except KeyError:
|
|
req_dir = "$home"
|
|
|
|
root_dirs = db.get_root_dirs()
|
|
|
|
try:
|
|
if req_dir == "$home" and root_dirs[0] == "$home":
|
|
req_dir = settings.USER_HOME_DIR
|
|
except IndexError:
|
|
pass
|
|
|
|
if req_dir == "$home":
|
|
folders = [Path(f) for f in root_dirs]
|
|
|
|
return {
|
|
"folders": [
|
|
Folder(
|
|
name=f.name if f.name != "" else str(f).replace("\\", "/"),
|
|
path=win_replace_slash(str(f)),
|
|
has_tracks=True,
|
|
is_sym=f.is_symlink(),
|
|
path_hash=create_folder_hash(*f.parts[1:]),
|
|
)
|
|
for f in folders
|
|
],
|
|
"tracks": [],
|
|
}
|
|
|
|
if is_windows():
|
|
req_dir = req_dir + "/"
|
|
# TODO: Test this on Windows
|
|
else:
|
|
req_dir = "/" + req_dir + "/" if not req_dir.startswith("/") else req_dir + "/"
|
|
|
|
tracks, folders = GetFilesAndDirs(req_dir)()
|
|
|
|
return {
|
|
"tracks": tracks,
|
|
"folders": sorted(folders, key=lambda i: i.name),
|
|
}
|
|
|
|
|
|
def get_all_drives(is_win: bool = False):
|
|
"""
|
|
Returns a list of all the drives on a Windows machine.
|
|
"""
|
|
drives = psutil.disk_partitions(all=False)
|
|
drives = [d.mountpoint for d in drives]
|
|
|
|
if is_win:
|
|
drives = [win_replace_slash(d) for d in drives]
|
|
else:
|
|
remove = ["/boot", "/boot/efi", "/tmp"]
|
|
drives = [d for d in drives if d not in remove]
|
|
|
|
return drives
|
|
|
|
|
|
@api.route("/folder/dir-browser", methods=["POST"])
|
|
def list_folders():
|
|
"""
|
|
Returns a list of all the folders in the given folder.
|
|
"""
|
|
data = request.get_json()
|
|
is_win = is_windows()
|
|
|
|
try:
|
|
req_dir: str = data["folder"]
|
|
except KeyError:
|
|
req_dir = "$home"
|
|
|
|
if req_dir == "$home":
|
|
# req_dir = settings.USER_HOME_DIR
|
|
# if is_win:
|
|
return {
|
|
"folders": [
|
|
{"name": d, "path": d}
|
|
for d in get_all_drives(is_win=is_win)
|
|
]
|
|
}
|
|
|
|
if is_win:
|
|
req_dir = req_dir + "/"
|
|
else:
|
|
req_dir = "/" + req_dir + "/"
|
|
req_dir = str(Path(req_dir).resolve())
|
|
|
|
try:
|
|
entries = os.scandir(req_dir)
|
|
except PermissionError:
|
|
return {"folders": []}
|
|
|
|
dirs = [e.name for e in entries if e.is_dir() and not e.name.startswith(".")]
|
|
dirs = [
|
|
{"name": d, "path": win_replace_slash(os.path.join(req_dir, d))} for d in dirs
|
|
]
|
|
|
|
return {
|
|
"folders": sorted(dirs, key=lambda i: i["name"]),
|
|
}
|