More files
This commit is contained in:
@ -1,5 +1,6 @@
|
||||
import os
|
||||
import importlib.util
|
||||
import importlib
|
||||
from flask import Flask
|
||||
from flask_sqlalchemy import SQLAlchemy
|
||||
from flask_migrate import Migrate
|
||||
@ -18,7 +19,7 @@ def create_app():
|
||||
db.init_app(app)
|
||||
migrate.init_app(app, db)
|
||||
login_manager.init_app(app)
|
||||
login_manager.login_view = 'auth.login' # Redirect for @login_required
|
||||
login_manager.login_view = 'auth.login'
|
||||
|
||||
# Register error handlers
|
||||
from .errors import bp as errors_bp
|
||||
@ -27,18 +28,41 @@ def create_app():
|
||||
# Plugin auto-loader
|
||||
plugin_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'plugins'))
|
||||
for plugin in os.listdir(plugin_path):
|
||||
full_path = os.path.join(plugin_path, plugin, 'routes.py')
|
||||
if os.path.isfile(full_path):
|
||||
spec = importlib.util.spec_from_file_location(f"plugins.{plugin}.routes", full_path)
|
||||
mod = importlib.util.module_from_spec(spec)
|
||||
spec.loader.exec_module(mod)
|
||||
if hasattr(mod, 'bp'):
|
||||
app.register_blueprint(mod.bp)
|
||||
|
||||
# Register CLI commands if the plugin has any
|
||||
if hasattr(mod, 'cli_commands'):
|
||||
for command in mod.cli_commands:
|
||||
app.cli.add_command(command)
|
||||
# Skip folders that end with `.noload`
|
||||
if plugin.endswith('.noload'):
|
||||
print(f"[⏭] Skipping plugin '{plugin}' (marked as .noload)")
|
||||
continue
|
||||
|
||||
plugin_dir = os.path.join(plugin_path, plugin)
|
||||
if not os.path.isdir(plugin_dir):
|
||||
continue
|
||||
|
||||
# Register routes
|
||||
route_file = os.path.join(plugin_dir, 'routes.py')
|
||||
if os.path.isfile(route_file):
|
||||
try:
|
||||
spec = importlib.util.spec_from_file_location(f"plugins.{plugin}.routes", route_file)
|
||||
mod = importlib.util.module_from_spec(spec)
|
||||
spec.loader.exec_module(mod)
|
||||
if hasattr(mod, 'bp'):
|
||||
app.register_blueprint(mod.bp)
|
||||
except Exception as e:
|
||||
print(f"[⚠️] Failed to load routes from plugin '{plugin}': {e}")
|
||||
|
||||
# Register CLI commands
|
||||
init_file = os.path.join(plugin_dir, '__init__.py')
|
||||
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)
|
||||
except Exception as e:
|
||||
print(f"[⚠️] Failed to load CLI from plugin '{plugin}': {e}")
|
||||
|
||||
@app.context_processor
|
||||
def inject_current_year():
|
||||
from datetime import datetime
|
||||
return {'current_year': datetime.now().year}
|
||||
|
||||
return app
|
||||
|
||||
|
Reference in New Issue
Block a user