mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-07 03:35:35 +00:00
142 lines
3.5 KiB
Python
142 lines
3.5 KiB
Python
"""
|
|
Contains the functions to prepare the server for use.
|
|
"""
|
|
import os
|
|
import shutil
|
|
from configparser import ConfigParser
|
|
import caribou # pylint: disable=import-error
|
|
|
|
from app import settings
|
|
from app.db.sqlite import create_connection, create_tables, queries
|
|
from app.db.store import Store
|
|
from app.settings import APP_DB_PATH, USERDATA_DB_PATH
|
|
from app.utils import get_home_res_path
|
|
|
|
config = ConfigParser()
|
|
|
|
config_path = get_home_res_path("pyinstaller.config.ini")
|
|
config.read(config_path)
|
|
|
|
|
|
try:
|
|
IS_BUILD = config["DEFAULT"]["BUILD"] == "True"
|
|
except KeyError:
|
|
# If the key doesn't exist, it means that the app is being executed in dev mode.
|
|
IS_BUILD = False
|
|
|
|
|
|
class CopyFiles:
|
|
"""Copies assets to the app directory."""
|
|
|
|
def __init__(self) -> None:
|
|
assets_dir = "assets"
|
|
|
|
if IS_BUILD:
|
|
assets_dir = get_home_res_path("assets")
|
|
|
|
files = [
|
|
{
|
|
"src": assets_dir,
|
|
"dest": os.path.join(settings.APP_DIR, "assets"),
|
|
"is_dir": True,
|
|
}
|
|
]
|
|
|
|
for entry in files:
|
|
src = os.path.join(os.getcwd(), entry["src"])
|
|
|
|
if entry["is_dir"]:
|
|
shutil.copytree(
|
|
src,
|
|
entry["dest"],
|
|
ignore=shutil.ignore_patterns(
|
|
"*.pyc",
|
|
),
|
|
copy_function=shutil.copy2,
|
|
dirs_exist_ok=True,
|
|
)
|
|
break
|
|
|
|
shutil.copy2(src, entry["dest"])
|
|
|
|
|
|
def create_config_dir() -> None:
|
|
"""
|
|
Creates the config directory if it doesn't exist.
|
|
"""
|
|
|
|
home_dir = os.path.expanduser("~")
|
|
config_folder = os.path.join(home_dir, settings.CONFIG_FOLDER)
|
|
|
|
thumb_path = os.path.join("images", "thumbnails")
|
|
small_thumb_path = os.path.join(thumb_path, "small")
|
|
large_thumb_path = os.path.join(thumb_path, "large")
|
|
|
|
artist_img_path = os.path.join("images", "artists")
|
|
small_artist_img_path = os.path.join(artist_img_path, "small")
|
|
large_artist_img_path = os.path.join(artist_img_path, "large")
|
|
|
|
playlist_img_path = os.path.join("images", "playlists")
|
|
|
|
dirs = [
|
|
"", # creates the config folder
|
|
"images",
|
|
thumb_path,
|
|
small_thumb_path,
|
|
large_thumb_path,
|
|
artist_img_path,
|
|
small_artist_img_path,
|
|
large_artist_img_path,
|
|
playlist_img_path,
|
|
]
|
|
|
|
for _dir in dirs:
|
|
path = os.path.join(config_folder, _dir)
|
|
exists = os.path.exists(path)
|
|
|
|
if not exists:
|
|
os.makedirs(path)
|
|
os.chmod(path, 0o755)
|
|
|
|
CopyFiles()
|
|
|
|
|
|
def setup_sqlite():
|
|
"""
|
|
Create Sqlite databases and tables.
|
|
"""
|
|
# if os.path.exists(DB_PATH):
|
|
# os.remove(DB_PATH)
|
|
|
|
app_db_conn = create_connection(APP_DB_PATH)
|
|
playlist_db_conn = create_connection(USERDATA_DB_PATH)
|
|
|
|
create_tables(app_db_conn, queries.CREATE_APPDB_TABLES)
|
|
create_tables(playlist_db_conn, queries.CREATE_USERDATA_TABLES)
|
|
|
|
userdb_migrations = get_home_res_path("app") / "migrations" / "userdata"
|
|
maindb_migrations = get_home_res_path("app") / "migrations" / "main"
|
|
|
|
caribou.upgrade(
|
|
APP_DB_PATH,
|
|
maindb_migrations,
|
|
)
|
|
|
|
caribou.upgrade(
|
|
str(USERDATA_DB_PATH),
|
|
str(userdb_migrations),
|
|
)
|
|
|
|
app_db_conn.close()
|
|
playlist_db_conn.close()
|
|
|
|
Store.load_all_tracks()
|
|
Store.process_folders()
|
|
Store.load_albums()
|
|
Store.load_artists()
|
|
|
|
|
|
def run_setup():
|
|
create_config_dir()
|
|
setup_sqlite()
|