suite-py 1.49.1__tar.gz → 1.52.0__tar.gz

{suite_py-1.49.1 → suite_py-1.52.0}/PKG-INFO

@@ -1,12 +1,11 @@
 Metadata-Version: 2.1
 Name: suite-py
-Version: 1.49.1
+Version: 1.52.0
 Summary: 
 Author: larrywax, EugenioLaghi, michelangelomo
 Author-email: devops@prima.it
-Requires-Python: >=3.8,<4.0
+Requires-Python: >=3.9,<4.0
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
@@ -16,14 +15,14 @@ Requires-Dist: InquirerPy (>=0.2.0)
 Requires-Dist: PyGithub (>=1.57)
 Requires-Dist: PyYaml (>=5.4)
 Requires-Dist: autoupgrade-prima (>=0.6)
-Requires-Dist: boto3 (>=1.17.84)
+Requires-Dist: boto3 (>=1.40)
 Requires-Dist: cement (>=3.0)
 Requires-Dist: colorama (>=0.4.3)
 Requires-Dist: halo (>=0.0.28)
 Requires-Dist: inquirer (>=3.1.4,<4.0.0)
 Requires-Dist: itsdangerous (==2.2.0)
 Requires-Dist: keyring (>=23.9.1,<26.0.0)
-Requires-Dist: pylint (>=2.14.5,<4.0.0)
+Requires-Dist: pylint (<5.0.0)
 Requires-Dist: pytest (>=7.0.0)
 Requires-Dist: python-dateutil (>=2.8.2)
 Requires-Dist: requests (>=2.26.0)

{suite_py-1.49.1 → suite_py-1.52.0}/pyproject.toml

@@ -2,7 +2,7 @@
 authors = ["larrywax, EugenioLaghi, michelangelomo <devops@prima.it>"]
 description = ""
 name = "suite-py"
-version = "1.49.1"
+version = "1.52.0"
 
 [tool.poetry.dependencies]
 Click = ">=7.0"
@@ -10,16 +10,16 @@ InquirerPy = ">=0.2.0"
 PyGithub = ">=1.57"
 PyYaml = ">=5.4"
 autoupgrade-prima = ">=0.6"
-boto3 = ">=1.17.84"
+boto3 = ">=1.40"
 cement = ">=3.0"
 colorama = ">=0.4.3"
 halo = ">=0.0.28"
 inquirer = "^3.1.4"
 itsdangerous = "==2.2.0"
 keyring = ">=23.9.1,<26.0.0"
-pylint = ">=2.14.5,<4.0.0"
+pylint = "<5.0.0"
 pytest = ">=7.0.0"
-python = ">=3.8,<4.0"
+python = ">=3.9,<4.0"
 python-dateutil = ">=2.8.2"
 requests = ">=2.26.0"
 requests-toolbelt = ">=0.9.1"
@@ -29,7 +29,7 @@ termcolor = ">=1.1.0"
 truststore = {version = ">=0.7,<0.11", python = ">=3.10"}
 
 [tool.poetry.dev-dependencies]
-black = ">=22.6,<25.0"
+black = ">=22.6,<26.0"
 
 [tool.poetry.scripts]
 suite-py = "suite_py.cli:main"

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/__version__.py

@@ -1,2 +1,2 @@
 # -*- encoding: utf-8 -*-
-__version__ = "1.49.1"
+__version__ = "1.52.0"

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/cli.py

@@ -46,6 +46,7 @@ from suite_py.lib import logger, metrics
 from suite_py.lib.config import Config
 from suite_py.lib.handler import git_handler as git
 from suite_py.lib.handler import prompt_utils
+from suite_py.lib.handler.backstage_handler import Backstage
 from suite_py.lib.handler.captainhook_handler import CaptainHook
 from suite_py.lib.handler.git_handler import GitHandler
 from suite_py.lib.handler.okta_handler import Okta
@@ -132,7 +133,7 @@ def catch_exceptions(func):
 @click.option(
     "--timeout",
     type=click.INT,
-    help="Timeout in seconds for Captainhook operations",
+    help="Timeout in seconds for Captainhook/Backstage operations",
 )
 @click.option("-v", "--verbose", count=True)
 @click.pass_context
@@ -177,16 +178,19 @@ def main(ctx, project, timeout, verbose):
         ctx.exit()
     if timeout:
         config.user["captainhook_timeout"] = timeout
