190 lines
6.6 KiB
Python
190 lines
6.6 KiB
Python
"""
|
|
Custom AdminSite for IGNY8 to organize models into proper groups
|
|
"""
|
|
from django.contrib import admin
|
|
from django.contrib.admin.apps import AdminConfig
|
|
from django.apps import apps
|
|
|
|
|
|
class Igny8AdminSite(admin.AdminSite):
|
|
"""
|
|
Custom AdminSite that organizes models into the planned groups:
|
|
1. Billing & Tenancy
|
|
2. Sites & Users
|
|
3. Global Reference Data
|
|
4. Planner
|
|
5. Writer Module
|
|
6. Thinker Module
|
|
7. System Configuration
|
|
"""
|
|
site_header = 'IGNY8 Administration'
|
|
site_title = 'IGNY8 Admin'
|
|
index_title = 'IGNY8 Administration'
|
|
|
|
def get_app_list(self, request):
|
|
"""
|
|
Customize the app list to organize models into proper groups
|
|
"""
|
|
# Get the default app list
|
|
app_dict = self._build_app_dict(request)
|
|
|
|
# Define our custom groups with their models (using object_name)
|
|
custom_groups = {
|
|
'Billing & Tenancy': {
|
|
'models': [
|
|
('igny8_core_auth', 'Plan'),
|
|
('igny8_core_auth', 'Account'),
|
|
('igny8_core_auth', 'Subscription'),
|
|
('billing', 'CreditTransaction'),
|
|
('billing', 'CreditUsageLog'),
|
|
('billing', 'Invoice'),
|
|
('billing', 'Payment'),
|
|
('billing', 'CreditPackage'),
|
|
('billing', 'CreditCostConfig'),
|
|
],
|
|
},
|
|
'Sites & Users': {
|
|
'models': [
|
|
('igny8_core_auth', 'Site'),
|
|
('igny8_core_auth', 'User'),
|
|
('igny8_core_auth', 'SiteUserAccess'),
|
|
('igny8_core_auth', 'PasswordResetToken'),
|
|
('igny8_core_auth', 'Sector'),
|
|
],
|
|
},
|
|
'Global Reference Data': {
|
|
'models': [
|
|
('igny8_core_auth', 'Industry'),
|
|
('igny8_core_auth', 'IndustrySector'),
|
|
('igny8_core_auth', 'SeedKeyword'),
|
|
('site_building', 'BusinessType'),
|
|
('site_building', 'AudienceProfile'),
|
|
('site_building', 'BrandPersonality'),
|
|
('site_building', 'HeroImageryDirection'),
|
|
],
|
|
},
|
|
'Planner': {
|
|
'models': [
|
|
('planner', 'Keywords'),
|
|
('planner', 'Clusters'),
|
|
('planner', 'ContentIdeas'),
|
|
],
|
|
},
|
|
'Writer Module': {
|
|
'models': [
|
|
('writer', 'Tasks'),
|
|
('writer', 'Content'),
|
|
('writer', 'Images'),
|
|
('writer', 'ContentTaxonomy'),
|
|
('writer', 'ContentAttribute'),
|
|
('writer', 'ContentTaxonomyRelation'),
|
|
('writer', 'ContentClusterMap'),
|
|
],
|
|
},
|
|
'Thinker Module': {
|
|
'models': [
|
|
('system', 'AIPrompt'),
|
|
('system', 'AuthorProfile'),
|
|
('system', 'Strategy'),
|
|
('ai', 'AITaskLog'),
|
|
],
|
|
},
|
|
'System Configuration': {
|
|
'models': [
|
|
('system', 'IntegrationSettings'),
|
|
('system', 'SystemLog'),
|
|
('system', 'SystemStatus'),
|
|
('system', 'SystemSettings'),
|
|
('system', 'AccountSettings'),
|
|
('system', 'UserSettings'),
|
|
('system', 'ModuleSettings'),
|
|
('system', 'AISettings'),
|
|
('system', 'ModuleEnableSettings'),
|
|
# Automation config lives under the automation app - include here
|
|
('automation', 'AutomationConfig'),
|
|
('automation', 'AutomationRun'),
|
|
],
|
|
},
|
|
'Payments': {
|
|
'models': [
|
|
('billing', 'PaymentMethodConfig'),
|
|
('billing', 'AccountPaymentMethod'),
|
|
],
|
|
},
|
|
'Integrations & Sync': {
|
|
'models': [
|
|
('integration', 'SiteIntegration'),
|
|
('integration', 'SyncEvent'),
|
|
],
|
|
},
|
|
'Publishing': {
|
|
'models': [
|
|
('publishing', 'PublishingRecord'),
|
|
('publishing', 'DeploymentRecord'),
|
|
],
|
|
},
|
|
'Optimization': {
|
|
'models': [
|
|
('optimization', 'OptimizationTask'),
|
|
],
|
|
},
|
|
'Django Internals': {
|
|
'models': [
|
|
('admin', 'LogEntry'),
|
|
('auth', 'Group'),
|
|
('auth', 'Permission'),
|
|
('contenttypes', 'ContentType'),
|
|
('sessions', 'Session'),
|
|
],
|
|
},
|
|
}
|
|
|
|
# Build the custom app list
|
|
app_list = []
|
|
|
|
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
|
|
if app_label in app_dict:
|
|
app_data = app_dict[app_label]
|
|
# Look for the model in the app's models
|
|
for model in app_data.get('models', []):
|
|
if model['object_name'] == model_name:
|
|
group_models.append(model)
|
|
break
|
|
|
|
# Only add the group if it has models
|
|
if group_models:
|
|
app_list.append({
|
|
'name': group_name,
|
|
'app_label': group_name.lower().replace(' ', '_').replace('&', ''),
|
|
'app_url': None,
|
|
'has_module_perms': True,
|
|
'models': group_models,
|
|
})
|
|
|
|
# Sort the app list by our custom order
|
|
order = [
|
|
'Billing & Tenancy',
|
|
'Sites & Users',
|
|
'Global Reference Data',
|
|
'Planner',
|
|
'Writer Module',
|
|
'Thinker Module',
|
|
'System Configuration',
|
|
'Payments',
|
|
'Integrations & Sync',
|
|
'Publishing',
|
|
'Optimization',
|
|
'Django Internals',
|
|
]
|
|
|
|
app_list.sort(key=lambda x: order.index(x['name']) if x['name'] in order else 999)
|
|
|
|
return app_list
|
|
|
|
|
|
|