221 lines
8.0 KiB
Python
221 lines
8.0 KiB
Python
from django.contrib import admin
|
|
from django.contrib import messages
|
|
from unfold.admin import ModelAdmin
|
|
from unfold.contrib.filters.admin import (
|
|
RangeDateFilter,
|
|
RangeNumericFilter,
|
|
RelatedDropdownFilter,
|
|
ChoicesDropdownFilter,
|
|
)
|
|
from igny8_core.admin.base import SiteSectorAdminMixin
|
|
from .models import Keywords, Clusters, ContentIdeas
|
|
from import_export.admin import ExportMixin
|
|
from import_export import resources
|
|
|
|
|
|
class KeywordsResource(resources.ModelResource):
|
|
"""Resource class for exporting Keywords"""
|
|
class Meta:
|
|
model = Keywords
|
|
fields = ('id', 'keyword', 'seed_keyword__keyword', 'site__name', 'sector__name',
|
|
'cluster__name', 'volume', 'difficulty', 'intent', 'status', 'created_at')
|
|
export_order = fields
|
|
|
|
|
|
@admin.register(Clusters)
|
|
class ClustersAdmin(SiteSectorAdminMixin, ModelAdmin):
|
|
list_display = ['name', 'site', 'sector', 'keywords_count', 'volume', 'status', 'created_at']
|
|
list_filter = [
|
|
('status', ChoicesDropdownFilter),
|
|
('site', RelatedDropdownFilter),
|
|
('sector', RelatedDropdownFilter),
|
|
('volume', RangeNumericFilter),
|
|
('created_at', RangeDateFilter),
|
|
]
|
|
search_fields = ['name']
|
|
ordering = ['name']
|
|
autocomplete_fields = ['site', 'sector']
|
|
|
|
def get_site_display(self, obj):
|
|
"""Safely get site name"""
|
|
try:
|
|
return obj.site.name if obj.site else '-'
|
|
except:
|
|
return '-'
|
|
get_site_display.short_description = 'Site'
|
|
|
|
def get_sector_display(self, obj):
|
|
"""Safely get sector name"""
|
|
try:
|
|
return obj.sector.name if obj.sector else '-'
|
|
except:
|
|
return '-'
|
|
|
|
|
|
@admin.register(Keywords)
|
|
class KeywordsAdmin(ExportMixin, SiteSectorAdminMixin, ModelAdmin):
|
|
resource_class = KeywordsResource
|
|
list_display = ['keyword', 'seed_keyword', 'site', 'sector', 'cluster', 'volume', 'difficulty', 'intent', 'status', 'created_at']
|
|
list_filter = [
|
|
('status', ChoicesDropdownFilter),
|
|
('intent', ChoicesDropdownFilter),
|
|
('site', RelatedDropdownFilter),
|
|
('sector', RelatedDropdownFilter),
|
|
('cluster', RelatedDropdownFilter),
|
|
('volume', RangeNumericFilter),
|
|
('difficulty', RangeNumericFilter),
|
|
('created_at', RangeDateFilter),
|
|
]
|
|
search_fields = ['keyword', 'seed_keyword__keyword']
|
|
ordering = ['-created_at']
|
|
autocomplete_fields = ['cluster', 'site', 'sector', 'seed_keyword']
|
|
actions = [
|
|
'bulk_assign_cluster',
|
|
'bulk_set_status_active',
|
|
'bulk_set_status_inactive',
|
|
]
|
|
|
|
def get_site_display(self, obj):
|
|
"""Safely get site name"""
|
|
try:
|
|
return obj.site.name if obj.site else '-'
|
|
except:
|
|
return '-'
|
|
get_site_display.short_description = 'Site'
|
|
|
|
def get_sector_display(self, obj):
|
|
"""Safely get sector name"""
|
|
try:
|
|
return obj.sector.name if obj.sector else '-'
|
|
except:
|
|
return '-'
|
|
|
|
def get_cluster_display(self, obj):
|
|
"""Safely get cluster name"""
|
|
try:
|
|
return obj.cluster.name if obj.cluster else '-'
|
|
except:
|
|
return '-'
|
|
get_cluster_display.short_description = 'Cluster'
|
|
|
|
def bulk_assign_cluster(self, request, queryset):
|
|
"""Assign selected keywords to a cluster"""
|
|
from django import forms
|
|
|
|
# If this is the POST request with cluster selection
|
|
if 'apply' in request.POST:
|
|
cluster_id = request.POST.get('cluster')
|
|
if cluster_id:
|
|
cluster = Clusters.objects.get(pk=cluster_id)
|
|
updated = queryset.update(cluster=cluster)
|
|
self.message_user(request, f'{updated} keyword(s) assigned to cluster: {cluster.name}', messages.SUCCESS)
|
|
return
|
|
|
|
# Get first keyword's site/sector for filtering clusters
|
|
first_keyword = queryset.first()
|
|
if first_keyword:
|
|
clusters = Clusters.objects.filter(site=first_keyword.site, sector=first_keyword.sector)
|
|
else:
|
|
clusters = Clusters.objects.all()
|
|
|
|
# Create form for cluster selection
|
|
class ClusterForm(forms.Form):
|
|
cluster = forms.ModelChoiceField(
|
|
queryset=clusters,
|
|
label="Select Cluster",
|
|
help_text=f"Assign {queryset.count()} selected keyword(s) to:"
|
|
)
|
|
|
|
if clusters.exists():
|
|
from django.shortcuts import render
|
|
return render(request, 'admin/bulk_action_form.html', {
|
|
'title': 'Assign Keywords to Cluster',
|
|
'queryset': queryset,
|
|
'form': ClusterForm(),
|
|
'action': 'bulk_assign_cluster',
|
|
})
|
|
else:
|
|
self.message_user(request, 'No clusters available for the selected keywords.', messages.WARNING)
|
|
bulk_assign_cluster.short_description = 'Assign to Cluster'
|
|
|
|
def bulk_set_status_active(self, request, queryset):
|
|
"""Set selected keywords to active status"""
|
|
updated = queryset.update(status='active')
|
|
self.message_user(request, f'{updated} keyword(s) set to active.', messages.SUCCESS)
|
|
bulk_set_status_active.short_description = 'Set status to Active'
|
|
|
|
def bulk_set_status_inactive(self, request, queryset):
|
|
"""Set selected keywords to inactive status"""
|
|
updated = queryset.update(status='inactive')
|
|
self.message_user(request, f'{updated} keyword(s) set to inactive.', messages.SUCCESS)
|
|
bulk_set_status_inactive.short_description = 'Set status to Inactive'
|
|
|
|
|
|
@admin.register(ContentIdeas)
|
|
class ContentIdeasAdmin(SiteSectorAdminMixin, ModelAdmin):
|
|
list_display = ['idea_title', 'site', 'sector', 'description_preview', 'content_type', 'content_structure', 'status', 'keyword_cluster', 'estimated_word_count', 'created_at']
|
|
list_filter = [
|
|
('status', ChoicesDropdownFilter),
|
|
('content_type', ChoicesDropdownFilter),
|
|
('content_structure', ChoicesDropdownFilter),
|
|
('site', RelatedDropdownFilter),
|
|
('sector', RelatedDropdownFilter),
|
|
('keyword_cluster', RelatedDropdownFilter),
|
|
('estimated_word_count', RangeNumericFilter),
|
|
('created_at', RangeDateFilter),
|
|
]
|
|
search_fields = ['idea_title', 'target_keywords', 'description']
|
|
ordering = ['-created_at']
|
|
readonly_fields = ['created_at', 'updated_at']
|
|
|
|
fieldsets = (
|
|
('Basic Info', {
|
|
'fields': ('idea_title', 'description', 'status', 'site', 'sector')
|
|
}),
|
|
('Content Planning', {
|
|
'fields': ('content_type', 'content_structure', 'estimated_word_count')
|
|
}),
|
|
('Keywords & Clustering', {
|
|
'fields': ('keyword_cluster', 'target_keywords', 'taxonomy')
|
|
}),
|
|
('Timestamps', {
|
|
'fields': ('created_at', 'updated_at'),
|
|
'classes': ('collapse',)
|
|
}),
|
|
)
|
|
|
|
def description_preview(self, obj):
|
|
"""Show a truncated preview of the description"""
|
|
if not obj.description:
|
|
return '-'
|
|
# Truncate to 100 characters
|
|
preview = obj.description[:100]
|
|
if len(obj.description) > 100:
|
|
preview += '...'
|
|
return preview
|
|
description_preview.short_description = 'Description'
|
|
|
|
def get_site_display(self, obj):
|
|
"""Safely get site name"""
|
|
try:
|
|
return obj.site.name if obj.site else '-'
|
|
except:
|
|
return '-'
|
|
get_site_display.short_description = 'Site'
|
|
|
|
def get_sector_display(self, obj):
|
|
"""Safely get sector name"""
|
|
try:
|
|
return obj.sector.name if obj.sector else '-'
|
|
except:
|
|
return '-'
|
|
|
|
def get_keyword_cluster_display(self, obj):
|
|
"""Safely get cluster name"""
|
|
try:
|
|
return obj.keyword_cluster.name if obj.keyword_cluster else '-'
|
|
except:
|
|
return '-'
|
|
get_keyword_cluster_display.short_description = 'Cluster'
|
|
|