326 lines
13 KiB
Python
326 lines
13 KiB
Python
"""
|
|
Custom AdminSite for IGNY8 to organize models into proper groups using Unfold
|
|
NO EMOJIS - Unfold handles all icons via Material Design
|
|
"""
|
|
from django.contrib import admin
|
|
from django.contrib.admin.apps import AdminConfig
|
|
from django.apps import apps
|
|
from django.urls import path, reverse_lazy
|
|
from django.shortcuts import redirect
|
|
from django.contrib.admin import sites
|
|
from unfold.admin import ModelAdmin as UnfoldModelAdmin
|
|
from unfold.sites import UnfoldAdminSite
|
|
|
|
|
|
class Igny8AdminSite(UnfoldAdminSite):
|
|
"""
|
|
Custom AdminSite based on Unfold that organizes models into the planned groups
|
|
"""
|
|
site_header = 'IGNY8 Administration'
|
|
site_title = 'IGNY8 Admin'
|
|
index_title = 'IGNY8 Administration'
|
|
|
|
def get_urls(self):
|
|
"""Get admin URLs with dashboard and reports available"""
|
|
from django.urls import path
|
|
from .dashboard import admin_dashboard
|
|
from .reports import revenue_report, usage_report, content_report, data_quality_report
|
|
|
|
urls = super().get_urls()
|
|
custom_urls = [
|
|
path('dashboard/', self.admin_view(admin_dashboard), name='dashboard'),
|
|
path('reports/revenue/', self.admin_view(revenue_report), name='report_revenue'),
|
|
path('reports/usage/', self.admin_view(usage_report), name='report_usage'),
|
|
path('reports/content/', self.admin_view(content_report), name='report_content'),
|
|
path('reports/data-quality/', self.admin_view(data_quality_report), name='report_data_quality'),
|
|
]
|
|
return custom_urls + urls
|
|
|
|
def index(self, request, extra_context=None):
|
|
"""Redirect to custom dashboard"""
|
|
from django.shortcuts import redirect
|
|
return redirect('admin:dashboard')
|
|
|
|
def get_sidebar_list(self, request):
|
|
"""
|
|
Override Unfold's get_sidebar_list to return our custom app groups
|
|
Convert Django app_list format to Unfold sidebar navigation format
|
|
"""
|
|
# Get our custom Django app list
|
|
django_apps = self.get_app_list(request, app_label=None)
|
|
|
|
# Convert to Unfold navigation format: {title, items: [{title, link, icon}]}
|
|
sidebar_groups = []
|
|
|
|
for app in django_apps:
|
|
group = {
|
|
'title': app['name'],
|
|
'collapsible': True,
|
|
'items': []
|
|
}
|
|
|
|
# Convert each model to navigation item
|
|
for model in app.get('models', []):
|
|
if model.get('perms', {}).get('view', False) or model.get('perms', {}).get('change', False):
|
|
item = {
|
|
'title': model['name'],
|
|
'link': model['admin_url'],
|
|
'icon': None, # Unfold will use default
|
|
'has_permission': True, # CRITICAL: Template checks this
|
|
}
|
|
group['items'].append(item)
|
|
|
|
# Only add groups that have items
|
|
if group['items']:
|
|
sidebar_groups.append(group)
|
|
|
|
return sidebar_groups
|
|
|
|
def each_context(self, request):
|
|
"""
|
|
Override context to ensure our custom app_list is always used
|
|
This is called by all admin templates for sidebar rendering
|
|
|
|
CRITICAL FIX: Force custom sidebar on ALL pages including model detail/list views
|
|
"""
|
|
# CRITICAL: Must call parent to get sidebar_navigation set
|
|
context = super().each_context(request)
|
|
|
|
# DEBUGGING: Print to console what parent returned
|
|
print(f"\n=== DEBUG each_context for {request.path} ===")
|
|
print(f"sidebar_navigation length from parent: {len(context.get('sidebar_navigation', []))}")
|
|
if context.get('sidebar_navigation'):
|
|
print(f"First sidebar group: {context['sidebar_navigation'][0].get('title', 'NO TITLE')}")
|
|
|
|
# Force our custom app list to be used everywhere - IGNORE app_label parameter
|
|
custom_apps = self.get_app_list(request, app_label=None)
|
|
context['available_apps'] = custom_apps
|
|
context['app_list'] = custom_apps # Also set app_list for compatibility
|
|
|
|
# CRITICAL FIX: Ensure sidebar_navigation is using our custom sidebar
|
|
# Parent's each_context already called get_sidebar_list(), which returns our custom sidebar
|
|
# So sidebar_navigation should already be correct, but let's verify
|
|
if not context.get('sidebar_navigation') or len(context.get('sidebar_navigation', [])) == 0:
|
|
# If sidebar_navigation is empty, force it
|
|
print("WARNING: sidebar_navigation was empty, forcing it!")
|
|
context['sidebar_navigation'] = self.get_sidebar_list(request)
|
|
|
|
print(f"Final sidebar_navigation length: {len(context['sidebar_navigation'])}")
|
|
print("=== END DEBUG ===\n")
|
|
|
|
return context
|
|
|
|
def get_app_list(self, request, app_label=None):
|
|
"""
|
|
Customize the app list to organize models into logical groups
|
|
NO EMOJIS - Unfold handles all icons via Material Design
|
|
|
|
Args:
|
|
request: The HTTP request
|
|
app_label: IGNORED - Always return full custom sidebar for consistency
|
|
"""
|
|
# CRITICAL: Always build full app_dict (ignore app_label) for consistent sidebar
|
|
app_dict = self._build_app_dict(request, None)
|
|
|
|
# Define our custom groups with their models (using object_name)
|
|
# Organized by business function - Material icons configured in Unfold
|
|
custom_groups = {
|
|
'Accounts & Users': {
|
|
'models': [
|
|
('igny8_core_auth', 'Account'),
|
|
('igny8_core_auth', 'User'),
|
|
('igny8_core_auth', 'Site'),
|
|
('igny8_core_auth', 'Sector'),
|
|
('igny8_core_auth', 'SiteUserAccess'),
|
|
('igny8_core_auth', 'Plan'),
|
|
('igny8_core_auth', 'Subscription'),
|
|
('igny8_core_auth', 'PasswordResetToken'),
|
|
('igny8_core_auth', 'Industry'),
|
|
('igny8_core_auth', 'IndustrySector'),
|
|
('igny8_core_auth', 'SeedKeyword'),
|
|
],
|
|
},
|
|
'Billing & Tenancy': {
|
|
'models': [
|
|
('billing', 'Invoice'),
|
|
('billing', 'Payment'),
|
|
('billing', 'CreditTransaction'),
|
|
('billing', 'CreditUsageLog'),
|
|
('billing', 'CreditPackage'),
|
|
('billing', 'PaymentMethodConfig'),
|
|
('billing', 'AccountPaymentMethod'),
|
|
('billing', 'CreditCostConfig'),
|
|
('billing', 'PlanLimitUsage'),
|
|
],
|
|
},
|
|
'Writer Module': {
|
|
'models': [
|
|
('writer', 'Content'),
|
|
('writer', 'Tasks'),
|
|
('writer', 'Images'),
|
|
('writer', 'ContentTaxonomy'),
|
|
('writer', 'ContentAttribute'),
|
|
('writer', 'ContentTaxonomyRelation'),
|
|
('writer', 'ContentClusterMap'),
|
|
],
|
|
},
|
|
'Planner': {
|
|
'models': [
|
|
('planner', 'Clusters'),
|
|
('planner', 'Keywords'),
|
|
('planner', 'ContentIdeas'),
|
|
],
|
|
},
|
|
'Publishing': {
|
|
'models': [
|
|
('publishing', 'PublishingRecord'),
|
|
('publishing', 'DeploymentRecord'),
|
|
],
|
|
},
|
|
'Optimization': {
|
|
'models': [
|
|
('optimization', 'OptimizationTask'),
|
|
],
|
|
},
|
|
'Automation': {
|
|
'models': [
|
|
('automation', 'AutomationConfig'),
|
|
('automation', 'AutomationRun'),
|
|
],
|
|
},
|
|
'Integration': {
|
|
'models': [
|
|
('integration', 'SiteIntegration'),
|
|
('integration', 'SyncEvent'),
|
|
],
|
|
},
|
|
'AI Framework': {
|
|
'models': [
|
|
('ai', 'AITaskLog'),
|
|
],
|
|
},
|
|
'System Configuration': {
|
|
'models': [
|
|
('system', 'AIPrompt'),
|
|
('system', 'Strategy'),
|
|
('system', 'AuthorProfile'),
|
|
('system', 'ContentTemplate'),
|
|
('system', 'TaxonomyConfig'),
|
|
('system', 'SystemSetting'),
|
|
('system', 'ContentTypeConfig'),
|
|
('system', 'PublishingChannel'),
|
|
('system', 'APIKey'),
|
|
('system', 'WebhookConfig'),
|
|
('system', 'NotificationConfig'),
|
|
('system', 'AuditLog'),
|
|
],
|
|
},
|
|
'Celery Results': {
|
|
'models': [
|
|
('django_celery_results', 'TaskResult'),
|
|
('django_celery_results', 'GroupResult'),
|
|
],
|
|
},
|
|
'Content Types': {
|
|
'models': [
|
|
('contenttypes', 'ContentType'),
|
|
],
|
|
},
|
|
'Administration': {
|
|
'models': [
|
|
('admin', 'LogEntry'),
|
|
],
|
|
},
|
|
'Authentication and Authorization': {
|
|
'models': [
|
|
('auth', 'Group'),
|
|
('auth', 'Permission'),
|
|
],
|
|
},
|
|
'Sessions': {
|
|
'models': [
|
|
('sessions', 'Session'),
|
|
],
|
|
},
|
|
}
|
|
|
|
# ALWAYS build and return our custom organized app list
|
|
# regardless of app_label parameter (for consistent sidebar on all pages)
|
|
organized_apps = []
|
|
|
|
# Add Dashboard link as first item
|
|
organized_apps.append({
|
|
'name': '📊 Dashboard',
|
|
'app_label': '_dashboard',
|
|
'app_url': '/admin/dashboard/',
|
|
'has_module_perms': True,
|
|
'models': [],
|
|
})
|
|
|
|
# Add Reports section with links to all reports
|
|
organized_apps.append({
|
|
'name': 'Reports & Analytics',
|
|
'app_label': '_reports',
|
|
'app_url': '#',
|
|
'has_module_perms': True,
|
|
'models': [
|
|
{
|
|
'name': 'Revenue Report',
|
|
'object_name': 'RevenueReport',
|
|
'admin_url': '/admin/reports/revenue/',
|
|
'view_only': True,
|
|
'perms': {'view': True},
|
|
},
|
|
{
|
|
'name': 'Usage Report',
|
|
'object_name': 'UsageReport',
|
|
'admin_url': '/admin/reports/usage/',
|
|
'view_only': True,
|
|
'perms': {'view': True},
|
|
},
|
|
{
|
|
'name': 'Content Report',
|
|
'object_name': 'ContentReport',
|
|
'admin_url': '/admin/reports/content/',
|
|
'view_only': True,
|
|
'perms': {'view': True},
|
|
},
|
|
{
|
|
'name': 'Data Quality Report',
|
|
'object_name': 'DataQualityReport',
|
|
'admin_url': '/admin/reports/data-quality/',
|
|
'view_only': True,
|
|
'perms': {'view': True},
|
|
},
|
|
],
|
|
})
|
|
|
|
for group_name, group_config in custom_groups.items():
|
|
group_models = []
|
|
|
|
for app_label, model_name in group_config['models']:
|
|
# Find the model in app_dict
|
|
for app in app_dict.values():
|
|
if app['app_label'] == app_label:
|
|
for model in app.get('models', []):
|
|
if model['object_name'] == model_name:
|
|
group_models.append(model)
|
|
break
|
|
|
|
if group_models:
|
|
# Get the first model's app_label to use as the real app_label
|
|
first_model_app_label = group_config['models'][0][0]
|
|
organized_apps.append({
|
|
'name': group_name,
|
|
'app_label': first_model_app_label, # Use real app_label, not fake one
|
|
'app_url': f'/admin/{first_model_app_label}/', # Real URL, not '#'
|
|
'has_module_perms': True,
|
|
'models': group_models,
|
|
})
|
|
|
|
return organized_apps
|
|
|
|
|
|
# Instantiate custom admin site
|
|
admin_site = Igny8AdminSite(name='admin')
|