calc top 10

journal/management/commands/top10.py

@@ -0,0 +1,68 @@
+from django.core.management.base import BaseCommand
+from django.db.models import Count, F, Q
+
+from catalog.models import *
+from journal.models import *
+from users.models import *
+from users.models import APIdentity
+
+
+class Command(BaseCommand):
+    help = "Calculate Top 10"
+
+    def add_arguments(self, parser):
+        parser.add_argument("year", type=int, help="year to calculate")
+        parser.add_argument("--save", default="", help="save as collection for user")
+        parser.add_argument(
+            "--top", default=10, type=int, help="save as collection for user"
+        )
+
+    def handle(self, year: int, top: int, save: str, **options):  # type: ignore
+        collector = APIdentity.objects.get(username=save, local=True) if save else None
+        year = 2023
+        types = [
+            [Edition],
+            [Movie],
+            [TVShow, TVSeason],
+            [Game],
+            [Podcast],
+            [Album],
+            [Performance],
+        ]
+        mapping = item_content_types()
+        for typ in types:
+            cids = [mapping[t] for t in typ]
+            title = f"{year}年标记最多的{typ[0].category.label}"
+            print(title)
+            top10 = list(
+                ShelfMember.objects.filter(
+                    created_time__year=year, item__polymorphic_ctype__in=cids
+                )
+                .values("item")
+                .annotate(c=Count("item"))
+                .order_by("-c")[:top]
+            )
+            print(options)
+            items = [Item.objects.get(pk=i["item"]) for i in top10]
+            _ = [print(i.title, i.absolute_url) for i in items]
+            if collector:
+                print(f"Saving to {collector}")
+                c, _ = Collection.objects.get_or_create(
+                    owner=collector,
+                    title=title,
+                    brief="*根据用户标记数统计*",
+                    defaults={"visibility": 2},
+                )
+                for i in items:
+                    c.append_item(i)
+
+        # top10 = list(
+        #     Comment.objects.filter(
+        #         created_time__year=2023, item__polymorphic_ctype=mapping[PodcastEpisode]
+        #     )
+        #     .values(i=F("item__podcastepisode__program_id"))
+        #     .annotate(c=Count("i"))
+        #     .order_by("-c")[:15]
+        # )
+        # items = [Item.objects.get(pk=i["i"]) for i in top10]
+        # _ = [print(i.title, i.absolute_url) for i in items]