feat: teams v1

forms_pro/api/team.py

@@ -0,0 +1,19 @@
+import frappe
+
+from forms_pro.utils.teams import GetTeamFormsResponseSchema
+from forms_pro.utils.teams import get_team_forms as get_team_forms_utils
+
+
+@frappe.whitelist()
+def get_team_forms(team_id: str) -> list[GetTeamFormsResponseSchema]:
+    """
+    Get the list of forms for the current team
+
+    Args:
+        team_id: ID of the team
+
+    Returns:
+        list[GetTeamFormsResponseSchema] - List of forms for the team
+    """
+    forms = get_team_forms_utils(team_id=team_id)
+    return forms

forms_pro/api/user.py

@@ -0,0 +1,61 @@
+import frappe
+from frappe.core.doctype.has_role.has_role import HasRole
+from pydantic import BaseModel, Field, field_validator
+
+from forms_pro.utils.teams import get_user_teams as get_user_teams_utils
+
+
+class GetUserTeamsResponseSchema(BaseModel):
+    name: str = Field(description="ID of the team")
+    team_name: str = Field(description="The name of the team")
+    is_current: bool = Field(description="Whether this is the current team")
+
+
+class GetUserResponseSchema(BaseModel):
+    email: str
+    first_name: str
+    last_name: str | None = None
+    full_name: str
+    username: str
+    desk_theme: str
+    roles: list[str]
+    has_desk_access: bool
+
+    @field_validator("roles", mode="before")
+    @classmethod
+    def extract_roles(cls, v: list[HasRole]) -> list[str]:
+        if not v:
+            return []
+
+        return [role.role for role in v]
+
+
+@frappe.whitelist()
+def get_user() -> GetUserResponseSchema:
+    """
+    Get Current User Data
+    """
+
+    user_id = frappe.session.user
+    user_doc = frappe.get_doc("User", user_id)
+    data = user_doc.as_dict()
+    data["roles"] = user_doc.get("roles")
+    data["has_desk_access"] = user_doc.has_desk_access()
+
+    return GetUserResponseSchema.model_validate(data).model_dump()
+
+
+@frappe.whitelist()
+def get_user_teams() -> list[GetUserTeamsResponseSchema]:
+    """
+    Get the list of teams for the current user
+    """
+
+    user = frappe.session.user
+
+    if user == "Guest":
+        return []
+
+    teams = get_user_teams_utils(user)
+
+    return [GetUserTeamsResponseSchema.model_validate(team).model_dump() for team in teams]

forms_pro/forms_pro/doctype/form/form.json

@@ -11,6 +11,7 @@
   "column_break_pqae",
   "title",
   "linked_doctype",
+  "linked_team_id",
   "settings_section",
   "login_required",
   "column_break_xocw",
@@ -107,12 +108,19 @@
    "fieldname": "allow_incomplete",
    "fieldtype": "Check",
    "label": "Allow Incomplete Forms"
+  },
+  {
+   "fieldname": "linked_team_id",
+   "fieldtype": "Link",
+   "label": "Linked Team",
+   "options": "FP Team",
+   "reqd": 1
   }
  ],
  "grid_page_length": 50,
  "index_web_pages_for_search": 1,
  "links": [],
- "modified": "2025-10-07 12:16:18.825427",
+ "modified": "2025-12-20 20:40:24.219032",
  "modified_by": "Administrator",
  "module": "Forms Pro",
  "name": "Form",

forms_pro/forms_pro/doctype/form/form.py

@@ -21,6 +21,7 @@ class Form(Document):
         fields: DF.Table[FormField]
         is_published: DF.Check
         linked_doctype: DF.Link
+        linked_team_id: DF.Link
         login_required: DF.Check
         metadata: DF.Code | None
         route: DF.Data | None

forms_pro/forms_pro/doctype/form/test_form.py

@@ -4,6 +4,8 @@
 import frappe
 from frappe.tests import IntegrationTestCase
 
+from forms_pro.utils.teams import get_user_teams
+
 # On IntegrationTestCase, the doctype test records and all
 # link-field test record dependencies are recursively loaded
 # Use these module variables to add/remove to/from that list
@@ -18,9 +20,18 @@ def setUp(self):
         self.test_doctype_name = "Test Form DocType"
         self.create_test_doctype()
 
+        self.test_user = "test_forms_pro_user@example.com"
+        self.test_team = get_user_teams(self.test_user)[0]["name"]
+
         # Create a test Form
         self.test_form = frappe.get_doc(
-            {"doctype": "Form", "title": "Test Form", "linked_doctype": self.test_doctype_name, "fields": []}
+            {
+                "doctype": "Form",
+                "title": "Test Form",
+                "linked_doctype": self.test_doctype_name,
+                "fields": [],
+                "linked_team_id": self.test_team,
+            }
         )
         self.test_form.insert()
 

