update

2026-06-01 09:56:50 +02:00
parent 7c6c3d7223
commit 5c2016a05c
14 changed files with 1140 additions and 261 deletions
@@ -0,0 +1,133 @@
+from dataclasses import dataclass, field
+from typing import Any
+
+from pydantic import BaseModel, Field
+from requests import get
+from requests.compat import quote
+
+from eta.common import EnvVar, JSONDataMap
+
+
+class GitlabProject(BaseModel):
+    name: str
+    name_with_namespace: str
+    path_with_namespace: str
+    web_url: str
+    id: int
+
+    issues: list["GitlabIssue"] = Field(default_factory=list, init=False)
+    merge_requests: list["GitlabMergeRequest"] = Field(default_factory=list, init=False)
+
+
+class GitlabMilestone(BaseModel):
+    id: int
+    title: str
+    state: str
+
+
+class GitlabIssue(BaseModel):
+    title: str
+    description: str | None
+    id: int
+    project_id: int
+    iid: int
+    state: str
+    labels: list[str]
+    milestone: GitlabMilestone | None
+    # web_url: str
+
+
+class GitlabMergeRequest(BaseModel):
+    id: int
+    iid: int
+    description: str | None
+    draft: bool
+    labels: list[str]
+    milestone: GitlabMilestone | None
+    source_branch: str
+    state: str
+    target_branch: str
+    project_id: int
+    title: str
+    web_url: str
+
+
+@dataclass
+class GitlabSession:
+    auth_key: str = field(
+        default_factory=EnvVar(
+            "GITLAB_AUTH",
+            "gitlab auth token is required to be set",
+        )
+    )
+    base_url: str = "https://git.extoll.de/api/v4"
+
+    def _get(self, endpoint: str, **query_params: str) -> JSONDataMap:
+        with get(
+            self.base_url + endpoint,
+            query_params,
+            headers={
+                "accept": "application/json",
+                "PRIVATE-TOKEN": self.auth_key,
+            },
+            verify=False,
+        ) as resp:
+            return resp.json()
+
+    def get_projects(self, **filters: str) -> list[GitlabProject]:
+        ret: list[GitlabProject] = []
+        page = 1
+        while projs := self._get(
+            "/projects",
+            active="true",
+            simple="true",
+            per_page="100",
+            page=str(page),
+        ):
+            ret.extend(GitlabProject.model_validate(proj) for proj in projs)
+            page = page + 1
+            if len(projs) < 100:
+                break
+
+        return ret
+
+    def get_issues(self, projects: list[GitlabProject]) -> list[GitlabIssue]:
+        ret: list[GitlabIssue] = []
+        pids = {p.id for p in projects}
+        page = 1
+        while True:
+            issues = self._get(
+                "/issues",
+                per_page="100",
+                page=str(page),
+            )
+            ret.extend(
+                GitlabIssue.model_validate(iss) for iss in issues if int(iss["project_id"]) in pids
+            )
+            page = page + 1
+            if len(issues) < 100:
+                break
+
+        return ret
+
+    def get_merge_requests(self, projects: list[GitlabProject]) -> list[GitlabMergeRequest]:
+        ret: list[GitlabMergeRequest] = []
+        pids = {p.id for p in projects}
+        page = 1
+        while True:
+            merge_requests = self._get(
+                "/merge_requests",
+                simple="true",
+                per_page="100",
+                page=str(page),
+            )
+            ret.extend(
+                GitlabMergeRequest.model_validate(mr)
+                for mr in merge_requests
+                if int(mr["project_id"]) in pids
+            )
+            page = page + 1
+            if len(merge_requests) < 100:
+                break
+
+        return ret