Add remaining association count sort options

Now that we have all association counts mapped to their respective
models we can easily allow sorting on them as well.

1 files changed, 11 insertions, 0 deletions

diff --git a/src/hircine/api/sort.py b/src/hircine/api/sort.py
index 17043a6..a4ccaf1 100644
--- a/src/hircine/api/sort.py
+++ b/src/hircine/api/sort.py
@@ -22,6 +22,10 @@ class ComicSort(enum.Enum):
     DATE = strawberry.enum_value(models.Comic.date)
     CREATED_AT = strawberry.enum_value(models.Comic.created_at)
     UPDATED_AT = strawberry.enum_value(models.Comic.updated_at)
+    ARTIST_COUNT = strawberry.enum_value(models.Comic.artist_count)
+    CHARACTER_COUNT = strawberry.enum_value(models.Comic.character_count)
+    CIRCLE_COUNT = strawberry.enum_value(models.Comic.circle_count)
+    WORLD_COUNT = strawberry.enum_value(models.Comic.world_count)
     TAG_COUNT = strawberry.enum_value(models.Comic.tag_count)
     PAGE_COUNT = strawberry.enum_value(models.Comic.page_count)
     RANDOM = "Random"
@@ -41,6 +45,7 @@ class ArtistSort(enum.Enum):
     NAME = strawberry.enum_value(models.Artist.name)
     CREATED_AT = strawberry.enum_value(models.Artist.created_at)
     UPDATED_AT = strawberry.enum_value(models.Artist.updated_at)
+    COMIC_COUNT = strawberry.enum_value(models.Artist.comic_count)
     RANDOM = "Random"
 
 
@@ -49,6 +54,7 @@ class CharacterSort(enum.Enum):
     NAME = strawberry.enum_value(models.Character.name)
     CREATED_AT = strawberry.enum_value(models.Character.created_at)
     UPDATED_AT = strawberry.enum_value(models.Character.updated_at)
+    COMIC_COUNT = strawberry.enum_value(models.Character.comic_count)
     RANDOM = "Random"
 
 
@@ -57,6 +63,7 @@ class CircleSort(enum.Enum):
     NAME = strawberry.enum_value(models.Circle.name)
     CREATED_AT = strawberry.enum_value(models.Circle.created_at)
     UPDATED_AT = strawberry.enum_value(models.Circle.updated_at)
+    COMIC_COUNT = strawberry.enum_value(models.Circle.comic_count)
     RANDOM = "Random"
 
 
@@ -66,6 +73,7 @@ class NamespaceSort(enum.Enum):
     NAME = strawberry.enum_value(models.Namespace.name)
     CREATED_AT = strawberry.enum_value(models.Namespace.created_at)
     UPDATED_AT = strawberry.enum_value(models.Namespace.updated_at)
+    TAG_COUNT = strawberry.enum_value(models.Namespace.tag_count)
     RANDOM = "Random"
 
 
@@ -74,6 +82,8 @@ class TagSort(enum.Enum):
     NAME = strawberry.enum_value(models.Tag.name)
     CREATED_AT = strawberry.enum_value(models.Tag.created_at)
     UPDATED_AT = strawberry.enum_value(models.Tag.updated_at)
+    COMIC_COUNT = strawberry.enum_value(models.Tag.comic_count)
+    NAMESPACE_COUNT = strawberry.enum_value(models.Tag.namespace_count)
     RANDOM = "Random"
 
 
@@ -82,6 +92,7 @@ class WorldSort(enum.Enum):
     NAME = strawberry.enum_value(models.World.name)
     CREATED_AT = strawberry.enum_value(models.World.created_at)
     UPDATED_AT = strawberry.enum_value(models.World.updated_at)
+    COMIC_COUNT = strawberry.enum_value(models.World.comic_count)
     RANDOM = "Random"