pyattackforge 0.2.1__tar.gz → 0.2.2__tar.gz

{pyattackforge-0.2.1 → pyattackforge-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyattackforge
-Version: 0.2.1
+Version: 0.2.2
 Summary: Python SDK for the AttackForge SSAPI
 Author: hackman238
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE

{pyattackforge-0.2.1 → pyattackforge-0.2.2}/pyattackforge/client.py

@@ -11,6 +11,7 @@ from .transport import AttackForgeTransport
 from .resources import (
     AssetsResource,
     ProjectsResource,
+    GroupsResource,
     FindingsResource,
     WriteupsResource,
     TestcasesResource,
@@ -51,6 +52,7 @@ class AttackForgeClient:
 
         self.assets = AssetsResource(self._transport)
         self.projects = ProjectsResource(self._transport)
+        self.groups = GroupsResource(self._transport)
         self.findings = FindingsResource(self._transport)
         self.writeups = WriteupsResource(self._transport)
         self.testcases = TestcasesResource(self._transport)

{pyattackforge-0.2.1 → pyattackforge-0.2.2}/pyattackforge/config.py

@@ -16,7 +16,7 @@ class ClientConfig:
     timeout: float = 30.0
     max_retries: int = 3
     backoff_factor: float = 0.5
-    user_agent: str = "pyattackforge/0.2.1"
+    user_agent: str = "pyattackforge/0.2.2"
     http2: bool = True
     # Default visibility for newly created findings. False = pending/hidden.
     default_findings_visible: bool = False

{pyattackforge-0.2.1 → pyattackforge-0.2.2}/pyattackforge/resources/__init__.py

@@ -2,6 +2,7 @@
 
 from .assets import AssetsResource
 from .projects import ProjectsResource
+from .groups import GroupsResource
 from .findings import FindingsResource
 from .writeups import WriteupsResource
 from .testcases import TestcasesResource
@@ -13,6 +14,7 @@ from .reports import ReportsResource
 __all__ = [
     "AssetsResource",
     "ProjectsResource",
+    "GroupsResource",
     "FindingsResource",
     "WriteupsResource",
     "TestcasesResource",

pyattackforge-0.2.2/pyattackforge/resources/groups.py

@@ -0,0 +1,20 @@
+"""Resource: groups."""
+
+from __future__ import annotations
+
+from typing import Any, Dict, Optional
+
+from .base import BaseResource
+
+
+class GroupsResource(BaseResource):
+    """Groups API resource wrapper."""
+
+    def create_group(self, payload: Dict[str, Any]) -> Any:
+        return self._post("/api/ss/group", json=payload)
+
+    def get_groups(self, params: Optional[Dict[str, Any]] = None) -> Any:
+        return self._get("/api/ss/groups", params=params)
+
+    def get_group(self, group_id: str, params: Optional[Dict[str, Any]] = None) -> Any:
+        return self._get(f"/api/ss/group/{group_id}", params=params)

{pyattackforge-0.2.1 → pyattackforge-0.2.2}/pyattackforge/resources/projects.py

@@ -130,6 +130,46 @@ class ProjectsResource(BaseResource):
     def update_user_access_on_project(self, project_id: str, user_id: str, payload: Dict[str, Any]) -> Any:
         return self._put(f"/api/ss/project/{project_id}/access/{user_id}", json=payload)
 
+    def add_project_to_group(
+        self,
+        project_id: str,
+        group_id: str,
+        *,
+        project_data: Optional[Dict[str, Any]] = None,
+    ) -> Any:
+        """
+        Convenience helper to add a project to a group by updating the project's groups list.
+        """
+        if not group_id:
+            raise ValueError("group_id is required")
+        data = project_data if project_data is not None else self.get_project(project_id)
+        project = data
+        if isinstance(project, dict) and isinstance(project.get("data"), dict):
+            project = project["data"]
+        if isinstance(project, dict) and isinstance(project.get("project"), dict):
+            project = project["project"]
+        groups = project.get("groups") or project.get("project_groups") or []
+        group_ids: List[str] = []
+        if isinstance(groups, list):
+            for entry in groups:
+                if isinstance(entry, str) and entry:
+                    group_ids.append(entry)
+                    continue
+                if isinstance(entry, dict):
+                    candidate = entry.get("id") or entry.get("group_id") or entry.get("_id")
+                    if isinstance(candidate, str) and candidate:
+                        group_ids.append(candidate)
+        seen: set = set()
+        deduped: List[str] = []
+        for value in group_ids:
+            if value in seen:
+                continue
+            seen.add(value)
+            deduped.append(value)
+        if group_id not in seen:
+            deduped.append(group_id)
+        return self.update_project(project_id, {"groups": deduped})
+
     def extract_projects_list(self, projects_data: Any) -> List[Dict[str, Any]]:
         if not isinstance(projects_data, dict):
             return []

{pyattackforge-0.2.1 → pyattackforge-0.2.2}/pyattackforge.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyattackforge
-Version: 0.2.1
+Version: 0.2.2
 Summary: Python SDK for the AttackForge SSAPI
 Author: hackman238
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE

{pyattackforge-0.2.1 → pyattackforge-0.2.2}/pyattackforge.egg-info/SOURCES.txt

@@ -16,6 +16,7 @@ pyattackforge/resources/__init__.py
 pyattackforge/resources/assets.py
 pyattackforge/resources/base.py
 pyattackforge/resources/findings.py
+pyattackforge/resources/groups.py
 pyattackforge/resources/notes.py
 pyattackforge/resources/projects.py
 pyattackforge/resources/reports.py

{pyattackforge-0.2.1 → pyattackforge-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pyattackforge"
-version = "0.2.1"
+version = "0.2.2"
 description = "Python SDK for the AttackForge SSAPI"
 readme = "README.md"
 license = {file = "LICENSE"}