mirror of
https://github.com/Dvorinka/swingmusic-extended.git
synced 2026-06-03 20:13:02 +00:00
cwilvx
128 lines
3.2 KiB
Python
128 lines
3.2 KiB
Python
"""
|
|
Handles arguments passed to the program.
|
|
"""
|
|
|
|
import sys
|
|
from getpass import getpass
|
|
|
|
import click
|
|
import PyInstaller.__main__ as bundler
|
|
|
|
from app import settings
|
|
from app.db.userdata import UserTable
|
|
from app.logger import log
|
|
from app.setup.sqlite import setup_sqlite
|
|
from app.utils.auth import hash_password
|
|
from app.utils.paths import getFlaskOpenApiPath
|
|
from app.utils.wintools import is_windows
|
|
|
|
ALLARGS = settings.ALLARGS
|
|
ARGS = sys.argv[1:]
|
|
|
|
|
|
def handle_build(*args, **kwargs):
|
|
"""
|
|
Handles the --build argument. Builds the project into a single executable.
|
|
"""
|
|
if not args[2]:
|
|
return
|
|
|
|
if settings.IS_BUILD:
|
|
click.echo("Can't build the project. Exiting ...")
|
|
sys.exit(0)
|
|
|
|
info_keys = [
|
|
"SWINGMUSIC_APP_VERSION",
|
|
"GIT_LATEST_COMMIT_HASH",
|
|
"GIT_CURRENT_BRANCH",
|
|
]
|
|
|
|
lines = []
|
|
|
|
for key in info_keys:
|
|
value = settings.Info.get(key)
|
|
|
|
if not value:
|
|
log.error(
|
|
f"WARNING: {key} not resolved. Can't build the project. Exiting ..."
|
|
)
|
|
sys.exit(0)
|
|
|
|
lines.append(f'{key} = "{value}"\n')
|
|
|
|
try:
|
|
# write the info to the config file
|
|
with open("./app/configs.py", "w", encoding="utf-8") as file:
|
|
# copy the api keys to the config file
|
|
file.writelines(lines)
|
|
|
|
_s = ";" if is_windows() else ":"
|
|
|
|
flask_openapi_path = getFlaskOpenApiPath()
|
|
|
|
bundler.run(
|
|
[
|
|
"main.py",
|
|
"--onefile",
|
|
"--name",
|
|
"swingmusic",
|
|
"--clean",
|
|
f"--add-data=assets{_s}assets",
|
|
f"--add-data=client{_s}client",
|
|
f"--add-data=version.txt{_s}.",
|
|
f"--add-data={flask_openapi_path}/templates/static{_s}flask_openapi3/templates/static",
|
|
f"--icon=assets/logo-fill.light.ico",
|
|
"-y",
|
|
]
|
|
)
|
|
finally:
|
|
# revert and remove the api keys for dev mode
|
|
with open("./app/configs.py", "w", encoding="utf-8") as file:
|
|
lines = [f'{key} = ""\n' for key in info_keys]
|
|
file.writelines(lines)
|
|
|
|
sys.exit(0)
|
|
|
|
|
|
def handle_password_reset(*args, **kwargs):
|
|
"""
|
|
Handles the --password-reset argument. Resets the password.
|
|
"""
|
|
if not args[2]:
|
|
return
|
|
|
|
setup_sqlite()
|
|
|
|
username: str = ""
|
|
password: str = ""
|
|
|
|
# collect username
|
|
try:
|
|
username = input("Enter username: ")
|
|
except KeyboardInterrupt:
|
|
click.echo("\nOperation cancelled! Exiting ...")
|
|
sys.exit(0)
|
|
|
|
username = username.strip()
|
|
user = UserTable.get_by_username(username)
|
|
|
|
if not user:
|
|
click.echo(f"User {username} not found")
|
|
sys.exit(0)
|
|
|
|
# collect password
|
|
try:
|
|
password = getpass("Enter new password: ")
|
|
except KeyboardInterrupt:
|
|
click.echo("\nOperation cancelled! Exiting ...")
|
|
sys.exit(0)
|
|
|
|
try:
|
|
UserTable.update_one({"id": user.id, "password": hash_password(password)})
|
|
click.echo("Password reset successfully!")
|
|
except Exception as e:
|
|
click.echo(f"Error resetting password: {e}")
|
|
sys.exit(0)
|
|
|
|
sys.exit(0)
|