Initial commit: Location Management Flask app
This commit is contained in:
111
app/__init__.py
Normal file
111
app/__init__.py
Normal file
@@ -0,0 +1,111 @@
|
||||
"""Flask application factory."""
|
||||
import os
|
||||
from flask import Flask
|
||||
from flask_sqlalchemy import SQLAlchemy
|
||||
from flask_login import LoginManager
|
||||
from flask_migrate import Migrate
|
||||
from flask_socketio import SocketIO
|
||||
|
||||
from config import config_map
|
||||
|
||||
db = SQLAlchemy()
|
||||
login_manager = LoginManager()
|
||||
migrate = Migrate()
|
||||
socketio = SocketIO()
|
||||
|
||||
|
||||
def create_app(config_name: str = "default") -> Flask:
|
||||
app = Flask(__name__, template_folder="templates", static_folder="static")
|
||||
app.config.from_object(config_map[config_name])
|
||||
|
||||
# ── extensions ────────────────────────────────────────────────────────────
|
||||
db.init_app(app)
|
||||
migrate.init_app(app, db)
|
||||
socketio.init_app(app, cors_allowed_origins="*", async_mode="threading")
|
||||
|
||||
login_manager.init_app(app)
|
||||
login_manager.login_view = "auth.login"
|
||||
login_manager.login_message = "Please log in to access this page."
|
||||
login_manager.login_message_category = "warning"
|
||||
|
||||
# ── blueprints ────────────────────────────────────────────────────────────
|
||||
from app.routes.auth import auth_bp
|
||||
from app.routes.dashboard import dashboard_bp
|
||||
from app.routes.boards import boards_bp
|
||||
from app.routes.workflows import workflows_bp
|
||||
from app.routes.api import api_bp
|
||||
from app.routes.admin import admin_bp
|
||||
|
||||
app.register_blueprint(auth_bp)
|
||||
app.register_blueprint(dashboard_bp)
|
||||
app.register_blueprint(boards_bp, url_prefix="/boards")
|
||||
app.register_blueprint(workflows_bp, url_prefix="/workflows")
|
||||
app.register_blueprint(api_bp, url_prefix="/api")
|
||||
app.register_blueprint(admin_bp, url_prefix="/admin")
|
||||
|
||||
# ── user loader ───────────────────────────────────────────────────────────
|
||||
from app.models.user import User
|
||||
|
||||
@login_manager.user_loader
|
||||
def load_user(user_id):
|
||||
return db.session.get(User, int(user_id))
|
||||
|
||||
# ── discover board drivers ─────────────────────────────────────────────────
|
||||
from app.drivers.registry import registry
|
||||
registry.discover()
|
||||
app.logger.info("Board drivers loaded: %s", [d.DRIVER_ID for d in registry.all()])
|
||||
|
||||
# ── create tables & seed admin on first run ───────────────────────────────
|
||||
with app.app_context():
|
||||
db.create_all()
|
||||
_seed_admin(app)
|
||||
_add_entities_column(app)
|
||||
_migrate_board_types(app)
|
||||
|
||||
return app
|
||||
|
||||
|
||||
def _add_entities_column(app: Flask) -> None:
|
||||
"""Add entities_json column to boards table if it doesn't exist yet."""
|
||||
from sqlalchemy import text
|
||||
with app.app_context():
|
||||
try:
|
||||
db.session.execute(text("ALTER TABLE boards ADD COLUMN entities_json TEXT DEFAULT '{}'"))
|
||||
db.session.commit()
|
||||
app.logger.info("Added entities_json column to boards table")
|
||||
except Exception:
|
||||
db.session.rollback() # Column already exists — safe to ignore
|
||||
|
||||
|
||||
def _migrate_board_types(app: Flask) -> None:
|
||||
"""Rename legacy board_type values to current driver IDs."""
|
||||
from app.models.board import Board
|
||||
renames = {
|
||||
"olimex_esp32_c6": "olimex_esp32_c6_evb",
|
||||
}
|
||||
total = 0
|
||||
for old, new in renames.items():
|
||||
rows = Board.query.filter_by(board_type=old).all()
|
||||
for b in rows:
|
||||
b.board_type = new
|
||||
total += 1
|
||||
if total:
|
||||
db.session.commit()
|
||||
app.logger.info("Migrated %d board(s) to updated driver IDs", total)
|
||||
|
||||
|
||||
def _seed_admin(app: Flask) -> None:
|
||||
"""Create default admin account if no users exist."""
|
||||
from app.models.user import User
|
||||
from werkzeug.security import generate_password_hash
|
||||
|
||||
if User.query.count() == 0:
|
||||
admin = User(
|
||||
username="admin",
|
||||
password_hash=generate_password_hash("admin"),
|
||||
role="admin",
|
||||
is_active=True,
|
||||
)
|
||||
db.session.add(admin)
|
||||
db.session.commit()
|
||||
app.logger.info("Default admin user created (username=admin password=admin)")
|
||||
Reference in New Issue
Block a user