artemis-model 0.1.133__py3-none-any.whl → 0.1.135__py3-none-any.whl

artemis_model/category.py

@@ -1,4 +1,4 @@
-from sqlalchemy import ARRAY, Computed, String, func, literal, text
+from sqlalchemy import Computed, func, literal, text
 from sqlalchemy.ext.associationproxy import AssociationProxy, association_proxy
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
@@ -16,12 +16,12 @@ def to_tsvector_ix(*columns):
 
 class CategoryMixin(TimeStampMixin):
     id: Mapped[int] = mapped_column(primary_key=True)
-    name: Mapped[str] = mapped_column(nullable=False)
-    sub_categories: Mapped[list[str]] = mapped_column(ARRAY(String), default=[])
+    main_category: Mapped[str] = mapped_column(nullable=False)
+    sub_category: Mapped[str] = mapped_column(nullable=False)
 
     name_tsv = mapped_column(
         TSVector(),
-        Computed("to_tsvector('english', name)", persisted=True),
+        Computed("to_tsvector('english', main_category || ' ' || sub_category)", persisted=True),
     )
 
     @declared_attr
@@ -39,18 +39,11 @@ class CategoryMixin(TimeStampMixin):
     # __table_args__ = (
     #     Index(
     #         "fts_ix_category_name_tsv",
-    #         to_tsvector_ix("name"),
+    #         to_tsvector_ix("main_category", "sub_category"),
     #         postgresql_using="gin",
     #     ),
     # )
 
-    @property
-    def is_subcategory(self) -> bool:
-        if self.sub_categories:
-            return True
-
-        return False
-
 
 class CategorySync(CustomSyncBase, CategoryMixin):
     pass

artemis_model/playlist.py

@@ -1,7 +1,7 @@
 import uuid
 from datetime import date, datetime
 
-from sqlalchemy import Computed, ForeignKey, func, literal, text
+from sqlalchemy import Computed, ForeignKey, func, literal, text, ARRAY, String
 from sqlalchemy.ext.hybrid import hybrid_property
 from sqlalchemy.ext.associationproxy import AssociationProxy, association_proxy
 from sqlalchemy.orm import Mapped, mapped_column, relationship
@@ -43,6 +43,8 @@ class PlaylistMixin(TimeStampMixin):
     published_at: Mapped[date | None] = mapped_column(nullable=True)
     is_private: Mapped[bool] = mapped_column(default=False)
     total_duration: Mapped[int] = mapped_column(default=0)
+    track_count: Mapped[int] = mapped_column(default=0)
+    artist_names: Mapped[list[str]] = mapped_column(ARRAY(String), nullable=False, default=[])
 
     name_tsv = mapped_column(
         TSVector(),

{artemis_model-0.1.133.dist-info → artemis_model-0.1.135.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: artemis-model
-Version: 0.1.133
+Version: 0.1.135
 Summary: 
 Author: Jukeboxy
 Requires-Python: >=3.10.6,<4.0.0

{artemis_model-0.1.133.dist-info → artemis_model-0.1.135.dist-info}/RECORD

@@ -4,7 +4,7 @@ artemis_model/artist.py,sha256=vjXlFN2mOEidAzUzxmsAP5GnVTvJUuhFdhQaH214lgw,1563
 artemis_model/auth.py,sha256=0Yg_6N1nj4U31F86u1PE0leIa3D64QlYrsDlavulN6s,4033
 artemis_model/banned_tracks.py,sha256=uU-F-6DL2EshPAUwLTTHjYZ7UEz4vm0Wfcif2F0lSKw,664
 artemis_model/base.py,sha256=zC20m8a1Sa11oEY0ay1iIRQeybGYvXY5p3Vr-pxoKnQ,6802
-artemis_model/category.py,sha256=jhCZrK-LxPmc0pXmVifAZWdCENssBTzdTmC3nqqZc6s,1761
+artemis_model/category.py,sha256=ERZC8YBTtmF72ykSiVEtf_Ws3mPuN28ECfTxWF7H8tE,1662
 artemis_model/dj_set.py,sha256=fOYnCu4n5TiqyiSojfdFnO7LuPe_mM2SUwBV5xHy2Kc,3782
 artemis_model/genre.py,sha256=8_-IuJS543wIhUVCES3bdrDpKPKx-plDuBKGBcoMIbc,1570
 artemis_model/location.py,sha256=-ovGxYAajdQxNoQvafUBynSgcrdBklynXM7Qe7gmY5c,4645
@@ -12,7 +12,7 @@ artemis_model/message.py,sha256=W4vhllsD4Nn11JIKeXlgsKC2NWCt3UMkWh-Sma71gBI,3325
 artemis_model/organization.py,sha256=yjGHJLTOtb9PUfxSPal782Rceb8xb8vftXX67tHRNqc,2351
 artemis_model/otp.py,sha256=guIRGtyFlHUBthCAEsTh5_Hs-1yiGN_qfEO4uHNcv4s,1017
 artemis_model/permission.py,sha256=Bn1Bg1aCS4Z4_3tqEqvtrzqAYDCImsvmGyIEMoVycEk,1452
-artemis_model/playlist.py,sha256=L22KSJgTZmNEc9CqM3TFD4Wnx3TW77GQJ5EFLJCxZ20,6393
+artemis_model/playlist.py,sha256=O60zlP1LrDkzNrrfNA7TD_kotd5A9i8PuFhwonXFhN0,6559
 artemis_model/redis/__init__.py,sha256=CEQ4Vy3lwSlW_Q66nFLGevJr3nRv8Ema59PG0pG6q9I,944
 artemis_model/redis/bucket.py,sha256=DlmIf6GxfKq9CzcXmMx5IcviaqOTvuWwIb6lAVcZQGs,1537
 artemis_model/redis/device.py,sha256=ZUiIyVmtWiEwayFIg9mktiFaVuHQ7Fg0rvSjjWyryCI,484
@@ -27,6 +27,6 @@ artemis_model/track.py,sha256=QwUF0QKVn1I64648B-NI75-IzGQvnLt9B0emD4GnS6E,3757
 artemis_model/user.py,sha256=eqIdCiBJRNLjCwPPCn-gQ6si0O5JUBGfp9oWJL5zVW4,2131
 artemis_model/zone.py,sha256=iGRUtzUwKh9LHT3MOfzzg1DnkPBts_ZBzZVTi2EmIgs,2282
 artemis_model/zone_activity.py,sha256=BY4iODavY9ceJ5oRChdjjxf26S3U30Yb7Pxm5YRFpCo,1590
-artemis_model-0.1.133.dist-info/METADATA,sha256=ouIOKMFLGpN1YJ3ZotW2ndqO-U3cYY8kAigu1p2Ibv8,3445
-artemis_model-0.1.133.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-artemis_model-0.1.133.dist-info/RECORD,,
+artemis_model-0.1.135.dist-info/METADATA,sha256=kcyxO1oNUACB9TMb7DzxvC71iefPlzvO7RobMqAQItE,3445
+artemis_model-0.1.135.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+artemis_model-0.1.135.dist-info/RECORD,,