add socketio to global state

This commit is contained in:
Simone Tesini 2025-08-02 00:48:21 +02:00
parent bc9bf71824
commit 85854e5673
2 changed files with 8 additions and 6 deletions

View file

@ -1,15 +1,15 @@
from typing import Any
import dotenv import dotenv
from flask import Flask, Response, jsonify, request from flask import Flask, Response, jsonify, request
from flask_cors import CORS from flask_cors import CORS
from flask_socketio import SocketIO, emit from flask_socketio import SocketIO, emit
from .state import State
from .connect import get_connection from .connect import get_connection
from .room import Room from .room import Room
from .song import Song, init_db, get_song_by_title_artist, add_song_in_db from .song import Song, add_song_in_db, get_song_by_title_artist, init_db
from .song_fetch import lastfm_query_search, download_song_mp3 from .song_fetch import download_song_mp3, lastfm_query_search
from .state import State
from typing import Any
dotenv.load_dotenv() dotenv.load_dotenv()
@ -20,7 +20,7 @@ socketio = SocketIO(app)
CORS(app) CORS(app)
db_conn = get_connection() db_conn = get_connection()
state = State(app, db_conn.cursor()) state = State(app, socketio, db_conn.cursor())
init_db(state.db) init_db(state.db)
state.rooms[1000] = Room( state.rooms[1000] = Room(

View file

@ -2,6 +2,7 @@ from dataclasses import dataclass, field
from sqlite3 import Cursor from sqlite3 import Cursor
from flask import Flask from flask import Flask
from flask_socketio import SocketIO
from .room import Room from .room import Room
@ -9,5 +10,6 @@ from .room import Room
@dataclass @dataclass
class State: class State:
app: Flask app: Flask
socketio: SocketIO
db: Cursor db: Cursor
rooms: dict[int, Room] = field(default_factory=dict) # { room_id: room, ... } rooms: dict[int, Room] = field(default_factory=dict) # { room_id: room, ... }