letsswing_admin/main.py

from nicegui import ui, app
from datetime import datetime, timedelta
from montydb import MontyClient
from montydb.types.objectid import ObjectId
import locale
from pages import manager, user, useredit, index
import signal
import os
import shutil

from fastapi.responses import RedirectResponse
from fastapi.responses import FileResponse

class CallbackHandler:
    def __init__(self):
        self.callbacks = {}

    def register_callback(self, event_name, func):
        if event_name not in self.callbacks:
            self.callbacks[event_name] = []
        self.callbacks[event_name].append(func)

    def execute_callbacks(self, event_name, *args, **kwargs):
        if event_name in self.callbacks:
            for callback in self.callbacks[event_name][:]:
                try:
                    callback(*args, **kwargs)
                except Exception as e:
                    print(f"Error executing callback {callback}: {e}")
                    self.callbacks[event_name].remove(callback)


handler = CallbackHandler()

passwords = {'admin': [0,'sw1ng3rs',0, "Matteo Benedetto"], 'user': [1,'pass',1, "Pippo"]}
# Set the locale to Italian
locale.setlocale(locale.LC_ALL, 'it_IT.utf8')

def styling():
    ui.colors(primary='black', secondary="lightgray")
    ui.query('body').style(f'font-family: sans-serif')
    ui.add_head_html('<meta name="theme-color" content="#000000"/>')

def login_check():
    if not app.storage.user.get('authenticated', False):
        return RedirectResponse('/login')

# Create a connection
conn = MontyClient("db")
# Access your database
db = conn.letsswing


# Routes
@ui.page('/manager/{sede}/{livello}', title="Let's Swing Admin", favicon="assets/favicon.ico")
def manager_page(sede:int, livello:int) -> None:
   login_check()
   styling()

   manage = manager.Manager(handler)
   manage.main(db, sede, livello)

# Routes
app.add_static_files('/static', 'static')

@ui.page('/', title="Let's Swing Admin", favicon="assets/favicon.ico")
def index_page() -> None:
    
    login_check()
    styling()
    index.main(db)
    

@ui.page('/user/{userid}', title="Let's Swing Admin", favicon="assets/favicon.ico")
def user_page(userid: str) -> None:
    login_check()
    styling()
    user.main(userid, db, handler)

@ui.page('/edit/{userid}/{sede}/{livello}', title="Let's Swing Admin", favicon="assets/favicon.ico")
def useredit_page(userid: str, sede: int, livello: int) -> None:
    login_check()
    styling()
    useredit.main(userid, sede, livello, db, handler)

@ui.page('/backup/{password}')
def api_backup(password:  str) -> None:
    if password != "1337swing":
        return {'msg':'password errata'}

    # Percorso completo del file zip
    filename = "backup_db_" +datetime.now().strftime('%Y%m%d%H%M%S')
    zip_path = os.path.join('backup', filename)

    # Crea il file zip
    shutil.make_archive(zip_path, 'zip', "db")

    return FileResponse(path=zip_path + '.zip', media_type='application/octet-stream', filename=filename+".zip")



@ui.page('/login',title="Let's Swing Admin", favicon="assets/favicon.ico")
def page_login() -> None:
    styling()
    def try_login() -> None:  # local function to avoid passing username and password as arguments
        if passwords.get(username.value)[1] == password.value:
            app.storage.user.update({'username': username.value, 'authenticated': True, 'id' : passwords.get(username.value)[0], "level":  passwords.get(username.value)[2], "name":  passwords.get(username.value)[3]})
            ui.open('/')
        else:
            ui.notify('username o password errata', color='negative')

    if app.storage.user.get('authenticated', False):
        return RedirectResponse('/')
    ui.add_head_html('''
                        <link rel="manifest" href="/manifest.json">
                        ''')
    with ui.card().classes('absolute-center items-center'):
        ui.image('assets/logo.svg')
        ui.label('Let\'s Swing Admin').classes('text-2xl')
        username = ui.input('Username').on('keydown.enter', try_login).classes('w-full')
        password = ui.input('Password', password=True, password_toggle_button=True).on('keydown.enter', try_login)
        ui.button('Log in', on_click=try_login)
        
        ui.add_body_html('''<script>if ('serviceWorker' in navigator) {
    window.addEventListener('load', function() {
        navigator.serviceWorker.register('/static/service-worker.js').then(function(registration) {
        // Registration was successful
        console.log('ServiceWorker registration successful with scope: ', registration.scope);
        }, function(err) {
        // registration failed :(
        console.log('ServiceWorker registration failed: ', err);
        });
    });
}</script>''')
                     

# PWA conversion
@ui.page('/manifest.json')
def manifest_json() -> None:

    return FileResponse(path='assets/site.webmanifest', filename="site.webmanifest")


# Run main loop
ui.run(host="0.0.0.0", port=9999, storage_secret='5w1ng4ndn0th1ngm0r3',workers=1)