137 lines
5 KiB
Python
137 lines
5 KiB
Python
import pprint
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from django.db.models import Count, F
|
|
|
|
from catalog.models import *
|
|
from journal.models import update_journal_for_merged_item
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "catalog app utilities"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
"--verbose",
|
|
action="store_true",
|
|
)
|
|
parser.add_argument(
|
|
"--fix",
|
|
action="store_true",
|
|
)
|
|
parser.add_argument(
|
|
"--purge",
|
|
action="store_true",
|
|
help="purge deleted items",
|
|
)
|
|
parser.add_argument(
|
|
"--integrity",
|
|
action="store_true",
|
|
help="check and fix integrity for merged and deleted items",
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
self.verbose = options["verbose"]
|
|
self.fix = options["fix"]
|
|
if options["purge"]:
|
|
self.purge()
|
|
if options["integrity"]:
|
|
self.integrity()
|
|
self.stdout.write(self.style.SUCCESS(f"Done."))
|
|
|
|
def purge(self):
|
|
for cls in Item.__subclasses__():
|
|
if self.fix:
|
|
self.stdout.write(f"Cleaning up {cls}...")
|
|
cls.objects.filter(is_deleted=True).delete()
|
|
|
|
def integrity(self):
|
|
self.stdout.write(f"Checking circulated merge...")
|
|
for i in Item.objects.filter(merged_to_item=F("id")):
|
|
self.stdout.write(f"! {i} : {i.absolute_url}?skipcheck=1")
|
|
if self.fix:
|
|
i.merged_to_item = None
|
|
i.save()
|
|
|
|
self.stdout.write(f"Checking chained merge...")
|
|
for i in (
|
|
Item.objects.filter(merged_to_item__isnull=False)
|
|
.annotate(n=Count("merged_from_items"))
|
|
.exclude(n=0)
|
|
):
|
|
self.stdout.write(f"! {i} : {i.absolute_url}?skipcheck=1")
|
|
if self.fix:
|
|
for j in i.merged_from_items.all():
|
|
j.merged_to_item = i.merged_to_item
|
|
j.save()
|
|
|
|
self.stdout.write(f"Checking deleted merge...")
|
|
for i in Item.objects.filter(merged_to_item__isnull=False, is_deleted=True):
|
|
self.stdout.write(f"! {i} : {i.absolute_url}?skipcheck=1")
|
|
if self.fix:
|
|
i.is_deleted = False
|
|
i.save()
|
|
|
|
self.stdout.write(f"Checking deleted item with external resources...")
|
|
for i in (
|
|
Item.objects.filter(is_deleted=True)
|
|
.annotate(n=Count("external_resources"))
|
|
.exclude(n=0)
|
|
):
|
|
self.stdout.write(f"! {i} : {i.absolute_url}?skipcheck=1")
|
|
if self.fix:
|
|
for r in i.external_resources.all():
|
|
r.item = None
|
|
r.save()
|
|
|
|
self.stdout.write(f"Checking merged item with external resources...")
|
|
for i in (
|
|
Item.objects.filter(merged_to_item__isnull=False)
|
|
.annotate(n=Count("external_resources"))
|
|
.exclude(n=0)
|
|
):
|
|
self.stdout.write(f"! {i} : {i.absolute_url}?skipcheck=1")
|
|
if self.fix:
|
|
for r in i.external_resources.all():
|
|
r.item = i.merged_to_item
|
|
r.save()
|
|
|
|
tvshow_ct_id = ContentType.objects.get_for_model(TVShow).id
|
|
self.stdout.write(f"Checking TVShow merged to other class...")
|
|
for i in (
|
|
TVShow.objects.filter(merged_to_item__isnull=False)
|
|
.filter(merged_to_item__isnull=False)
|
|
.exclude(merged_to_item__polymorphic_ctype_id=tvshow_ct_id)
|
|
):
|
|
if i.child_items.all().exists():
|
|
self.stdout.write(f"! with season {i} : {i.absolute_url}?skipcheck=1")
|
|
if self.fix:
|
|
i.merged_to_item = None
|
|
i.save()
|
|
else:
|
|
self.stdout.write(f"! no season {i} : {i.absolute_url}?skipcheck=1")
|
|
if self.fix:
|
|
i.recast_to(i.merged_to_item.__class__)
|
|
|
|
self.stdout.write(f"Checking TVSeason is child of other class...")
|
|
for i in TVSeason.objects.filter(show__isnull=False).exclude(
|
|
show__polymorphic_ctype_id=tvshow_ct_id
|
|
):
|
|
self.stdout.write(f"! {i.show} : {i.show.absolute_url}?skipcheck=1")
|
|
if self.fix:
|
|
i.show = None
|
|
i.save()
|
|
|
|
self.stdout.write(f"Checking deleted item with child TV Season...")
|
|
for i in TVSeason.objects.filter(show__is_deleted=True):
|
|
self.stdout.write(f"! {i.show} : {i.show.absolute_url}?skipcheck=1")
|
|
if self.fix:
|
|
i.show.is_deleted = False
|
|
i.show.save()
|
|
|
|
self.stdout.write(f"Checking merged item with child TV Season...")
|
|
for i in TVSeason.objects.filter(show__merged_to_item__isnull=False):
|
|
self.stdout.write(f"! {i.show} : {i.show.absolute_url}?skipcheck=1")
|
|
if self.fix:
|
|
i.show = i.show.merged_to_item
|
|
i.save()
|