suite-py 1.44.0__tar.gz → 1.46.0__tar.gz

{suite_py-1.44.0 → suite_py-1.46.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: suite-py
-Version: 1.44.0
+Version: 1.46.0
 Summary: 
 Author: larrywax, EugenioLaghi, michelangelomo
 Author-email: devops@prima.it
@@ -28,7 +28,7 @@ Requires-Dist: pytest (>=7.0.0)
 Requires-Dist: python-dateutil (>=2.8.2)
 Requires-Dist: requests (>=2.26.0)
 Requires-Dist: requests-toolbelt (>=0.9.1)
-Requires-Dist: rich (==14.0.0)
+Requires-Dist: rich (==14.1.0)
 Requires-Dist: semver (>=3.0.4,<4.0.0)
 Requires-Dist: termcolor (>=1.1.0)
 Requires-Dist: truststore (>=0.7,<0.11) ; python_version >= "3.10"

{suite_py-1.44.0 → suite_py-1.46.0}/pyproject.toml

@@ -2,7 +2,7 @@
 authors = ["larrywax, EugenioLaghi, michelangelomo <devops@prima.it>"]
 description = ""
 name = "suite-py"
-version = "1.44.0"
+version = "1.46.0"
 
 [tool.poetry.dependencies]
 Click = ">=7.0"
@@ -23,7 +23,7 @@ python = ">=3.8,<4.0"
 python-dateutil = ">=2.8.2"
 requests = ">=2.26.0"
 requests-toolbelt = ">=0.9.1"
-rich = "==14.0.0"
+rich = "==14.1.0"
 semver = "^3.0.4"
 termcolor = ">=1.1.0"
 truststore = {version = ">=0.7,<0.11", python = ">=3.10"}

{suite_py-1.44.0 → suite_py-1.46.0}/suite_py/__version__.py

@@ -1,2 +1,2 @@
 # -*- encoding: utf-8 -*-
-__version__ = "1.44.0"
+__version__ = "1.46.0"

{suite_py-1.44.0 → suite_py-1.46.0}/suite_py/cli.py

@@ -265,6 +265,28 @@ def cli_unlock_project(obj, environment):
     obj.call(ProjectLock, env=environment, action="unlock").run()
 
 
+@main.command(
+    "estimate-cone",
+    help="Point-in-time estimate of the time needed to complete all unresolved children of a card using the Cone of Uncertainty",
+)
+@click.option("--issue", prompt="Issue ID", type=str)
+@click.option("--sprint-board", prompt="Sprint Board ID", type=str)
+@click.option("--previous-sprints", required=False, type=int, default=6)
+@click.pass_obj
+@catch_exceptions
+def estimate_cone(
+    obj: Context,
+    issue: Optional[str],
+    sprint_board: Optional[str],
+    previous_sprints: int,
+):
+    from suite_py.commands.estimate_cone import EstimateCone
+
+    obj.call(EstimateCone).run(
+        issue=issue, sprint_board=sprint_board, previous_sprints=previous_sprints
+    )
+
+
 @main.command("open-pr", help="Open a PR on GitHub")
 @click.pass_obj
 @catch_exceptions

suite_py-1.46.0/suite_py/commands/estimate_cone.py

@@ -0,0 +1,91 @@
+# -*- encoding: utf-8 -*-
+from suite_py.lib import metrics
+from suite_py.lib.handler.youtrack_handler import YoutrackHandler
+
+
+class EstimateCone:
+    def __init__(self, project, config, tokens):
+        self._project = project
+        self._config = config
+        self._youtrack = YoutrackHandler(config, tokens)
+
+    @metrics.command("estimate-cone")
+    def run(self, issue: str, sprint_board: str, previous_sprints: int):
+        remaining_story_points = self._sum_of_unresolved_descendant_story_points(issue)
+
+        current_sprint = self._youtrack.get_current_sprint(sprint_board)
+        sprints = self._youtrack.get_sprints(sprint_board)
+
+        current_sprint_index = next(
+            (
+                index
+                for index, sprint in enumerate(sprints)
+                if sprint["id"] == current_sprint["id"]
+            ),
+            None,
+        )
+
+        recent_sprints = sprints[
+            current_sprint_index - previous_sprints : current_sprint_index
+        ]
+
+        sprints_resolved_story_points = [
+            self._youtrack.get_sprint_resolved_story_points(sprint_board, sprint["id"])
+            for sprint in recent_sprints
+        ]
+
+        average_story_points_per_sprint = sum(sprints_resolved_story_points) / len(
+            sprints_resolved_story_points
+        )
+        max_story_points_per_sprint = (
+            max(sprints_resolved_story_points) or 1
+        )  # Avoid division by zero
+        min_story_points_per_sprint = (
+            min(sprints_resolved_story_points) or 1
+        )  # Avoid division by zero
+
+        average_estimate = remaining_story_points / average_story_points_per_sprint
+        best_estimate = remaining_story_points / max_story_points_per_sprint
+        worst_estimate = remaining_story_points / min_story_points_per_sprint
+
+        print(f"Remaining story points: {remaining_story_points}")
+
+        print(f"Max story points per sprint: {max_story_points_per_sprint}")
+        print(f"Average story points per sprint: {average_story_points_per_sprint:.2f}")
+        print(f"Min story points per sprint: {min_story_points_per_sprint}")
+
+        print(f"Average estimate: {average_estimate:.2f} sprints")
+        print(f"Best estimate: {best_estimate:.2f} sprints")
+        print(f"Worst estimate: {worst_estimate:.2f} sprints")
+
+        print(f"Sprints resolved story points: {sprints_resolved_story_points}")
+
+    def _sum_of_unresolved_descendant_story_points(self, _issue_readable_id):
+        descendants = self._youtrack.get_issue_descendants(_issue_readable_id)
+
+        descendants_digested = [
+            {
+                "id": descendant.get("idReadable"),
+                "summary": descendant.get("summary"),
+                "resolved": bool(descendant.get("resolved")),
+                "story_points": next(
+                    (
+                        field["value"]
+                        for field in descendant["customFields"]
+                        if field["name"] == "Story Points"
+                    )
+                )
+                or 0,
+            }
+            for descendant in descendants
+        ]
+
+        for descendant in descendants_digested:
+            if not descendant["resolved"]:
+                print(descendant)
+
+        return sum(
+            descendant["story_points"]
+            for descendant in descendants_digested
+            if not descendant["resolved"]
+        )

{suite_py-1.44.0 → suite_py-1.46.0}/suite_py/lib/handler/youtrack_handler.py

@@ -23,6 +23,7 @@ WORKABLE_CARD_TYPES = [
 ]
 
 
+# pylint: disable=too-many-public-methods
 class YoutrackHandler:
     def __init__(self, config, tokens):
         headers = {"Accept": "application/json", "Content-Type": "application/json"}
@@ -43,9 +44,15 @@ class YoutrackHandler:
 
     def get_issue(self, issue_id):
         logger.debug(f"YouTrack issue_id: {issue_id}")
-        params = {
-            "fields": "$type,id,idReadable,summary,parent(issues(id)),customFields(name,value(name))"
-        }
+        _fields = [
+            "$type",
+            "id",
+            "idReadable",
+            "summary",
+            "parent(issues(id))",
+            "customFields(name,value(name))",
+        ]
+        params = {"fields": ",".join(_fields)}
         issue = self._client.get(f"issues/{issue_id}", params=params).json()
         issue["Type"] = self._get_issue_type_name(issue)
         logger.debug(f"YouTrack Issue json {issue}")
@@ -214,3 +221,69 @@ class YoutrackHandler:
         parent = issue["parent"]["issues"][0]
         # recursively get parent issue's type
         return self.get_issue(parent["id"])["Type"]
+
+    def get_issue_descendants(self, issue_id):
+        _fields = [
+            "resolved",
+            "subtasks(issues(id,idReadable,resolved,summary,customFields(name,value),subtasks(issues(id))))",
+        ]
+        params = {"fields": ",".join(_fields)}
+        issue = self._client.get(f"issues/{issue_id}", params=params).json()
+
+        descendants = []
+        for child in issue.get("subtasks", {}).get("issues", []):
+            descendants.append(child)
+            if child.get("subtasks", {}).get("issues"):
+                descendants.extend(self.get_issue_descendants(child["id"]))
+        return descendants
+
+    def get_current_sprint(self, agile_board_id):
+        params = {
+            "fields": "id",
+        }
+        return self._client.get(
+            f"agiles/{agile_board_id}/sprints/current", params=params
+        ).json()
+
+    def get_sprints(self, agile_board_id):
+        params = {
+            "fields": "id",
+        }
+        return self._client.get(
+            f"agiles/{agile_board_id}/sprints", params=params
+        ).json()
+
+    def get_sprint_resolved_story_points(self, agile_board_id, sprint_id):
+        """Get the sum of the story point estimates of all resolved issues of a sprint."""
+        params = {
+            "fields": "id,name,goal,issues(id,idReadable,summary,resolved,customFields(name,value))",
+        }
+        sprint = self._client.get(
+            f"agiles/{agile_board_id}/sprints/{sprint_id}", params=params
+        ).json()
+
+        resolved_issues = [issue for issue in sprint["issues"] if issue["resolved"]]
+
+        total_story_points = 0
+        for issue in resolved_issues:
+            # Find the Story Points custom field
+            story_points_field = next(
+                (
+                    field
+                    for field in issue.get("customFields", [])
+                    if field["name"] == "Story Points"
+                ),
+                None,
+            )
+            if story_points_field and story_points_field.get("value"):
+                try:
+                    story_points = int(story_points_field["value"])
+                    total_story_points += story_points
+                except (ValueError, TypeError):
+                    # Skip issues with invalid story points values
+                    logger.debug(
+                        f"Invalid story points value for issue {issue.get('idReadable', 'unknown')}: {story_points_field.get('value')}"
+                    )
+                    continue
+
+        return total_story_points