""" System Module Admin """ from django.contrib import admin from unfold.admin import ModelAdmin from igny8_core.admin.base import AccountAdminMixin, Igny8ModelAdmin from .models import AIPrompt, IntegrationSettings, AuthorProfile, Strategy from .global_settings_models import ( GlobalModuleSettings, GlobalIntegrationSettings, GlobalAIPrompt, GlobalAuthorProfile, GlobalStrategy, ) from django.contrib import messages from import_export.admin import ExportMixin, ImportExportMixin from import_export import resources class AIPromptResource(resources.ModelResource): """Resource class for importing/exporting AI Prompts""" class Meta: model = AIPrompt fields = ('id', 'account__name', 'prompt_type', 'prompt_value', 'is_active', 'created_at') export_order = fields import_id_fields = ('id',) skip_unchanged = True # Import settings admin from .settings_admin import ( SystemSettingsAdmin, AccountSettingsAdmin, UserSettingsAdmin, ModuleSettingsAdmin, AISettingsAdmin ) try: from .models import SystemLog, SystemStatus @admin.register(SystemLog) class SystemLogAdmin(AccountAdminMixin, Igny8ModelAdmin): list_display = ['id', 'account', 'module', 'level', 'action', 'message', 'created_at'] list_filter = ['module', 'level', 'created_at', 'account'] search_fields = ['message', 'action'] readonly_fields = ['created_at', 'updated_at'] date_hierarchy = 'created_at' @admin.register(SystemStatus) class SystemStatusAdmin(AccountAdminMixin, Igny8ModelAdmin): list_display = ['component', 'account', 'status', 'message', 'last_check'] list_filter = ['status', 'component', 'account'] search_fields = ['component', 'message'] readonly_fields = ['last_check'] except ImportError: pass @admin.register(AIPrompt) class AIPromptAdmin(ImportExportMixin, AccountAdminMixin, Igny8ModelAdmin): resource_class = AIPromptResource list_display = ['id', 'prompt_type', 'account', 'is_active', 'updated_at'] list_filter = ['prompt_type', 'is_active', 'account'] search_fields = ['prompt_type'] readonly_fields = ['created_at', 'updated_at', 'default_prompt'] actions = [ 'bulk_activate', 'bulk_deactivate', 'bulk_reset_to_default', ] fieldsets = ( ('Basic Info', { 'fields': ('account', 'prompt_type', 'is_active') }), ('Prompt Content', { 'fields': ('prompt_value', 'default_prompt') }), ('Timestamps', { 'fields': ('created_at', 'updated_at') }), ) def get_account_display(self, obj): """Safely get account name""" try: account = getattr(obj, 'account', None) return account.name if account else '-' except: return '-' get_account_display.short_description = 'Account' def bulk_activate(self, request, queryset): updated = queryset.update(is_active=True) self.message_user(request, f'{updated} AI prompt(s) activated.', messages.SUCCESS) bulk_activate.short_description = 'Activate selected prompts' def bulk_deactivate(self, request, queryset): updated = queryset.update(is_active=False) self.message_user(request, f'{updated} AI prompt(s) deactivated.', messages.SUCCESS) bulk_deactivate.short_description = 'Deactivate selected prompts' def bulk_reset_to_default(self, request, queryset): count = 0 for prompt in queryset: if prompt.default_prompt: prompt.prompt_value = prompt.default_prompt prompt.save() count += 1 self.message_user(request, f'{count} AI prompt(s) reset to default values.', messages.SUCCESS) bulk_reset_to_default.short_description = 'Reset to default values' class IntegrationSettingsResource(resources.ModelResource): """Resource class for exporting Integration Settings (config masked)""" class Meta: model = IntegrationSettings fields = ('id', 'account__name', 'integration_type', 'is_active', 'created_at') export_order = fields @admin.register(IntegrationSettings) class IntegrationSettingsAdmin(ExportMixin, AccountAdminMixin, Igny8ModelAdmin): resource_class = IntegrationSettingsResource list_display = ['id', 'integration_type', 'account', 'is_active', 'updated_at'] list_filter = ['integration_type', 'is_active', 'account'] search_fields = ['integration_type'] readonly_fields = ['created_at', 'updated_at'] actions = [ 'bulk_activate', 'bulk_deactivate', 'bulk_test_connection', ] fieldsets = ( ('Basic Info', { 'fields': ('account', 'integration_type', 'is_active') }), ('Configuration', { 'fields': ('config',), 'description': 'JSON configuration containing API keys and settings. Example: {"apiKey": "sk-...", "model": "gpt-4.1", "enabled": true}' }), ('Timestamps', { 'fields': ('created_at', 'updated_at') }), ) def get_readonly_fields(self, request, obj=None): """Make config readonly when viewing to prevent accidental exposure""" if obj: # Editing existing object return self.readonly_fields + ['config'] return self.readonly_fields def get_account_display(self, obj): """Safely get account name""" try: account = getattr(obj, 'account', None) return account.name if account else '-' except: return '-' get_account_display.short_description = 'Account' def bulk_activate(self, request, queryset): updated = queryset.update(is_active=True) self.message_user(request, f'{updated} integration setting(s) activated.', messages.SUCCESS) bulk_activate.short_description = 'Activate selected integrations' def bulk_deactivate(self, request, queryset): updated = queryset.update(is_active=False) self.message_user(request, f'{updated} integration setting(s) deactivated.', messages.SUCCESS) bulk_deactivate.short_description = 'Deactivate selected integrations' def bulk_test_connection(self, request, queryset): """Test connection for selected integration settings""" count = queryset.filter(is_active=True).count() self.message_user( request, f'{count} integration(s) queued for connection test. (Test logic to be implemented)', messages.INFO ) bulk_test_connection.short_description = 'Test connections' class AuthorProfileResource(resources.ModelResource): """Resource class for importing/exporting Author Profiles""" class Meta: model = AuthorProfile fields = ('id', 'name', 'account__name', 'tone', 'language', 'is_active', 'created_at') export_order = fields import_id_fields = ('id',) skip_unchanged = True @admin.register(AuthorProfile) class AuthorProfileAdmin(ImportExportMixin, AccountAdminMixin, Igny8ModelAdmin): resource_class = AuthorProfileResource list_display = ['name', 'account', 'tone', 'language', 'is_active', 'created_at'] list_filter = ['is_active', 'tone', 'language', 'account'] search_fields = ['name', 'description', 'tone'] readonly_fields = ['created_at', 'updated_at'] actions = [ 'bulk_activate', 'bulk_deactivate', 'bulk_clone', ] fieldsets = ( ('Basic Info', { 'fields': ('account', 'name', 'description', 'is_active') }), ('Writing Style', { 'fields': ('tone', 'language', 'structure_template') }), ('Timestamps', { 'fields': ('created_at', 'updated_at') }), ) def get_account_display(self, obj): """Safely get account name""" try: account = getattr(obj, 'account', None) return account.name if account else '-' except: return '-' get_account_display.short_description = 'Account' def bulk_activate(self, request, queryset): updated = queryset.update(is_active=True) self.message_user(request, f'{updated} author profile(s) activated.', messages.SUCCESS) bulk_activate.short_description = 'Activate selected profiles' def bulk_deactivate(self, request, queryset): updated = queryset.update(is_active=False) self.message_user(request, f'{updated} author profile(s) deactivated.', messages.SUCCESS) bulk_deactivate.short_description = 'Deactivate selected profiles' def bulk_clone(self, request, queryset): count = 0 for profile in queryset: profile_copy = AuthorProfile.objects.get(pk=profile.pk) profile_copy.pk = None profile_copy.name = f"{profile.name} (Copy)" profile_copy.is_active = False profile_copy.save() count += 1 self.message_user(request, f'{count} author profile(s) cloned.', messages.SUCCESS) bulk_clone.short_description = 'Clone selected profiles' class StrategyResource(resources.ModelResource): """Resource class for importing/exporting Strategies""" class Meta: model = Strategy fields = ('id', 'name', 'account__name', 'sector__name', 'is_active', 'created_at') export_order = fields import_id_fields = ('id',) skip_unchanged = True @admin.register(Strategy) class StrategyAdmin(ImportExportMixin, AccountAdminMixin, Igny8ModelAdmin): resource_class = StrategyResource list_display = ['name', 'account', 'sector', 'is_active', 'created_at'] list_filter = ['is_active', 'account'] search_fields = ['name', 'description'] readonly_fields = ['created_at', 'updated_at'] actions = [ 'bulk_activate', 'bulk_deactivate', 'bulk_clone', ] fieldsets = ( ('Basic Info', { 'fields': ('account', 'name', 'description', 'sector', 'is_active') }), ('Strategy Configuration', { 'fields': ('prompt_types', 'section_logic') }), ('Timestamps', { 'fields': ('created_at', 'updated_at') }), ) def get_account_display(self, obj): """Safely get account name""" try: account = getattr(obj, 'account', None) return account.name if account else '-' except: return '-' get_account_display.short_description = 'Account' def get_sector_display(self, obj): """Safely get sector name""" try: return obj.sector.name if obj.sector else 'Global' except: return 'Global' get_sector_display.short_description = 'Sector' def bulk_activate(self, request, queryset): updated = queryset.update(is_active=True) self.message_user(request, f'{updated} strategy/strategies activated.', messages.SUCCESS) bulk_activate.short_description = 'Activate selected strategies' def bulk_deactivate(self, request, queryset): updated = queryset.update(is_active=False) self.message_user(request, f'{updated} strategy/strategies deactivated.', messages.SUCCESS) bulk_deactivate.short_description = 'Deactivate selected strategies' def bulk_clone(self, request, queryset): count = 0 for strategy in queryset: strategy_copy = Strategy.objects.get(pk=strategy.pk) strategy_copy.pk = None strategy_copy.name = f"{strategy.name} (Copy)" strategy_copy.is_active = False strategy_copy.save() count += 1 self.message_user(request, f'{count} strategy/strategies cloned.', messages.SUCCESS) bulk_clone.short_description = 'Clone selected strategies' @admin.register(GlobalModuleSettings) class GlobalModuleSettingsAdmin(ModelAdmin): """Admin for Global Module Settings (Singleton)""" list_display = [ 'id', 'planner_enabled', 'writer_enabled', 'thinker_enabled', 'automation_enabled', 'site_builder_enabled', 'linker_enabled', 'optimizer_enabled', 'publisher_enabled', ] fieldsets = ( ('Module Toggles', { 'fields': ( 'planner_enabled', 'writer_enabled', 'thinker_enabled', 'automation_enabled', 'site_builder_enabled', 'linker_enabled', 'optimizer_enabled', 'publisher_enabled', ), 'description': 'Platform-wide module enable/disable controls. Changes affect all accounts immediately.' }), ) def has_add_permission(self, request): """Only allow one instance (singleton)""" return not self.model.objects.exists() def has_delete_permission(self, request, obj=None): """Prevent deletion of singleton""" return False # ===================================================================================== # GLOBAL SETTINGS ADMIN - Platform-wide defaults # ===================================================================================== @admin.register(GlobalIntegrationSettings) class GlobalIntegrationSettingsAdmin(Igny8ModelAdmin): """Admin for global integration settings (singleton)""" list_display = ["id", "is_active", "last_updated", "updated_by"] readonly_fields = ["last_updated"] fieldsets = ( ("OpenAI Settings", { "fields": ("openai_api_key", "openai_model", "openai_temperature", "openai_max_tokens"), "description": "Global OpenAI configuration used by all accounts (unless overridden)" }), ("Image Generation - Default Service", { "fields": ("default_image_service",), "description": "Choose which image generation service is used by default for all accounts" }), ("Image Generation - DALL-E", { "fields": ("dalle_api_key", "dalle_model", "dalle_size"), "description": "Global DALL-E (OpenAI) image generation configuration" }), ("Image Generation - Runware", { "fields": ("runware_api_key", "runware_model"), "description": "Global Runware image generation configuration" }), ("Universal Image Settings", { "fields": ("image_quality", "image_style", "max_in_article_images", "desktop_image_size", "mobile_image_size"), "description": "Image quality, style, and sizing settings that apply to ALL providers (DALL-E, Runware, etc.)" }), ("Status", { "fields": ("is_active", "last_updated", "updated_by") }), ) def has_add_permission(self, request): """Only allow one instance (singleton pattern)""" return not GlobalIntegrationSettings.objects.exists() def has_delete_permission(self, request, obj=None): """Don't allow deletion of singleton""" return False @admin.register(GlobalAIPrompt) class GlobalAIPromptAdmin(ExportMixin, Igny8ModelAdmin): """Admin for global AI prompt templates""" list_display = ["prompt_type", "version", "is_active", "last_updated"] list_filter = ["is_active", "prompt_type", "version"] search_fields = ["prompt_type", "description"] readonly_fields = ["last_updated", "created_at"] fieldsets = ( ("Basic Info", { "fields": ("prompt_type", "description", "is_active", "version") }), ("Prompt Content", { "fields": ("prompt_value", "variables"), "description": "Variables should be a list of variable names used in the prompt" }), ("Timestamps", { "fields": ("created_at", "last_updated") }), ) actions = ["increment_version"] def increment_version(self, request, queryset): """Increment version for selected prompts""" for prompt in queryset: prompt.version += 1 prompt.save() self.message_user(request, f"{queryset.count()} prompt(s) version incremented.", messages.SUCCESS) increment_version.short_description = "Increment version" @admin.register(GlobalAuthorProfile) class GlobalAuthorProfileAdmin(ImportExportMixin, Igny8ModelAdmin): """Admin for global author profile templates""" list_display = ["name", "category", "tone", "language", "is_active", "created_at"] list_filter = ["is_active", "category", "tone", "language"] search_fields = ["name", "description"] readonly_fields = ["created_at", "updated_at"] fieldsets = ( ("Basic Info", { "fields": ("name", "description", "category", "is_active") }), ("Writing Style", { "fields": ("tone", "language", "structure_template") }), ("Timestamps", { "fields": ("created_at", "updated_at") }), ) @admin.register(GlobalStrategy) class GlobalStrategyAdmin(ImportExportMixin, Igny8ModelAdmin): """Admin for global strategy templates""" list_display = ["name", "category", "is_active", "created_at"] list_filter = ["is_active", "category"] search_fields = ["name", "description"] readonly_fields = ["created_at", "updated_at"] fieldsets = ( ("Basic Info", { "fields": ("name", "description", "category", "is_active") }), ("Strategy Configuration", { "fields": ("prompt_types", "section_logic") }), ("Timestamps", { "fields": ("created_at", "updated_at") }), )