migration and testing

This commit is contained in:
2025-05-27 05:19:03 -05:00
parent d920660366
commit bba83af409
7 changed files with 64 additions and 45 deletions

View File

@ -1,7 +1,8 @@
import os
import json
import importlib.util
import importlib
import importlib.util
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate
@ -9,9 +10,10 @@ from flask_login import LoginManager
from flask_wtf.csrf import CSRFProtect
from dotenv import load_dotenv
# Load environment variables from .env
# Load environment variables
load_dotenv()
# Initialize extensions
db = SQLAlchemy()
migrate = Migrate()
login_manager = LoginManager()
@ -21,9 +23,9 @@ csrf = CSRFProtect()
def create_app():
app = Flask(__name__)
app.config.from_object('app.config.Config')
csrf.init_app(app)
# Initialize core extensions
csrf.init_app(app)
db.init_app(app)
migrate.init_app(app, db)
login_manager.init_app(app)
@ -33,8 +35,7 @@ def create_app():
from .errors import bp as errors_bp
app.register_blueprint(errors_bp)
# Plugin auto-loader
# Plugin auto-loader
# Auto-discover and register plugins
plugin_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'plugins'))
for plugin in os.listdir(plugin_path):
if plugin.endswith('.noload'):
@ -45,7 +46,7 @@ def create_app():
if not os.path.isdir(plugin_dir):
continue
# Register routes
# 1. Register routes
route_file = os.path.join(plugin_dir, 'routes.py')
if os.path.isfile(route_file):
try:
@ -57,25 +58,35 @@ def create_app():
except Exception as e:
print(f"[⚠️] Failed to load routes from plugin '{plugin}': {e}")
# Register CLI commands and plugin entry points
init_file = os.path.join(plugin_dir, '__init__.py')
# Define paths
init_file = os.path.join(plugin_dir, '__init__.py')
plugin_json = os.path.join(plugin_dir, 'plugin.json')
model_file = os.path.join(plugin_dir, 'models.py')
# 2. Register CLI commands and run entry point
if os.path.isfile(init_file):
try:
cli_module = importlib.import_module(f"plugins.{plugin}")
if hasattr(cli_module, 'register_cli'):
cli_module.register_cli(app)
plugin_json = os.path.join(plugin_dir, 'plugin.json')
if os.path.isfile(plugin_json):
try:
meta = json.load(open(plugin_json, 'r'))
entry = meta.get('entry_point')
if entry and hasattr(cli_module, entry):
getattr(cli_module, entry)(app)
except Exception as e:
print(f"[⚠️] Failed to run entry_point '{entry}' for plugin '{plugin}': {e}")
with open(plugin_json, 'r') as f:
meta = json.load(f)
entry = meta.get('entry_point')
if entry and hasattr(cli_module, entry):
getattr(cli_module, entry)(app)
except Exception as e:
print(f"[⚠️] Failed to load CLI from plugin '{plugin}': {e}")
print(f"[⚠️] Failed to load CLI for plugin '{plugin}': {e}")
# 3. Auto-load plugin models for migrations
if os.path.isfile(model_file):
try:
spec = importlib.util.spec_from_file_location(f"plugins.{plugin}.models", model_file)
mod = importlib.util.module_from_spec(spec)
spec.loader.exec_module(mod)
except Exception as e:
print(f"[⚠️] Failed to load models from plugin '{plugin}': {e}")
@app.context_processor
def inject_current_year():