suite-py 1.41.9__py3-none-any.whl → 1.42.0__py3-none-any.whl

suite_py/lib/handler/drone_handler.py

@@ -1,252 +0,0 @@
-# -*- encoding: utf-8 -*-
-import os
-import subprocess
-import sys
-import time
-from distutils.version import StrictVersion  # pylint: disable=W0402
-from subprocess import CalledProcessError
-
-import requests
-import yaml
-from halo import Halo
-
-from suite_py.lib import logger
-
-
-class DroneHandler:
-    def __init__(self, config, tokens, repo=None):
-        self._token = tokens.drone
-        self._config = config
-        os.environ["DRONE_TOKEN"] = self._token
-        self._baseurl = os.environ.get("DRONE_SERVER", "https://drone-1.prima.it")
-        self._repo = repo
-        if repo:
-            self._path = os.path.join(config.user["projects_home"], repo)
-
-    def get_last_build_url(self, prefix=None):
-        with Halo(text="Contacting drone...", spinner="dots", color="magenta"):
-            # necessario per far comparire la build che abbiamo appena pushato
-            time.sleep(2)
-            try:
-                # pylint: disable-next=missing-timeout
-                builds = requests.get(
-                    f"{self._baseurl}/api/repos/primait/{self._repo}/builds",
-                    headers={"Authorization": f"Bearer {self._token}"},
-                ).json()
-
-                if prefix:
-                    builds = [b for b in builds if b["target"].startswith(prefix)]
-
-                return f"{self._baseurl}/primait/{self._repo}/{builds[0]['number']}"
-            except Exception:
-                return ""
-
-    def get_pr_build_number(self, commit_sha):
-        with Halo(text="Contacting drone...", spinner="dots", color="magenta"):
-            tries = 10
-            while tries > 0:
-                tries -= 1
-                try:
-                    # pylint: disable-next=missing-timeout
-                    builds = requests.get(
-                        f"{self._baseurl}/api/repos/primait/{self._repo}/builds?per_page=100",
-                        headers={"Authorization": f"Bearer {self._token}"},
-                    ).json()
-                    builds = [b for b in builds if b["after"] == commit_sha]
-                    return builds[0]["number"]
-                except Exception:
-                    time.sleep(1)
-
-            return None
-
-    def get_user(self):
-        try:
-            # pylint: disable-next=missing-timeout
-            user = requests.get(
-                f"{self._baseurl}/api/user",
-                headers={"Authorization": f"Bearer {self._token}"},
-            ).json()
-            return user
-        except Exception:
-            return None
-
-    def get_repo_build(self, repo, build_number):
-        # pylint: disable-next=missing-timeout
-        build = requests.get(
-            f"{self._baseurl}/api/repos/primait/{repo}/builds/{build_number}",
-            headers={"Authorization": f"Bearer {self._token}"},
-        ).json()
-
-        return build
-
-    def get_build_url(self, build_number):
-        if build_number:
-            return f"{self._baseurl}/primait/{self._repo}/{build_number}"
-        return None
-
-    def get_repo_build_url(self, repo, build_number):
-        if build_number:
-            return f"{self._baseurl}/primait/{repo}/{build_number}"
-        return None
-
-    def get_build_and_pipeline_url(self, repo, build_number, pipeline_number):
-        if build_number:
-            return f"{self._baseurl}/primait/{repo}/{build_number}/{pipeline_number}"
-        return None
-
-    def get_tags_from_builds(self):
-        tags = []
-        # pylint: disable-next=missing-timeout
-        builds = requests.get(
-            f"{self._baseurl}/api/repos/primait/{self._repo}/builds?per_page=100",
-            headers={"Authorization": f"Bearer {self._token}"},
-        ).json()
-
-        for build in builds:
-            if build["event"] == "tag":
-                tags.append(build["ref"].replace("refs/tags/", ""))
-
-        tags = list(dict.fromkeys(tags))
-        tags.sort(key=StrictVersion, reverse=True)
-        return tags
-
-    def get_builds_from_tag(self, tag):
-        try:
-            # pylint: disable-next=missing-timeout
-            builds = requests.get(
-                f"{self._baseurl}/api/repos/primait/{self._repo}/builds?per_page=100&tag={tag}",
-                headers={"Authorization": f"Bearer {self._token}"},
-            ).json()
-
-        except Exception:
-            builds = []
-
-        if isinstance(builds, dict) and builds.get("message") == "Unauthorized":
-            raise Exception("Drone API Error: Unauthorized - run suite-py check")
-
-        return builds
-
-    def get_builds_from_branch(self, branch):
-        try:
-            # pylint: disable-next=missing-timeout
-            builds = requests.get(
-                f"{self._baseurl}/api/repos/primait/{self._repo}/builds?branch={branch}&per_page=100",
-                headers={"Authorization": f"Bearer {self._token}"},
-            ).json()
-
-            return [b for b in builds if b["event"] == "push"]
-        except Exception:
-            return []
-
-    def get_build_number_from_tag(self, tag):
-        attempts = 0
-        while attempts < 3:
-
-            try:
-                # pylint: disable-next=missing-timeout
-                builds = requests.get(
-                    f"{self._baseurl}/api/repos/primait/{self._repo}/builds?per_page=100&tag={tag}",
-                    headers={"Authorization": f"Bearer {self._token}"},
-                ).json()
-
-                for build in builds:
-                    if build["event"] == "tag":
-                        return build["number"]
-            except Exception:
-                pass
-
-            time.sleep(2)
-            attempts += 1
-
-        return None
-
-    def promote(self, build, target, querystring):
-        try:
-            # pylint: disable-next=missing-timeout
-            return requests.post(
-                f"{self._baseurl}/api/repos/primait/{self._repo}/builds/{build}/promote?target={target}&{querystring}",
-                headers={"Authorization": f"Bearer {self._token}"},
-            ).json()
-        except Exception as e:
-            logger.error(f"Unable to trigger drone build, exiting with error: {e}")
-            sys.exit(-1)
-
-    def promote_staging(self, build, target, querystring):
-        return self.promote(build, target, querystring)
-
-    def promote_production(self, tag, target, querystring):
-        build = self.get_build_number_from_tag(tag)
-        if build:
-            return self.promote(build, target, querystring)
-
-        logger.error(
-            f"Unable to retrieve drone build for repo {self._repo} with tag {tag}"
-        )
-        sys.exit(-1)
-
-    def launch_build(self, build):
-        # pylint: disable-next=missing-timeout
-        return requests.post(
-            f"{self._baseurl}/api/repos/primait/{self._repo}/builds/{build}",
-            headers={"Authorization": f"Bearer {self._token}"},
-        ).json()
-
-    def prestart_success(self, build_number):
-        with Halo(text="Contacting drone...", spinner="dots", color="magenta"):
-            tries = 10
-            while build_number and tries > 0:
-                tries -= 1
-                # pylint: disable-next=missing-timeout
-                build_status = requests.get(
-                    f"{self._baseurl}/api/repos/primait/{self._repo}/builds/{build_number}",
-                    headers={"Authorization": f"Bearer {self._token}"},
-                ).json()
-
-                steps = build_status["stages"][0]["steps"]
-
-                for step in steps:
-                    if step["name"] == "pre-start":
-                        if step["status"] == "success":
-                            return True
-                        break
-                time.sleep(3)
-            return False
-
-    def fmt(self):
-        try:
-            subprocess.run(
-                ["drone", "fmt", "--save", ".drone.yml"], cwd=self._path, check=True
-            )
-        except CalledProcessError as e:
-            logger.error(f"{self._repo}: unable to format the .drone.yml {e}")
-            sys.exit(-1)
-
-    def validate(self):
-        try:
-            subprocess.run(
-                ["drone", "lint", "--trusted", ".drone.yml"], cwd=self._path, check=True
-            )
-        except CalledProcessError as e:
-            logger.error(f"{self._path}: the .drone.yml is not valid {e}")
-            sys.exit(-1)
-
-    def sign(self):
-        try:
-            subprocess.run(
-                ["drone", "sign", f"primait/{self._repo}", "--save"],
-                cwd=self._path,
-                check=True,
-            )
-        except CalledProcessError as e:
-            logger.error(f"{self._repo}: unable to sign the .drone.yml {e}")
-            sys.exit(-1)
-
-    def parse_yaml(self):
-        repo_path = os.path.join(self._config.user["projects_home"], self._repo)
-        yaml_path = os.path.join(repo_path, ".drone.yml")
-        try:
-            return yaml.load_all(
-                open(yaml_path, "r", encoding="utf-8"), Loader=yaml.FullLoader
-            )
-        except FileNotFoundError:
-            return None

suite_py/lib/handler/vault_handler.py

@@ -1,28 +0,0 @@
-# -*- encoding: utf-8 -*-
-import os
-import subprocess
-import sys
-from subprocess import CalledProcessError
-
-from suite_py.lib import logger
-
-
-class VaultHandler:
-    def __init__(self, repo, config):
-        self._repo = repo
-        self._projects_home = config.user["projects_home"]
-        self._path = os.path.join(self._projects_home, repo)
-
-    def exec(self, profile, command, additional_args=""):
-        try:
-            c = subprocess.Popen(  # pylint: disable=consider-using-with
-                f"aws-vault exec {profile} {additional_args} -- {command}",
-                stdout=subprocess.PIPE,
-                shell=True,
-                cwd=self._path,
-            )  # .stdout.read()
-            c.wait()
-            return c
-        except CalledProcessError as e:
-            logger.error(f"Error during command execution: {e}")
-            sys.exit(-1)