You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
3.0 KiB
61 lines
3.0 KiB
from django.core.management.base import BaseCommand
|
|
from django.db import transaction
|
|
from apps.video.models import VideoCategory, VideoCollection, VideoPlaylist, PlaylistItem
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Delete all data from VideoCategory, VideoCollection, and VideoPlaylist (keeps Video model data)'
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
'--confirm',
|
|
action='store_true',
|
|
help='Confirm deletion without prompting'
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
confirm = options.get('confirm', False)
|
|
|
|
# Count current data
|
|
category_count = VideoCategory.objects.count()
|
|
collection_count = VideoCollection.objects.count()
|
|
playlist_count = VideoPlaylist.objects.count()
|
|
playlist_item_count = PlaylistItem.objects.count()
|
|
|
|
self.stdout.write(self.style.WARNING('\n=== Current Data Count ==='))
|
|
self.stdout.write(f'VideoCategory: {category_count}')
|
|
self.stdout.write(f'VideoCollection: {collection_count}')
|
|
self.stdout.write(f'VideoPlaylist: {playlist_count}')
|
|
self.stdout.write(f'PlaylistItem: {playlist_item_count}')
|
|
self.stdout.write(self.style.WARNING('\n=== Video Data Will NOT Be Deleted ===\n'))
|
|
|
|
if not confirm:
|
|
user_input = input('Are you sure you want to delete this data? Type "yes" to confirm: ')
|
|
if user_input.lower() != 'yes':
|
|
self.stdout.write(self.style.ERROR('Operation cancelled.'))
|
|
return
|
|
|
|
try:
|
|
with transaction.atomic():
|
|
# Delete in order to respect foreign key constraints
|
|
# 1. Delete PlaylistItem first (references VideoPlaylist)
|
|
deleted_playlist_items = PlaylistItem.objects.all().delete()
|
|
self.stdout.write(self.style.SUCCESS(f'✓ Deleted {deleted_playlist_items[0]} PlaylistItems'))
|
|
|
|
# 2. Delete VideoPlaylist (may reference VideoCategory and VideoCollection through M2M)
|
|
deleted_playlists = VideoPlaylist.objects.all().delete()
|
|
self.stdout.write(self.style.SUCCESS(f'✓ Deleted {deleted_playlists[0]} VideoPlaylists'))
|
|
|
|
# 3. Delete VideoCollection
|
|
deleted_collections = VideoCollection.objects.all().delete()
|
|
self.stdout.write(self.style.SUCCESS(f'✓ Deleted {deleted_collections[0]} VideoCollections'))
|
|
|
|
# 4. Delete VideoCategory
|
|
deleted_categories = VideoCategory.objects.all().delete()
|
|
self.stdout.write(self.style.SUCCESS(f'✓ Deleted {deleted_categories[0]} VideoCategories'))
|
|
|
|
self.stdout.write(self.style.SUCCESS('\n✓ All data deleted successfully!'))
|
|
|
|
except Exception as e:
|
|
self.stdout.write(self.style.ERROR(f'\n✗ Error during deletion: {str(e)}'))
|
|
raise
|