+        config.user["backstage_timeout"] = timeout
 
     tokens = Tokens()
     okta = Okta(config, tokens)
     captainhook = CaptainHook(config, okta, tokens)
+    backstage = Backstage(config, okta, tokens)
     project = os.path.basename(project)
     git_handler = GitHandler(project, config)
     youtrack_handler = YoutrackHandler(config, tokens)
 
     ctx.obj = Context(
         captainhook=captainhook,
+        backstage=backstage,
         config=config,
         git_handler=git_handler,
         okta=okta,
@@ -233,21 +237,21 @@ def bump(obj: Context, project: Optional[str] = None, version: Optional[str] = N
 @main.command(
     "create-branch", help="Create local branch and set the YouTrack card in progress"
 )
-@click.option("--card", type=click.STRING, help="YouTrack card number (ex. PRIMA-123)")
 @click.option(
     "--autostash",
     is_flag=True,
     help="Stash uncommitted changes before creating the branch and reapply them afterward",
 )
 @click.option(
-    "--parent-branch",
+    "--branch-name",
     type=click.STRING,
-    help="Parent branch to create the new branch from",
+    help="Branch name template. Supports {card_id}, {type}, {summary} placeholders (ex. '{card_id}/{type}/{summary}')",
 )
+@click.option("--card", type=click.STRING, help="YouTrack card ID (ex. PRIMA-4423)")
 @click.option(
-    "--branch-name",
+    "--parent-branch",
     type=click.STRING,
-    help="Branch name template. Supports {card_id}, {type}, {summary} placeholders (ex. '{card_id}/{type}/{summary}')",
+    help="Parent branch to create the new branch from",
 )
 @click.pass_obj
 @catch_exceptions
@@ -255,10 +259,10 @@ def cli_create_branch(obj, card, autostash, parent_branch, branch_name):
     from suite_py.commands.create_branch import CreateBranch
 
     obj.call(CreateBranch).run(
-        card_id=card,
         autostash=autostash,
-        parent_branch=parent_branch,
         branch_name=branch_name,
+        card_id=card,
+        parent_branch=parent_branch,
     )
 
 

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/commands/check.py

@@ -8,18 +8,22 @@ from suite_py.lib.symbol import CHECKMARK, CROSSMARK
 
 
 class Check:
-    def __init__(self, captainhook, config, tokens):
+    # delete this after `captainhook` has been removed
+    # pylint: disable=too-many-instance-attributes
+    def __init__(self, captainhook, backstage, config, tokens):
         self.config = config
         self._invalid_or_missing_tokens = []
         self._tokens = tokens
         self._youtrack = YoutrackHandler(config, tokens)
         self._github = GithubHandler(tokens)
         self._captainhook = captainhook
+        self._backstage = backstage
 
         self._checks = [
             ("Github", self._check_github),
             ("Youtrack", self._check_youtrack),
             ("CaptainHook", self._check_captainhook),
+            ("Backstage", self._check_backstage),
             ("AWS", self._check_aws),
         ]
 
@@ -69,6 +73,13 @@ class Check:
         except Exception:
             return "unreachable"
 
+    def _check_backstage(self):
+        try:
+            self._backstage.check()
+            return "ok"
+        except Exception:
+            return "unreachable"
+
     def _check_aws(self):
         try:
             if self.config.aws != "":

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/commands/context.py

@@ -2,6 +2,7 @@ import dataclasses
 from inspect import signature
 
 from suite_py.lib.config import Config
+from suite_py.lib.handler.backstage_handler import Backstage
 from suite_py.lib.handler.captainhook_handler import CaptainHook
 from suite_py.lib.handler.git_handler import GitHandler
 from suite_py.lib.handler.okta_handler import Okta
@@ -9,9 +10,12 @@ from suite_py.lib.handler.youtrack_handler import YoutrackHandler
 from suite_py.lib.tokens import Tokens
 
 
+# delete this after `captainhook` has been removed
+# pylint: disable=too-many-instance-attributes
 @dataclasses.dataclass
 class Context:
     captainhook: CaptainHook
+    backstage: Backstage
     config: Config
     git_handler: GitHandler
     okta: Okta

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/commands/create_branch.py

@@ -1,7 +1,6 @@
 # -*- coding: utf-8 -*-
 import re
 import sys
-
 import requests
 
 from suite_py.lib import logger, metrics
@@ -19,7 +18,7 @@ class CreateBranch:
     ):
         self._config = config
         self._git_handler = git_handler
-        self._youtrack = youtrack_handler
+        self._youtrack_handler = youtrack_handler
 
     @metrics.command("create-branch")
     def run(
@@ -41,22 +40,22 @@ class CreateBranch:
 
         try:
             if card_id:
-                issue = self._youtrack.get_issue(card_id)
+                issue = self._youtrack_handler.get_issue(card_id)
             else:
-                issue = self._youtrack.get_issue(self._ask_card())
-        except Exception:
+                issue = self._youtrack_handler.get_issue(self._ask_card_id())
+        except Exception as e:
             logger.error(
-                "There was a problem retrieving the issue from YouTrack. Check that the issue number is correct"
+                f"There was a problem retrieving the issue `{card_id}` from YouTrack. {e}"
             )
             sys.exit(-1)
 
         self._checkout_branch(issue, autostash, parent_branch, branch_name)
 
-        user = self._youtrack.get_current_user()
-        self._youtrack.assign_to(issue["id"], user["login"])
+        user = self._youtrack_handler.get_current_user()
+        self._youtrack_handler.assign_to(issue["id"], user["login"])
 
         try:
-            self._youtrack.update_state(
+            self._youtrack_handler.update_state(
                 issue["id"], self._config.youtrack["picked_state"]
             )
         except requests.exceptions.HTTPError:
@@ -87,7 +86,7 @@ class CreateBranch:
             "Insert the YouTrack issue number:", self._config.user["default_slug"]
         )
 
-    def _ask_card(self):
+    def _ask_card_id(self):
         suggestions = self._get_card_suggestions()
         user_choice = (
             self._select_card(suggestions)
@@ -98,8 +97,9 @@ class CreateBranch:
 
     def _get_card_suggestions(self):
         try:
-            return self._youtrack.search_issues(
-                self._config.user["card_suggest_query"], 5
+            return self._youtrack_handler.search_issues(
+                self._config.user["card_suggest_query"],
+                self._config.user["card_suggestions_limit"],
             )
         except Exception:
             logger.warning(

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/commands/merge_pr.py

@@ -6,6 +6,7 @@ from halo import Halo
 from suite_py.lib import logger, metrics
 from suite_py.lib.handler import git_handler as git
 from suite_py.lib.handler import prompt_utils
+from suite_py.lib.handler.backstage_handler import Backstage
 from suite_py.lib.handler.captainhook_handler import CaptainHook
 from suite_py.lib.handler.git_handler import GitHandler
 from suite_py.lib.handler.github_handler import GithubHandler
@@ -14,11 +15,14 @@ from suite_py.lib.symbol import CHECKMARK, CROSSMARK
 
 
 class MergePR:
-    def __init__(self, project, captainhook: CaptainHook, config, tokens):
+    def __init__(
+        self, project, captainhook: CaptainHook, backstage: Backstage, config, tokens
+    ):
         self._project = project
         self._config = config
         self._youtrack = YoutrackHandler(config, tokens)
         self._captainhook = captainhook
+        self._backstage = backstage
         self._git = GitHandler(project, config)
         self._github = GithubHandler(tokens)
 
@@ -63,6 +67,7 @@ class MergePR:
             default=False,
         ):
             self._captainhook.lock_project(self._project, "staging")
+            self._backstage.lock_project(self._project, "staging")
 
         if youtrack_id:
             logger.info("Updating card status on YouTrack...")

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/commands/open_pr.py

@@ -1,5 +1,6 @@
 # -*- encoding: utf-8 -*-
 import sys
+from string import Template
 
 from github import GithubException
 
@@ -77,7 +78,10 @@ class OpenPR:
 
             default_title = self._youtrack.get_issue(youtrack_id)["summary"]
             title_without_prefix = _ask_for_title(default_title)
-            title = f"[{youtrack_id}]: {title_without_prefix}"
+
+            title = Template(self._config.user["pr_title_template"]).substitute(
+                card_id=youtrack_id, title=title_without_prefix
+            )
         else:
             logger.warning(
                 f"Creating pull request on the {self._project} project for the {self._branch_name} branch NOT linked to YouTrack card"

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/commands/project_lock.py

@@ -7,11 +7,12 @@ from suite_py.lib import logger, metrics
 
 
 class ProjectLock:
-    def __init__(self, project, env, action, captainhook):
+    def __init__(self, project, env, action, captainhook, backstage):
         self._project = project
         self._env = _parse_env(env)
         self._action = action
         self._captainhook = captainhook
+        self._backstage = backstage
 
     @metrics.command("manage-project-lock")
     def run(self):
@@ -25,6 +26,8 @@ class ProjectLock:
                     "Captainhook request timed out. Try with suite-py --timeout=60 lock-project lock"
                 )
                 sys.exit(1)
+
+            self._backstage.lock_project(self._project, self._env)
         elif self._action == "unlock":
             try:
                 req = self._captainhook.unlock_project(self._project, self._env)
@@ -37,6 +40,8 @@ class ProjectLock:
                     "Captainhook request timed out. Try with suite-py --timeout=60 lock-project lock"
                 )
                 sys.exit(1)
+
+            self._backstage.unlock_project(self._project, self._env)
         else:
             logger.warning("I'm confused. Make the correct choice.")
             sys.exit(-1)

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/commands/release.py

@@ -42,6 +42,8 @@ class Release:
         self._repo = self._github.get_repo(project)
         self._git = GitHandler(project, config)
         self._version = VersionHandler(self._repo, self._git, self._github)
+        # Cache for workflow events {check_run_database_id: event_type}
+        self.workflow_events = {}
 
     @metrics.command("release")
     def run(self):
@@ -145,6 +147,12 @@ class Release:
         if self._commit or not commits:
             return
 
+        # Batch fetch workflow events for all commits upfront for efficiency
+        shas = [c.sha for c in commits]
+        self.workflow_events = self._github.get_workflow_events_for_commits(
+            self._repo, shas
+        )
+
         choices = []
 
         def build_choice(c, icon):
@@ -196,9 +204,20 @@ class Release:
             return []
 
     def _classify_from_check_runs(self, check_runs):
-        statuses = {r.status for r in check_runs if getattr(r, "status", None)}
+        """Classify commit status from check runs, filtering by event type.
+
+        Args:
+            check_runs: List of check run objects from GitHub API
+
+        Returns:
+            str: One of "in_progress", "failure", "cancelled", "success", "unknown"
+        """
+        # Filter check runs by event type if workflow events are available
+        filtered_runs = self._filter_check_runs_by_event(check_runs)
+
+        statuses = {r.status for r in filtered_runs if getattr(r, "status", None)}
         conclusions = {
-            r.conclusion for r in check_runs if getattr(r, "conclusion", None)
+            r.conclusion for r in filtered_runs if getattr(r, "conclusion", None)
         }
 
         if self._any_in_progress(statuses):
@@ -211,6 +230,39 @@ class Release:
             return "success"
         return "unknown"
 
+    def _filter_check_runs_by_event(self, check_runs):
+        """Filter check runs to include only push events and unknown events.
+
+        Args:
+            check_runs: List of check run objects from GitHub API
+
+        Returns:
+            list: Filtered list of check runs
+        """
+        # If no workflow events cached, return all check runs (fallback)
+        if not self.workflow_events:
+            return check_runs
+
+        filtered = []
+
+        for check_run in check_runs:
+            # Get the database ID for this check run
+            database_id = getattr(check_run, "id", None)
+            if not database_id:
+                continue
+
+            # Look up the event type for this check run
+            event_type = self.workflow_events.get(database_id, "unknown")
+
+            # Include check runs with "push" or "unknown" event types
+            # "unknown" covers manual checks and status checks without workflows
+            if event_type in ("push", "unknown"):
+                filtered.append(check_run)
+
+        # If filtering resulted in no check runs, fall back to all check runs
+        # This handles edge cases where GraphQL data might be incomplete
+        return filtered if filtered else check_runs
+
     @staticmethod
     def _any_in_progress(statuses):
         return any(
@@ -228,7 +280,11 @@ class Release:
     def _is_success(conclusions):
         return bool(conclusions) and (
             all(c == "success" for c in conclusions)
-            or ("success" in conclusions and not Release._any_failure(conclusions))
+            or (
+                "success" in conclusions
+                and not Release._any_failure(conclusions)
+                and "cancelled" not in conclusions
+            )
         )
 
     @staticmethod

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/lib/config.py

@@ -36,11 +36,15 @@ class Config:
         conf["user"].setdefault("default_slug", "PRIMA-XXX")
         default_search = f"in:{conf['user']['default_slug'].split('-')[0]} #{{To Do}}"
         conf["user"].setdefault("card_suggest_query", default_search)
+        conf["user"].setdefault("card_suggestions_limit", 5)
         # This is in seconds
         conf["user"].setdefault("captainhook_timeout", 30)
         conf["user"].setdefault("captainhook_url", "https://captainhook.prima.it")
+        conf["user"].setdefault("backstage_timeout", 30)
+        conf["user"].setdefault("backstage_url", "https://backstage.helloprima.com")
         conf["user"].setdefault("use_commits_in_pr_body", False)
         conf["user"].setdefault("frequent_reviewers_max_number", 5)
+        conf["user"].setdefault("pr_title_template", "[$card_id]: $title")
 
         conf["youtrack"].setdefault("add_reviewers_tags", True)
         conf["youtrack"].setdefault("default_issue_type", "Task")

suite_py-1.52.0/suite_py/lib/handler/backstage_handler.py

@@ -0,0 +1,82 @@
+# -*- encoding: utf-8 -*-
+import requests
+
+from suite_py.__version__ import __version__
+from suite_py.lib.handler.github_handler import GithubHandler
+from suite_py.lib.handler.okta_handler import Okta
+
+
+class BackstageResponseError(Exception): ...
+
+
+class Backstage:
+    _okta: Okta
+    _github: GithubHandler
+
+    def __init__(self, config, okta: Okta, tokens=None):
+        backstage_url = config.user["backstage_url"]
+        self._baseurl = f"{backstage_url}/api/entity-locking"
+        self._timeout = config.user["backstage_timeout"]
+        self._okta = okta
+
+        if tokens is not None:
+            self._github = GithubHandler(tokens)
+
+    def lock_project(self, project, env):
+        return self.send_request(
+            "post",
+            "/projects/manage-lock",
+            json={
+                "project": project,
+                "status": "locked",
+                "user": self._get_user(),
+                "environment": env,
+            },
+        )
+
+    def unlock_project(self, project, env):
+        return self.send_request(
+            "post",
+            "/projects/manage-lock",
+            json={
+                "project": project,
+                "status": "unlocked",
+                "user": self._get_user(),
+                "environment": env,
+            },
+        )
+
+    def status(self, project, env):
+        return self.send_request(
+            "get", f"/projects/check?project={project}&environment={env}"
+        )
+
+    def check(self):
+        response = self.send_request("get", "/ping", timeout=(2, self._timeout))
+        assert response.text == "pong"
+
+    def send_request(self, method, endpoint, data=None, json=None, timeout=None):
+        response = requests.request(
+            method,
+            f"{self._baseurl}{endpoint}",
+            headers=self._headers(),
+            data=data,
+            json=json,
+            timeout=(timeout or self._timeout),
+        )
+
+        if response.status_code == 200:
+            return response
+
+        raise BackstageResponseError(
+            f"Got unexpected status code from Backstage: {response.status_code}"
+        )
+
+    def _get_user(self):
+        return self._github.get_user().login
+
+    def _headers(self):
+        return {
+            "User-Agent": f"suite-py/{__version__}",
+            "Authorization": f"Bearer {self._okta.get_id_token()}",
+        }

suite_py-1.52.0/suite_py/lib/handler/github_handler.py

@@ -0,0 +1,263 @@
+# -*- encoding: utf-8 -*-
+from github import Auth, Github
+from github.GitRelease import GitRelease  # pylint: disable=unused-import
+from github.Repository import Repository  # pylint: disable=unused-import
+
+from suite_py.lib import logger
+
+
+class GithubHandler:
+    # pylint: disable=too-many-public-methods
+    _organization = "primait"
+
+    def __init__(self, tokens):
+        auth = Auth.Token(tokens.github)
+        self._client = Github(auth=auth)
+
+    def get_repo(self, repo_name):
+        return self._client.get_repo(f"{self._organization}/{repo_name}")
+
+    def get_organization(self):
+        return self._client.get_organization(self._organization)
+
+    def get_user(self):
+        return self._client.get_user()
+
+    # pylint: disable=too-many-positional-arguments
+    def create_pr(self, repo, branch, title, body="", base="master", is_draft=False):
+        return self.get_repo(repo).create_pull(
+            title=title, head=branch, base=base, body=body, draft=is_draft
+        )
+
+    def get_pr(self, repo, pr_number):
+        return self.get_repo(repo).get_pull(pr_number)
+
+    def get_branch_from_pr(self, repo, pr_number):
+        repo = self.get_repo(repo)
+        return repo.get_branch(repo.get_pull(pr_number).head.ref)
+
+    def get_tags(self, repo):
+        return self.get_repo(repo).get_tags()
+
+    def get_team_members(self, team_name=""):
+        return self.get_organization().get_team_by_slug(team_name).get_members()
+
+    def get_all_users(self):
+        return self.get_organization().get_members()
+
+    def get_list_pr(self, repo):
+        pulls = self.get_repo(repo).get_pulls(
+            state="open", sort="created", base="master"
+        )
+        return pulls
+
+    def get_pr_from_branch(self, repo, branch):
+        return self.get_repo(repo).get_pulls(head=f"primait:{branch}")
+
+    def get_link_from_pr(self, repo, pr_number):
+        return f"https://github.com/primait/{repo}/pull/{pr_number}"
+
+    def get_commits_since_release(self, repo, tag):
+        release_commit = repo.get_commit(tag)
+        commits = []
+        for c in repo.get_commits():
+            if c == release_commit:
+                break
+            commits.append(c)
+        return commits
+
+    def get_latest_release_if_exists(self, repo):
+        try:
+            return repo.get_latest_release()
+        except Exception:
+            return None
+
+    def user_is_admin(self, repo):
+        return self.get_repo(repo).permissions.admin
+
+    def get_release_if_exists(self, repo, release):
+        try:
+            return repo.get_release(release)
+        except Exception:
+            return None
+
+    def get_build_status(self, repo, ref):
+        return self.get_repo(repo).get_commit(ref).get_combined_status()
+
+    def get_releases(self, repo):
+        return self.get_repo(repo).get_releases()
+
+    def get_branches(self, repo):
+        return self.get_repo(repo).get_branches()
+
+    def get_raw_content(self, repo, ref, file):
+        repo = self.get_repo(repo)
+        content = repo.get_contents(file, ref)
+        return content.decoded_content
+
+    def _execute_graphql(self, query, variables):
+        """
+        Execute a GraphQL query against GitHub's GraphQL API.
+
+        Args:
+            query: The GraphQL query string
+            variables: Dictionary of variables for the query (can be None)
+
+        Returns:
+            dict: The parsed JSON response, or None if an error occurs
+        """
+        try:
+            requester = self._client.requester
+            response = requester.graphql_query(query, variables)
+            return response
+        except Exception as error:
+            logger.error(
+                "GraphQL query failed: %s. Variables: %s",
+                str(error),
+                variables,
+            )
+            return None
+
+    def get_workflow_events_for_commits(self, repo, shas):
+        """Fetch workflow events for a list of commit SHAs using GraphQL.
+
+        Args:
+            repo: Repository object from PyGithub
+            shas: List of commit SHA strings
+
+        Returns:
+            dict: Mapping of {check_run_database_id: event_type}
+                  Returns empty dict if GraphQL query fails or response contains errors.
+                  This allows callers to continue with unfiltered check runs as fallback.
+
+        Raises:
+            ValueError: If response parsing fails (raised by _parse_workflow_events_response)
+        """
+        if not shas:
+            return {}
+
+        # Build GraphQL query using variables for commit SHAs (not string interpolation)
+        # The GitHub GraphQL API only accepts single oid per object() call
+        commit_fragments = []
+        variables = {
+            "owner": repo.owner.login,
+            "repo": repo.name,
+        }
+
+        # Add each SHA as a variable and build corresponding query fragment
+        for idx, sha in enumerate(shas):
+            sha_var = f"sha{idx}"
+            variables[sha_var] = sha
+
+            commit_fragments.append(
+                f"""
+            commit{idx}: object(oid: ${sha_var}) {{
+              ... on Commit {{
+                oid
+                checkSuites(first: 100) {{
+                  nodes {{
+                    workflowRun {{
+                      event
+                    }}
+                    checkRuns(first: 100) {{
+                      nodes {{
+                        databaseId
+                        name
+                      }}
+                    }}
+                  }}
+                }}
+              }}
+            }}
+            """
+            )
+
+        # Build query signature with all SHA variables
+        sha_params = ", ".join([f"$sha{i}: GitObjectID!" for i in range(len(shas))])
+        commit_fragments_str = "\n".join(commit_fragments)
+        query = f"""
+        query($owner: String!, $repo: String!, {sha_params}) {{
+          repository(owner: $owner, name: $repo) {{
+            {commit_fragments_str}
+          }}
+        }}
+        """
+
+        response = self._execute_graphql(query, variables)
+
+        if not response or "errors" in response or "data" not in response:
+            if response and "errors" in response:
+                logger.error(
+                    "GraphQL errors for workflow events (repo=%s/%s, commits=%d): %s",
+                    repo.owner.login,
+                    repo.name,
+                    len(shas),
+                    response["errors"],
+                )
+            elif not response:
+                logger.warning(
+                    "No GraphQL response for workflow events (repo=%s/%s, commits=%d)",
+                    repo.owner.login,
+                    repo.name,
+                    len(shas),
+                )
+            return {}
+
+        return self._parse_workflow_events_response(response, shas)
+
+    def _parse_workflow_events_response(self, response, shas):
+        """Parse GraphQL response to extract workflow events for commits.
+
+        Args:
+            response: GraphQL response dict from GitHub API
+            shas: List of commit SHA strings (used for indexing aliased queries)
+
+        Returns:
+            dict: Mapping of {check_run_database_id: event_type}
+
+        Raises:
+            ValueError: If response structure is invalid or unexpected
+        """
+        result = {}
+        try:
+            repository = response.get("data", {}).get("repository", {})
+
+            # Iterate through the aliased commit responses
+            for idx in range(len(shas)):
+                commit_alias = f"commit{idx}"
+                commit_data = repository.get(commit_alias)
+
+                if not commit_data:
+                    continue
+
+                check_suites = commit_data.get("checkSuites", {}).get("nodes", [])
+                for suite in check_suites:
+                    if not suite:
+                        continue
+
+                    # Handle case where workflowRun is None (manual checks, status checks)
+                    workflow_run = suite.get("workflowRun")
+                    event_type = (
+                        workflow_run.get("event") if workflow_run else "unknown"
+                    )
+
+                    check_runs = suite.get("checkRuns", {}).get("nodes", [])
+                    for check_run in check_runs:
+                        if not check_run:
+                            continue
+
+                        database_id = check_run.get("databaseId")
+                        if database_id:
+                            result[database_id] = event_type
+
+        except (AttributeError, KeyError, TypeError) as error:
+            # Log and raise if there's any issue parsing the response
+            logger.error(
+                "Failed to parse workflow events response: %s. Response structure may be unexpected.",
+                str(error),
+            )
+            raise ValueError(
+                f"Failed to parse GraphQL workflow events response: {error}"
+            ) from error
+
+        return result

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/lib/handler/okta_handler.py

@@ -5,7 +5,7 @@ from suite_py.lib import logger, oauth
 from suite_py.lib.config import Config
 from suite_py.lib.tokens import Tokens
 
-_SCOPE = "openid offline_access"
+_SCOPE = "openid email offline_access"
 
 
 class OktaError(Exception):

{suite_py-1.49.1 → suite_py-1.52.0}/suite_py/lib/handler/youtrack_handler.py

@@ -19,6 +19,7 @@ WORKABLE_CARD_TYPES = [
     "Rework",
     "Analysis",
     "Sec Issue",
+    "Task",
     "UXUI",
 ]
 
@@ -53,10 +54,12 @@ class YoutrackHandler:
             "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}")
 
+        issue["Type"] = self._get_issue_type_name(issue)
+
         return issue
 
     def search_issues(self, search, amount=5):
@@ -210,17 +213,50 @@ class YoutrackHandler:
         return re.sub(REGEX, f"[\\1]({self._issue_url}\\1)", text)
 
     def _get_issue_type_name(self, issue):
-        type_field = [x for x in issue["customFields"] if x["name"] == "Type"][0]
+        try:
+            type_field = [x for x in issue["customFields"] if x["name"] == "Type"][0]
+
+            if isinstance(type_field, list):
+                logger.debug(
+                    "Type field is a list. Defaulting to first value!",
+                    extra={"card_id": issue["idReadable"], "type_field": type_field},
+                )
+                type_field = type_field[0]
+
+            if type_field["value"]["name"] in WORKABLE_CARD_TYPES:
+                return type_field["value"]["name"]
+
+            logger.debug(
+                "Type field %s is not workable",
+                type_field["value"]["name"],
+                extra={"card_id": issue["idReadable"]},
+            )
 
-        if type_field["value"]["name"] in WORKABLE_CARD_TYPES:
-            return type_field["value"]["name"]
+            if len(issue["parent"]["issues"]) == 0:
+                return self._default_issue_type
 
-        if len(issue["parent"]["issues"]) == 0:
-            return self._default_issue_type
+            parent = issue["parent"]["issues"][0]
+            # recursively get parent issue's type
+            return self.get_issue(parent["id"])["Type"]
 
-        parent = issue["parent"]["issues"][0]
-        # recursively get parent issue's type
-        return self.get_issue(parent["id"])["Type"]
+        except Exception:
+            logger.warning(
+                """
+Failed to get the issue type from youtrack card: %s. Defaulting to %s.
+This is likely caused by either a misconfiguration of your YouTrack board or a suite-py bug. Ask the DevEx team for help.
+            """,
+                issue["idReadable"],
+                self._default_issue_type,
+            )
+
+            # Log exception in debug mode
+            logger.debug(
+                "Failed to get issue type from youtrack card %s",
+                issue["idReadable"],
+                exc_info=True,
+            )
+
+            return self._default_issue_type
 
     def get_issue_descendants(self, issue_id):
         _fields = [

suite_py-1.49.1/suite_py/lib/handler/github_handler.py

@@ -1,92 +0,0 @@
-# -*- encoding: utf-8 -*-
-from github import Github
-from github.GitRelease import GitRelease  # pylint: disable=unused-import
-from github.Repository import Repository  # pylint: disable=unused-import
-
-
-class GithubHandler:
-    _organization = "primait"
-
-    def __init__(self, tokens):
-        self._client = Github(tokens.github)
-
-    def get_repo(self, repo_name):
-        return self._client.get_repo(f"{self._organization}/{repo_name}")
-
-    def get_organization(self):
-        return self._client.get_organization(self._organization)
-
-    def get_user(self):
-        return self._client.get_user()
-
-    # pylint: disable=too-many-positional-arguments
-    def create_pr(self, repo, branch, title, body="", base="master", is_draft=False):
-        return self.get_repo(repo).create_pull(
-            title=title, head=branch, base=base, body=body, draft=is_draft
-        )
-
-    def get_pr(self, repo, pr_number):
-        return self.get_repo(repo).get_pull(pr_number)
-
-    def get_branch_from_pr(self, repo, pr_number):
-        repo = self.get_repo(repo)
-        return repo.get_branch(repo.get_pull(pr_number).head.ref)
-
-    def get_tags(self, repo):
-        return self.get_repo(repo).get_tags()
-
-    def get_team_members(self, team_name=""):
-        return self.get_organization().get_team_by_slug(team_name).get_members()
-
-    def get_all_users(self):
-        return self.get_organization().get_members()
-
-    def get_list_pr(self, repo):
-        pulls = self.get_repo(repo).get_pulls(
-            state="open", sort="created", base="master"
-        )
-        return pulls
-
-    def get_pr_from_branch(self, repo, branch):
-        return self.get_repo(repo).get_pulls(head=f"primait:{branch}")
-
-    def get_link_from_pr(self, repo, pr_number):
-        return f"https://github.com/primait/{repo}/pull/{pr_number}"
-
-    def get_commits_since_release(self, repo, tag):
-        release_commit = repo.get_commit(tag)
-        commits = []
-        for c in repo.get_commits():
-            if c == release_commit:
-                break
-            commits.append(c)
-        return commits
-
-    def get_latest_release_if_exists(self, repo):
-        try:
-            return repo.get_latest_release()
-        except Exception:
-            return None
-
-    def user_is_admin(self, repo):
-        return self.get_repo(repo).permissions.admin
-
-    def get_release_if_exists(self, repo, release):
-        try:
-            return repo.get_release(release)
-        except Exception:
-            return None
-
-    def get_build_status(self, repo, ref):
-        return self.get_repo(repo).get_commit(ref).get_combined_status()
-
-    def get_releases(self, repo):
-        return self.get_repo(repo).get_releases()
-
-    def get_branches(self, repo):
-        return self.get_repo(repo).get_branches()
-
-    def get_raw_content(self, repo, ref, file):
-        repo = self.get_repo(repo)
-        content = repo.get_contents(file, ref)
-        return content.decoded_content