69 lines
1.8 KiB
Python
69 lines
1.8 KiB
Python
import sys, logging
|
|
from multiprocessing import Manager
|
|
|
|
from flask import Flask, request
|
|
from flask_cors import CORS
|
|
|
|
from api.child_api import child_api
|
|
from api.image_api import image_api
|
|
from api.reward_api import reward_api
|
|
from api.task_api import task_api
|
|
from events.broadcaster import Broadcaster
|
|
from events.sse import sse_response_for_user, send_to_user, set_shared_manager
|
|
from db.default import initializeImages
|
|
|
|
# Configure logging once at application startup
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S',
|
|
stream=sys.stdout,
|
|
force=True # Override any existing config
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Create shared manager
|
|
manager = Manager()
|
|
set_shared_manager(manager)
|
|
|
|
app = Flask(__name__)
|
|
#CORS(app, resources={r"/api/*": {"origins": ["http://localhost:3000", "http://localhost:5173"]}})
|
|
app.register_blueprint(child_api)
|
|
app.register_blueprint(reward_api)
|
|
app.register_blueprint(task_api)
|
|
app.register_blueprint(image_api)
|
|
CORS(app)
|
|
|
|
@app.route("/events")
|
|
def events():
|
|
# Authenticate user or read a token
|
|
user_id = request.args.get("user_id")
|
|
if not user_id:
|
|
return {"error": "Missing user_id"}, 400
|
|
|
|
|
|
return sse_response_for_user(user_id)
|
|
|
|
|
|
@app.route("/notify/<user_id>")
|
|
def notify_user(user_id):
|
|
# Example trigger
|
|
send_to_user(user_id, {
|
|
"type": "notification",
|
|
"message": f"Hello {user_id}, this is a private message!"
|
|
})
|
|
|
|
return {"status": "sent"}
|
|
|
|
def start_background_threads():
|
|
broadcaster = Broadcaster()
|
|
broadcaster.daemon = True
|
|
broadcaster.start()
|
|
|
|
# Initialize background workers on server start
|
|
initializeImages()
|
|
start_background_threads()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=False, host='0.0.0.0', port=5000, threaded=True) |