forms_pro/forms_pro/doctype/fp_team/fp_team.py

@@ -20,4 +20,24 @@ class FPTeam(Document):
         users: DF.TableMultiSelect[FPTeamMember]
     # end: auto-generated types
 
-    pass
+    @property
+    def team_members(self) -> list[str]:
+        """
+        Get the list of team members
+
+        Returns:
+            list[str] - List of team member email addresses
+        """
+        return [member.user for member in self.users] if self.users else []
+
+    def is_team_member(self, user: str) -> bool:
+        """
+        Check if a user is a member of the team
+
+        Args:
+            user: The user email address
+
+        Returns:
+            bool - True if the user is a member of the team, False otherwise
+        """
+        return user in self.team_members

forms_pro/hooks.py

@@ -166,7 +166,7 @@
 # Testing
 # -------
 
-# before_tests = "forms_pro.install.before_tests"
+before_tests = "forms_pro.install.before_tests"
 
 # Overriding Methods
 # ------------------------------

forms_pro/install.py

@@ -0,0 +1,30 @@
+import frappe
+from frappe.core.doctype.user.user import User
+
+from forms_pro.roles import FORMS_PRO_ROLE
+
+
+def before_tests():
+    give_admin_forms_pro_role()
+    create_test_user()
+
+
+def give_admin_forms_pro_role():
+    admin = frappe.get_doc("User", "Administrator")
+    admin.append("roles", {"role": FORMS_PRO_ROLE})
+    admin.save()
+
+
+def create_test_user():
+    test_user = "test_forms_pro_user@example.com"
+
+    if frappe.db.exists("User", test_user):
+        return
+
+    user: User = frappe.new_doc("User")
+    user.email = test_user
+    user.first_name = "Test"
+    user.last_name = "Forms Pro User"
+    user.insert(ignore_permissions=True)
+    user.append("roles", {"role": FORMS_PRO_ROLE})
+    user.save(ignore_permissions=True)

forms_pro/overrides/roles.py

@@ -2,7 +2,7 @@
 from frappe.core.doctype.user.user import User
 
 from forms_pro.roles import FORMS_PRO_ROLE
-from forms_pro.utils.teams import get_user_teams
+from forms_pro.utils.teams import get_user_teams, set_current_team
 
 
 def handle_forms_pro_role_change(doc, method) -> None:
@@ -27,7 +27,7 @@ def handle_forms_pro_role_change(doc, method) -> None:
             create_default_team_for_user(user)
 
 
-def create_default_team_for_user(user: User):
+def create_default_team_for_user(user: User) -> None:
     from forms_pro.forms_pro.doctype.fp_team.fp_team import FPTeam
 
     team: FPTeam = frappe.new_doc("FP Team")
@@ -40,3 +40,4 @@ def create_default_team_for_user(user: User):
         },
     )
     team.save(ignore_permissions=True)
+    set_current_team(team.name, user.name)

forms_pro/utils/form_generator.py

@@ -4,12 +4,12 @@
 
 
 @frappe.whitelist()
-def create_form_with_doctype(doctype: str):
+def create_form_with_doctype(team_id: str, doctype: str):
     roles = frappe.get_roles(frappe.session.user)
     if FORMS_PRO_ROLE not in roles:
         frappe.throw("You are not authorized to create a form")
 
-    form_generator = FormGenerator(linked_doctype=doctype)
+    form_generator = FormGenerator(team_id=team_id, linked_doctype=doctype)
     form_generator.generate()
 
     return {
@@ -19,12 +19,12 @@ def create_form_with_doctype(doctype: str):
 
 
 @frappe.whitelist()
-def create_form():
+def create_form(team_id: str):
     roles = frappe.get_roles(frappe.session.user)
     if FORMS_PRO_ROLE not in roles:
         frappe.throw("You are not authorized to create a form")
 
-    form_generator = FormGenerator()
+    form_generator = FormGenerator(team_id=team_id)
     form_generator.generate()
 
     return {
@@ -34,8 +34,13 @@ def create_form():
 
 
 class FormGenerator:
-    def __init__(self, linked_doctype: str | None = None) -> None:
+    def __init__(
+        self,
+        team_id: str,
+        linked_doctype: str | None = None,
+    ) -> None:
         self.doctype = None
+        self.team_id = team_id
         if linked_doctype:
             self.doctype = frappe.get_doc("DocType", linked_doctype)
 
@@ -80,6 +85,7 @@ def _initialize_form_document(self) -> None:
         form_document = frappe.new_doc("Form")
         form_document.linked_doctype = self.doctype.name
         form_document.title = "Untitled Form"
+        form_document.linked_team_id = self.team_id
         form_document.insert(ignore_permissions=True)
         self.form_document = form_document