Add some routes for group

Author: andyzhouty

flemi/__init__.py

@@ -51,7 +51,7 @@ def register_extensions(app: Flask) -> None:
 
 
 def register_blueprints(app: Flask) -> None:
-    for mod_name in ("auth", "me", "post", "user"):
+    for mod_name in ("auth", "me", "post", "user", "group"):
         mod = il.import_module(f".api.{mod_name}.views", "flemi")
         blueprint = getattr(mod, f"{mod_name}_bp")
         CORS(blueprint)

flemi/api/group/__init__.py

@@ -0,0 +1,21 @@
+from apiflask import Schema
+from apiflask.fields import Boolean
+from apiflask.fields import Integer
+from apiflask.fields import List
+from apiflask.fields import Nested
+from apiflask.fields import String
+
+from ..user.schemas import PublicUserOutSchema
+
+
+class GroupInSchema(Schema):
+    name = String(required=True)
+    members = List(Integer())
+    private = Boolean()
+
+
+class GroupOutSchema(Schema):
+    name = String()
+    members = List(Nested(PublicUserOutSchema))
+    manager = Nested(PublicUserOutSchema)
+    private = Boolean()

flemi/api/group/schemas.py

@@ -0,0 +1,34 @@
+from apiflask import APIBlueprint
+from flask import request
+
+from ...models import Group
+from ..auth.views import auth
+from ..auth.views import verify_token
+from .schemas import GroupOutSchema
+
+group_bp = APIBlueprint("group", __name__, url_prefix="/group")
+
+
+@group_bp.get("/all")
+@group_bp.output(GroupOutSchema(many=True))
+def all():
+    token = request.headers.get("Authorization", "")
+    public_groups = Group.query.filter_by(private=False).all()
+
+    try:
+        token = token[7:]
+        user = verify_token(token)
+    except IndexError:
+        return public_groups
+
+    if user is None:
+        return public_groups
+    else:
+        return [g for g in Group.query.all() if not g.private or user in g.members]
+
+
+@group_bp.get("/me")
+@group_bp.auth_required(auth)
+@group_bp.output(GroupOutSchema(many=True))
+def me():
+    return auth.current_user.groups

flemi/api/group/views.py

@@ -6,7 +6,7 @@
 from marshmallow.fields import Url
 
 
-class PrivateUserOutputSchema(Schema):
+class PrivateUserOutSchema(Schema):
     avatar_url = Function(lambda obj: obj.avatar_url())
 
     class Meta:

flemi/api/me/schemas.py

@@ -6,15 +6,15 @@
 from .schemas import AboutEditSchema
 from .schemas import AvatarEditSchema
 from .schemas import BasicProfileEditSchema
-from .schemas import PrivateUserOutputSchema
+from .schemas import PrivateUserOutSchema
 
 
 me_bp = APIBlueprint("me", __name__, url_prefix="/me")
 
 
 @me_bp.get("")
 @me_bp.auth_required(auth)
-@me_bp.output(PrivateUserOutputSchema)
+@me_bp.output(PrivateUserOutSchema)
 def self_profile():
     """
     profile of the current user

flemi/api/me/views.py

@@ -0,0 +1,31 @@
+from flemi.models import db, Group
+from .base import Base
+
+
+class GroupTestCase(Base):
+    def setUp(self) -> None:
+        super().setUp()
+        self.set_headers()
+
+    def test_all_groups(self):
+        g = Group(name="test")
+        g2 = Group(name="test2", members=[self.test_user], private=True)
+        db.session.add(g)
+        db.session.add(g2)
+        db.session.commit()
+
+        resp = self.client.get(
+            "/group/all",
+        )
+        data = resp.get_json()
+        self.assertEqual(resp.status_code, 200)
+        self.assertEqual(data[0]["name"], "test")
+        self.assertTrue(len(data) == 1)
+
+        resp = self.client.get(
+            "/group/all",
+            headers=self.headers
+        )
+        data = resp.get_json()
+        print(len(data))
+        self.assertTrue(len(data) == 2)