"""Notifications plugin main class.""" import json import logging from pathlib import Path from typing import List, Dict, Optional, Type from flask import Flask, Blueprint import click from shopdb.plugins.base import BasePlugin, PluginMeta from shopdb.extensions import db from .models import Notification, NotificationType from .api import notifications_bp logger = logging.getLogger(__name__) class NotificationsPlugin(BasePlugin): """ Notifications plugin - manages announcements and notifications. Provides functionality for: - Creating and managing notifications/announcements - Displaying banner notifications - Calendar view of notifications """ def __init__(self): self._manifest = self._load_manifest() def _load_manifest(self) -> Dict: """Load plugin manifest from JSON file.""" manifest_path = Path(__file__).parent / 'manifest.json' if manifest_path.exists(): with open(manifest_path, 'r') as f: return json.load(f) return {} @property def meta(self) -> PluginMeta: """Return plugin metadata.""" return PluginMeta( name=self._manifest.get('name', 'notifications'), version=self._manifest.get('version', '1.0.0'), description=self._manifest.get( 'description', 'Notifications and announcements management' ), author=self._manifest.get('author', 'ShopDB Team'), dependencies=self._manifest.get('dependencies', []), core_version=self._manifest.get('core_version', '>=1.0.0'), api_prefix=self._manifest.get('api_prefix', '/api/notifications'), ) def get_blueprint(self) -> Optional[Blueprint]: """Return Flask Blueprint with API routes.""" return notifications_bp def get_models(self) -> List[Type]: """Return list of SQLAlchemy model classes.""" return [Notification, NotificationType] def init_app(self, app: Flask, db_instance) -> None: """Initialize plugin with Flask app.""" logger.info(f"Notifications plugin initialized (v{self.meta.version})") def on_install(self, app: Flask) -> None: """Called when plugin is installed.""" with app.app_context(): self._ensure_notification_types() logger.info("Notifications plugin installed") def _ensure_notification_types(self) -> None: """Ensure default notification types exist.""" default_types = [ ('Awareness', 'General awareness notification', '#17a2b8', 'info-circle'), ('Change', 'Planned change notification', '#ffc107', 'exchange-alt'), ('Incident', 'Incident or outage notification', '#dc3545', 'exclamation-triangle'), ('Maintenance', 'Scheduled maintenance notification', '#6c757d', 'wrench'), ('General', 'General announcement', '#28a745', 'bullhorn'), ] for typename, description, color, icon in default_types: existing = NotificationType.query.filter_by(typename=typename).first() if not existing: t = NotificationType( typename=typename, description=description, color=color, icon=icon ) db.session.add(t) logger.debug(f"Created notification type: {typename}") db.session.commit() def on_uninstall(self, app: Flask) -> None: """Called when plugin is uninstalled.""" logger.info("Notifications plugin uninstalled") def get_cli_commands(self) -> List: """Return CLI commands for this plugin.""" @click.group('notifications') def notifications_cli(): """Notifications plugin commands.""" pass @notifications_cli.command('list-types') def list_types(): """List all notification types.""" from flask import current_app with current_app.app_context(): types = NotificationType.query.filter_by(isactive=True).all() if not types: click.echo('No notification types found.') return click.echo('Notification Types:') for t in types: click.echo(f" [{t.notificationtypeid}] {t.typename} ({t.color})") @notifications_cli.command('stats') def stats(): """Show notification statistics.""" from flask import current_app from datetime import datetime with current_app.app_context(): now = datetime.utcnow() total = Notification.query.filter( Notification.isactive == True ).count() active = Notification.query.filter( Notification.isactive == True, Notification.startdate <= now, db.or_( Notification.enddate.is_(None), Notification.enddate >= now ) ).count() click.echo(f"Total notifications: {total}") click.echo(f"Currently active: {active}") @notifications_cli.command('create') @click.option('--title', required=True, help='Notification title') @click.option('--message', required=True, help='Notification message') @click.option('--type', 'type_name', default='General', help='Notification type') def create_notification(title, message, type_name): """Create a new notification.""" from flask import current_app with current_app.app_context(): ntype = NotificationType.query.filter_by(typename=type_name).first() if not ntype: click.echo(f"Error: Notification type '{type_name}' not found.") return n = Notification( title=title, message=message, notificationtypeid=ntype.notificationtypeid ) db.session.add(n) db.session.commit() click.echo(f"Created notification #{n.notificationid}: {title}") return [notifications_cli] def get_dashboard_widgets(self) -> List[Dict]: """Return dashboard widget definitions.""" return [ { 'name': 'Active Notifications', 'component': 'NotificationsWidget', 'endpoint': '/api/notifications/dashboard/summary', 'size': 'small', 'position': 1, }, ] def get_navigation_items(self) -> List[Dict]: """Return navigation menu items.""" return [ { 'name': 'Notifications', 'icon': 'bell', 'route': '/notifications', 'position': 5, }, { 'name': 'Calendar', 'icon': 'calendar', 'route': '/calendar', 'position': 6, }, ]