trick-import

orm/community.py

@@ -1,21 +1,19 @@
 import enum
 import time
 
-from sqlalchemy import ARRAY, Column, ForeignKey, Integer, String, distinct, func
+from sqlalchemy import Column, ForeignKey, Integer, String, Text, distinct, func
 from sqlalchemy.ext.hybrid import hybrid_property
 
 from orm.author import Author
-from orm.shout import Shout
 from services.db import Base
 
 
 class CommunityRole(enum.Enum):
-    AUTHOR = "author"
-    READER = "reader"
-    EDITOR = "editor"
-    CRITIC = "critic"
-    EXPERT = "expert"
-    ARTIST = "artist"
+    READER = "reader"  # can read and comment
+    AUTHOR = "author"  # + can vote and invite collaborators
+    ARTIST = "artist"  # + can be credited as featured artist
+    EXPERT = "expert"  # + can add proof or disproof to shouts, can manage topics
+    EDITOR = "editor"  # + can manage topics, comments and community settings
 
     @classmethod
     def as_string_array(cls, roles):
@@ -28,7 +26,7 @@ class CommunityFollower(Base):
     author = Column(ForeignKey("author.id"), primary_key=True)
     community = Column(ForeignKey("community.id"), primary_key=True)
     joined_at = Column(Integer, nullable=False, default=lambda: int(time.time()))
-    roles = Column(ARRAY(String), nullable=False, default=lambda: CommunityRole.as_string_array([CommunityRole.READER]))
+    roles = Column(Text, nullable=True, comment="Roles (comma-separated)")
 
     def set_roles(self, roles):
         self.roles = CommunityRole.as_string_array(roles)
@@ -51,6 +49,14 @@ class Community(Base):
     def stat(self):
         return CommunityStats(self)
 
+    @property
+    def role_list(self):
+        return self.roles.split(",") if self.roles else []
+
+    @role_list.setter
+    def role_list(self, value):
+        self.roles = ",".join(value) if value else None
+
 
 class CommunityStats:
     def __init__(self, community):
@@ -58,11 +64,9 @@ class CommunityStats:
 
     @property
     def shouts(self):
-        return (
-            self.community.session.query(func.count(Shout.id))
-            .filter(Shout.community == self.community.id)
-            .scalar()
-        )
+        from orm.shout import Shout
+
+        return self.community.session.query(func.count(Shout.id)).filter(Shout.community == self.community.id).scalar()
 
     @property
     def followers(self):
@@ -74,12 +78,29 @@ class CommunityStats:
 
     @property
     def authors(self):
+        from orm.shout import Shout
+
         # author has a shout with community id and its featured_at is not null
         return (
             self.community.session.query(func.count(distinct(Author.id)))
             .join(Shout)
-            .filter(
-                Shout.community == self.community.id, Shout.featured_at.is_not(None), Author.id.in_(Shout.authors)
-            )
+            .filter(Shout.community == self.community.id, Shout.featured_at.is_not(None), Author.id.in_(Shout.authors))
             .scalar()
         )
+
+
+class CommunityAuthor(Base):
+    __tablename__ = "community_author"
+
+    id = Column(Integer, primary_key=True)
+    community_id = Column(Integer, ForeignKey("community.id"))
+    author_id = Column(Integer, ForeignKey("author.id"))
+    roles = Column(Text, nullable=True, comment="Roles (comma-separated)")
+
+    @property
+    def role_list(self):
+        return self.roles.split(",") if self.roles else []
+
+    @role_list.setter
+    def role_list(self, value):
+        self.roles = ",".join(value) if value else None