suite-py 1.41.10__py3-none-any.whl → 1.42.1__py3-none-any.whl

suite_py/lib/handler/captainhook_handler.py

@@ -52,13 +52,6 @@ class CaptainHook:
     def get_users_list(self):
         return self.send_get_request("/users/all")
 
-    def aggregators(self):
-        return self.send_get_request("/cloudflare/aggregators/available")
-
-    def change_aggregator(self, aggregator, qa_address):
-        data = {"aggregator": aggregator, "qa_address": qa_address}
-        return self.send_put_request("/cloudflare/aggregators", data)
-
     def send_metrics(self, metrics):
         self.send_post_request("/suite_py/metrics/", json=metrics).raise_for_status()
 

suite_py/lib/handler/okta_handler.py

@@ -8,6 +8,10 @@ from suite_py.lib.tokens import Tokens
 _SCOPE = "openid offline_access"
 
 
+class OktaError(Exception):
+    pass
+
+
 class Okta:
     def __init__(self, config: Config, tokens: Tokens) -> None:
         self._config = config
@@ -34,8 +38,8 @@ class Okta:
 
         refresh_token = self._get_refresh_token()
         if not isinstance(refresh_token, str):
-            raise Exception(
-                "Invalid okta refresh token. Try logging in with `suite_py login`"
+            raise OktaError(
+                "Invalid okta refresh token. Try logging in with `suite-py login`"
             )
         res = oauth.do_refresh_token(
             self._config.okta["client_id"],
@@ -48,9 +52,9 @@ class Okta:
 
     def _update_tokens(self, tokens: oauth.OAuthTokenResponse) -> str:
         if not tokens.id_token:
-            raise Exception("Okta didn't return a new id_token. This shouldn't happen.")
+            raise OktaError("Okta didn't return a new id_token. This shouldn't happen.")
         if not tokens.refresh_token:
-            raise Exception(
+            raise OktaError(
                 "Okta didn't return a new refresh_token. This shouldn't happen."
             )
 

suite_py/lib/metrics.py

@@ -12,7 +12,7 @@ def _metrics() -> Metrics:
     if _metrics_handler:
         return _metrics_handler
 
-    raise Exception(
+    raise RuntimeError(
         "command_executed called before logger.setup(). This is a bug, please report it"
     )
 

suite_py/lib/oauth.py

@@ -23,6 +23,10 @@ class OAuthTokenResponse:
     expires_in: float
 
 
+class OAuthError(Exception):
+    pass
+
+
 def retrieve_token(base_url, params) -> OAuthTokenResponse:
     url = f"{base_url}/token"
     headers = {
@@ -34,7 +38,7 @@ def retrieve_token(base_url, params) -> OAuthTokenResponse:
     logger.debug(data)
 
     if error := data.get("error_description", None):
-        raise Exception(f"OAuth error: {error}")
+        raise OAuthError(f"OAuth error: {error}")
 
     return OAuthTokenResponse(
         access_token=data["access_token"],
@@ -125,12 +129,12 @@ def authorization_code_flow(
     server.handle_request()
 
     if state != server.received_state:
-        raise Exception(
+        raise OAuthError(
             "Error: session replay or similar attack in progress. Please log out of all connections."
         )
 
     if server.error_message:
-        raise Exception(server.error_message)
+        raise OAuthError(server.error_message)
 
     # Step4: Request tokens: Exchange your authorization_code and code_verifier for tokens.
     body = {

suite_py/lib/tokens.py

@@ -10,7 +10,7 @@ from InquirerPy import prompt
 
 from suite_py.lib import logger
 
-TOKENS = ["github", "youtrack", "drone"]
+TOKENS = ["github", "youtrack"]
 
 
 def platform_has_chmod():
@@ -32,7 +32,7 @@ def should_use_keyring():
             pass
 
         if not test_success:
-            raise Exception("Loaded keyring didn't match the saved test keyring")
+            raise RuntimeError("Loaded keyring didn't match the saved test keyring")
 
         return True
     except Exception as error:
@@ -139,7 +139,7 @@ class Tokens:
         if os.path.exists(self._file_name):
             try:
                 if not platform_has_chmod():
-                    raise Exception("chmod is not supported on this OS")
+                    raise RuntimeError("chmod is not supported on this OS")
 
                 os.chmod(self._file_name, 0o600)
             except Exception as error:
@@ -162,9 +162,5 @@ class Tokens:
     def youtrack(self):
         return self._tokens["youtrack"]
 
-    @property
-    def drone(self):
-        return self._tokens["drone"]
-
     def okta(self):
         return self._tokens.get("okta", {})

{suite_py-1.41.10.dist-info → suite_py-1.42.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: suite-py
-Version: 1.41.10
+Version: 1.42.1
 Summary: 
 Author: larrywax, EugenioLaghi, michelangelomo
 Author-email: devops@prima.it
@@ -13,7 +13,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: Click (>=7.0)
 Requires-Dist: InquirerPy (>=0.2.0)
-Requires-Dist: Jinja2 (>=2.11,<3.0.0)
+Requires-Dist: Jinja2 (>=3.1.4,<4.0.0)
 Requires-Dist: PyGithub (>=1.57)
 Requires-Dist: PyYaml (>=5.4)
 Requires-Dist: autoupgrade-prima (>=0.6)
@@ -21,12 +21,12 @@ Requires-Dist: black (>=22.6,<25.0)
 Requires-Dist: boto3 (>=1.17.84)
 Requires-Dist: cement (>=3.0)
 Requires-Dist: colorama (>=0.4.3)
-Requires-Dist: cryptography (==42.0.7)
+Requires-Dist: cryptography (==43.0.0)
 Requires-Dist: halo (>=0.0.28)
 Requires-Dist: inquirer (==3.1.4)
 Requires-Dist: itsdangerous (==2.2.0)
 Requires-Dist: keyring (>=23.9.1,<26.0.0)
-Requires-Dist: kubernetes (==29.0.0)
+Requires-Dist: kubernetes (==30.1.0)
 Requires-Dist: logzero (==1.7.0)
 Requires-Dist: markupsafe (==2.0.1)
 Requires-Dist: pptree (==3.1)

{suite_py-1.41.10.dist-info → suite_py-1.42.1.dist-info}/RECORD

@@ -1,54 +1,44 @@
 suite_py/__init__.py,sha256=REmi3D0X2G1ZWnYpKs8Ffm3NIj-Hw6dMuvz2b9NW344,142
-suite_py/__version__.py,sha256=i6HmzASsryZBCC4p1Cf_qbUBilep7P848kIqMZUnTNs,50
-suite_py/cli.py,sha256=oGKcY9TUvGgXJl1tYBlkvSCcz6A6Aa7Fye5yEPqB4F8,15159
+suite_py/__version__.py,sha256=2qe7wju_9Za3gMD66khHaby5fLHqRazj7tWdFCm5iWk,49
+suite_py/cli.py,sha256=nQmnOXTLoO12vyH_6d41bd7h8xElRusT_tLGNtYt-Tg,9953
 suite_py/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-suite_py/commands/aggregator.py,sha256=_xyyX7MOMZzDEvzj2AI03OF3sut5PpSIyuRjUiCp5aI,5747
 suite_py/commands/ask_review.py,sha256=yN__Ac-fiZBPShjRDhyCCQZGfVlQE16KozoJk4UtiNw,3788
-suite_py/commands/batch_job.py,sha256=flTSxmNaUv1VcL91Lv-BwEjFpEyF8APTNysQE_2jPYE,7474
 suite_py/commands/bump.py,sha256=oFZU1hPfD11ujFC5G7wFyQOf2alY3xp2SO1h1ldjf3s,5406
-suite_py/commands/check.py,sha256=0e2NsPi3cqvCwtNYzhR1UroT59bCojLlGo69vv3FiOA,4074
+suite_py/commands/check.py,sha256=jCX59g6DgTA55yD_75mgcqJ5zCjRwl_eIRGDeUFjUWY,3316
 suite_py/commands/common.py,sha256=aWCEvO3hqdheuMUmZcHuc9EGZPQTk7VkzkHJk283MxQ,566
 suite_py/commands/context.py,sha256=6dssSqoABPb9gLmL3Y7W2iAGxrY_oma665X8zrSkNNQ,1079
 suite_py/commands/create_branch.py,sha256=cDpeDsQk1AK70GkWz-hTduaEeU65x1wck1b-5nKIMew,4424
-suite_py/commands/deploy.py,sha256=kadgbVKUMtE_X4b8oWaQ1ufS4Qy9b2WuutPDXnjm4t4,8088
-suite_py/commands/docker.py,sha256=POz_VXOXEQaFZCafkH-grgB2_HZFrckAc0CpB9IgOiU,2932
-suite_py/commands/generator.py,sha256=-wQFRS0UNc-EvuYvnj3gk6DHTVSsg9lA-NMU2kQewb8,8510
-suite_py/commands/id.py,sha256=qMQMSH_bGDInarYaGOpX2lEGf1tyHBeAV5GQiNr5Kiw,1998
-suite_py/commands/ip.py,sha256=pbyVuee_cN507qUYSBv5gWcvKLYolOeuU_w_P7P7nVc,2396
 suite_py/commands/login.py,sha256=A59e1HsbN7Ocv2L_2H0Eb7MZK7AzLkLb72QxBthnIqU,258
-suite_py/commands/merge_pr.py,sha256=gUpoDx3q23X6gF9PLXCZnIL9DfRw_3D0LlqBlGVt7rA,5676
-suite_py/commands/open_pr.py,sha256=U7MVl-JFKu1mdfxC_UvxUHtPLEln0g4kKl-SvP-6zd8,7159
+suite_py/commands/merge_pr.py,sha256=fXIE8mT9MjvvpqE-uVdXGBVFGhn0eQzcBxNr-N8SyAY,5171
+suite_py/commands/open_pr.py,sha256=e6IT6f8L_HKsOEtrGHZQ2HWrFwmvIgXGNCK_TU0WE9k,7009
 suite_py/commands/project_lock.py,sha256=b7OkGysue_Sl13VIT7B5CTBppCvrB_Q6iC0IJRBSHp8,1909
-suite_py/commands/release.py,sha256=clKgmsNgR9DdgBkYjXI3NqVaw8_mCe2TZHegby3ESn4,16634
-suite_py/commands/secret.py,sha256=Us7wvsqAPJhe1nUSHOmzO1RSyZriKAGKwi2YEhPNIA4,7976
+suite_py/commands/release.py,sha256=Mh6esivJ234cRE5mYWPOZSgpujO47RprySh0U_9BD9s,7857
 suite_py/commands/set_token.py,sha256=fehIqKjKhE-BJGFhgkPTo3Ntr0MvpgLd6EC5yjKuRs8,1508
 suite_py/commands/status.py,sha256=0JUK53_d1-U3WNS742JD2QTiGmCGZONo3jJx8WR7q70,1122
 suite_py/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-suite_py/lib/config.py,sha256=4uKXAav8E-VXlXGPnkYQ_fZYyi6058S0FM2JmyV8L2k,3970
+suite_py/lib/config.py,sha256=52YV0K0GK1WRbeM70IDXkpG8oZD1Zvxipn1b-XekwNA,3907
 suite_py/lib/handler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 suite_py/lib/handler/aws_handler.py,sha256=dRvRDicikfRbuFtCLPbevaX-yC-fO4LwXFdyqLPJ8OI,8815
-suite_py/lib/handler/captainhook_handler.py,sha256=xZ9NueGfh14L46S8AWSCZAGJdOFUCa2cFbbguq5SVv8,3281
+suite_py/lib/handler/captainhook_handler.py,sha256=sp1dxZXO1QJaq304JhtkTrQjn8Z4971e2uFTM7Dh5wg,2983
 suite_py/lib/handler/changelog_handler.py,sha256=-ppnRl3smBA_ys8tPqXmytS4eyntlwfawC2fiXFcwlw,4818
-suite_py/lib/handler/drone_handler.py,sha256=rmtzu30OQyG3vRPlbZKsQhHN9zbguPtXO0RpDjYOTPA,8967
 suite_py/lib/handler/frequent_reviewers_handler.py,sha256=EIJX5FEMWzrxuXS9A17hu1vfxgJSOHSBX_ahCEZ2FVA,2185
 suite_py/lib/handler/git_handler.py,sha256=boxhl9lQz6fjEJ10ib1KrDW-geCVjhA_6nKwv2ll01g,11333
 suite_py/lib/handler/github_handler.py,sha256=AnFL54yOZ5GDIU91wQat4s-d1WTcmg_B_5M7-Rop3wA,2900
 suite_py/lib/handler/metrics_handler.py,sha256=-Tp62pFIiYsBkDga0nQG3lWU-gxH68wEjIIIJeU1jHk,3159
-suite_py/lib/handler/okta_handler.py,sha256=3GEnJxbLXlu2zjFWniYwG0m1mFKJGfske1t-uUUdpZU,2545
+suite_py/lib/handler/okta_handler.py,sha256=QOt7fhCy-1Pw_tQ0Psfuy2fXP-gbxvJ1zVNmrGvwRXw,2584
 suite_py/lib/handler/prompt_utils.py,sha256=vgk1O7h-iYEAZv1sXtMh8xIgH1djI398rzxRIgZWZcg,2474
-suite_py/lib/handler/vault_handler.py,sha256=r4osw7qwz3ZFmLg2U1oFPdtRFcXzDXiaWBZC01cYK_w,871
 suite_py/lib/handler/version_handler.py,sha256=DXTx4yCAbFVC6CdMqPJ-LiN5YM-dT2zklG8POyKTP5A,6774
 suite_py/lib/handler/youtrack_handler.py,sha256=eTGBBXjlN_ay1cawtnZ2IG6l78dDyKdMN1x6PxcvtA0,7499
 suite_py/lib/logger.py,sha256=q_qRtpg0Eh7r5tB-Rwz87dnWBwP-a2dIvggkiQikH8M,782
-suite_py/lib/metrics.py,sha256=hGGrWg_c3uTz4xhpb7POGZh_xhIAYU3drRo-KgwpBvM,1626
-suite_py/lib/oauth.py,sha256=Y_I8a9wDKOVQ75xJH5ECDRH4xrJHSPVPUvVVM2fTQdA,5014
+suite_py/lib/metrics.py,sha256=CqAIwPIRm0AJR4qmCTj1rw-NCwdSL4Huj2eKhLxgLgU,1629
+suite_py/lib/oauth.py,sha256=NOX96coz_pb10C4pZ2Sk33beJQTTi_N-LXPUjDYuRKM,5057
 suite_py/lib/requests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 suite_py/lib/requests/auth.py,sha256=wN_WtGFmDUWRFilWzOmUaRBvP2n3EPpPMqex9Zjddko,228
 suite_py/lib/requests/session.py,sha256=P32H3cWnCWunu91WIj2iDM5U3HzaBglg60VN_C9JBL4,267
 suite_py/lib/symbol.py,sha256=z3QYBuNIwD3qQ3zF-cLOomIr_-C3bO_u5UIDAHMiyTo,60
-suite_py/lib/tokens.py,sha256=kK4vatd5iKEFaue0BZwK1f66YOh9zLdLzP41ZOhjMCU,5534
+suite_py/lib/tokens.py,sha256=4DbsHDFLIxs40t3mRw_ZyhmejZQ0Bht7iAL8dTCTQd4,5458
 suite_py/templates/login.html,sha256=fJLls2SB84oZTSrxTdA5q1PqfvIHcCD4fhVWfyco7Ig,861
-suite_py-1.41.10.dist-info/METADATA,sha256=A_GGYDs7Uq5qzCIiUW-PKqxk0QdgS4su2FMwpKWc4A4,1533
-suite_py-1.41.10.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-suite_py-1.41.10.dist-info/entry_points.txt,sha256=dVKLC-9Infy-dHJT_MkK6LcDjOgBCJ8lfPkURJhBjxE,46
-suite_py-1.41.10.dist-info/RECORD,,
+suite_py-1.42.1.dist-info/METADATA,sha256=eotAZFpnIsQMp5sbP7CK4YwP-m5DjWrzpkDA75KdYMs,1533
+suite_py-1.42.1.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+suite_py-1.42.1.dist-info/entry_points.txt,sha256=dVKLC-9Infy-dHJT_MkK6LcDjOgBCJ8lfPkURJhBjxE,46
+suite_py-1.42.1.dist-info/RECORD,,

suite_py/commands/aggregator.py

@@ -1,159 +0,0 @@
-# -*- coding: utf-8 -*-
-import re
-import sys
-
-from halo import Halo
-from kubernetes import client, config
-from rich.console import Console
-from rich.table import Table
-
-from suite_py.lib import logger, metrics
-from suite_py.lib.handler import prompt_utils
-
-
-class Aggregator:
-    def __init__(self, captainhook, command):
-        self._captainhook = captainhook
-        self._command = command
-
-    @metrics.command("aggregator")
-    def run(self):
-        if self._command == "list":
-            self._list_aggregators()
-            return
-
-        if self._command == "change":
-            aggregator = self._select_aggregator()
-            address = prompt_utils.ask_questions_input(
-                "Insert QA url or press enter to set staging URL: ",
-                default_text="staging.prima.it",
-            )
-
-            if address.startswith("http"):
-                address = re.sub(r"https?://?", "", address)
-
-            split_addr = re.split(r"([a-z]+)\-([a-z0-9]+)", address)
-
-            change_request = self._captainhook.change_aggregator(
-                aggregator["name"], address
-            )
-
-            self._handle_captainhook_response(
-                change_request, aggregator["name"], address
-            )
-
-            if "www" in split_addr:
-                split_addr[1] = "prima"
-                self._update_k8s_ingress_route(aggregator, split_addr[1], split_addr[2])
-            elif "evvivass" in split_addr:
-                self._update_k8s_ingress_route(aggregator, split_addr[1], split_addr[2])
-            else:
-                return
-
-    def _handle_captainhook_response(self, request, aggregator, address):
-        if request.status_code == 200:
-            change_request = request.json()
-            if change_request["success"]:
-                logger.info(f"CNAME updated! Now {aggregator} is pointing to {address}")
-            else:
-                cases = {
-                    "cloudflare_error": "Error during Cloudflare invocation.",
-                    "unknown_dns_record": "Impossible to find DNS record associated with aggregator.",
-                    "unknown_aggregator": "Aggregator not found.",
-                    "invalid_qa_address": "The QA address does not meet the requirements.",
-                }
-                logger.error(cases.get(change_request["error"], "unknown error"))
-                sys.exit(-1)
-        else:
-            logger.error("An error has occurred on Captainhook.")
-            sys.exit(-1)
-
-    def _select_aggregator(self):
-        with Halo(text="Loading aggregators...", spinner="dots", color="magenta"):
-            choices = [
-                {"name": agg["name"], "value": agg}
-                for agg in self._captainhook.aggregators().json()
-            ]
-        if choices:
-            choices.sort(key=lambda x: x["name"])
-            return prompt_utils.ask_choices("Select aggregator: ", choices)
-
-        logger.error("There are no aggregators on Captainhook.")
-        sys.exit(-1)
-
-    def _list_aggregators(self):
-        with Halo(text="Loading...", spinner="dots", color="magenta"):
-            aggregators = self._captainhook.aggregators()
-
-        if aggregators.status_code != 200:
-            logger.error("Unable to retrieve the list of aggregators.")
-            return
-
-        console = Console()
-
-        aggregators_table = Table()
-        aggregators_table.add_column("Name", style="green")
-        aggregators_table.add_column("Address", style="white")
-
-        for a in aggregators.json():
-            aggregators_table.add_row(a["name"], a["content"])
-
-        logger.info("Available aggregators:")
-        console.print(aggregators_table)
-
-    def _update_k8s_ingress_route(self, aggregator, service_name, namespace):
-        try:
-            config.load_kube_config()
-        except Exception:
-            logger.error(
-                "\n\nYou need to authenticate, run the following command:\n$ aws eks update-kubeconfig --name main-qa\n\n"
-            )
-            sys.exit(-1)
-
-        api = client.CustomObjectsApi()
-        res = {
-            "apiVersion": "traefik.containo.us/v1alpha1",
-            "kind": "IngressRoute",
-            "metadata": {
-                "name": f"{aggregator['name']}",
-                "namespace": f"{namespace}",
-                "annotations": {
-                    "traefik.ingress.kubernetes.io/router.tls": "true",
-                },
-                "labels": {
-                    "com.prima.environment": "qa",
-                    "com.prima.country": "it",
-                    "app.kubernetes.io/name": f"{aggregator['name']}",
-                },
-            },
-            "spec": {
-                "entryPoints": ["websecure"],
-                "routes": [
-                    {
-                        "kind": "Rule",
-                        "match": f"Host(`{aggregator['address']}{aggregator['domain']}`)",
-                        "services": [{"name": f"{service_name}", "port": 80}],
-                    }
-                ],
-            },
-        }
-
-        try:
-            api.create_namespaced_custom_object(
-                group="traefik.containo.us",
-                version="v1alpha1",
-                namespace=namespace,
-                plural="ingressroutes",
-                body=res,
-            )
-            logger.info(f"Aggregator {aggregator['name']} created!")
-        except Exception:
-            api.patch_namespaced_custom_object(
-                group="traefik.containo.us",
-                version="v1alpha1",
-                namespace=namespace,
-                plural="ingressroutes",
-                name=aggregator["name"],
-                body=res,
-            )
-            logger.info(f"Aggregator {aggregator['name']} updated!")

suite_py/commands/batch_job.py

@@ -1,215 +0,0 @@
-# -*- coding: utf-8 -*-
-import re
-import sys
-import textwrap
-
-import semver
-
-from suite_py.lib import logger, metrics
-from suite_py.lib.handler import prompt_utils
-from suite_py.lib.handler.drone_handler import DroneHandler
-from suite_py.lib.handler.git_handler import GitHandler
-from suite_py.lib.handler.github_handler import GithubHandler
-
-
-class BatchJob:
-    def __init__(
-        self, project, config, tokens, environment, cpu_request, memory_request
-    ):
-        self._project = project
-        self._config = config
-        self._tokens = tokens
-        self._environment = environment
-        self._cpu_request = cpu_request
-        self._memory_request = memory_request
-        self._git = GitHandler(project, config)
-        self._github = GithubHandler(tokens)
-        self._drone = DroneHandler(config, tokens, repo=project)
-        self._countries = _parse_available_countries(self._drone)
-        self._repo = self._github.get_repo(project)
-
-    @metrics.command("batch-job")
-    def run(self):
-        country = prompt_utils.ask_choices(
-            "Which country do you want to run the job on?",
-            self._countries,
-        )
-
-        if not _has_job_pipeline(self._drone, country, self._environment):
-            logger.error(
-                "There is no job pipeline for the specified country/environment"
-            )
-            sys.exit(1)
-
-        command = prompt_utils.ask_questions_input(
-            "What command do you want to execute?", f"/app/bin/{self._project} "
-        )
-
-        self._start_job(country, command)
-
-    def _start_job(self, country, command):
-        promotion = {}
-
-        if self._environment == "staging":
-            build = self._get_latest_master_build()
-
-            self._ask_confirm(
-                country,
-                command,
-                f'Current master: {build["message"]} (#{build["number"]})',
-            )
-
-            promotion = self._drone.promote_staging(
-                build["number"],
-                f"job-{country}-{self._environment}",
-                f"BATCH_COMMAND={command}&JOB_CPU={self._cpu_request}&JOB_MEMORY={self._memory_request}",
-            )
-        elif self._environment == "production":
-            version = self._get_latest_tag(country)
-
-            self._ask_confirm(country, command, f"Current tag: {version}")
-
-            promotion = self._drone.promote_production(
-                version,
-                f"job-{country}-{self._environment}",
-                f"DRONE_TAG={version}&BATCH_COMMAND={command}&JOB_CPU={self._cpu_request}&JOB_MEMORY={self._memory_request}",
-            )
-
-        if "number" not in promotion:
-            logger.warning(f"Unable to promote drone build. Response: {promotion}")
-            return
-
-        logger.info("Drone build started successfully!")
-        logger.info(
-            f"You can follow the build status here: {self._drone.get_build_url(promotion['number'])}"
-        )
-
-    def _get_latest_master_build(self):
-        try:
-            builds = self._drone.get_builds_from_branch("master")
-
-            latest_build = None
-            non_green_builds = []
-            for b in builds:
-                if b["status"] == "success":
-                    if latest_build is None:
-                        latest_build = b
-                    break
-
-                non_green_builds.append(b)
-
-            if len(non_green_builds) > 0:
-                logger.warning(
-                    "There are recent builds on master still running or failed"
-                )
-                for b in non_green_builds:
-                    logger.warning(
-                        f'* {b["message"]} (Status: {b["status"]}, Number: #{b["number"]})'
-                    )
-
-            if not latest_build:
-                logger.error("Unable to find latest build on master")
-                sys.exit(255)
-
-            return latest_build
-
-        except Exception as e:
-            print(e)
-            logger.error(
-                "An error has occurred retrieving current master version during batch job.\nPlease ask #team-platform-operations for help."
-            )
-            sys.exit(255)
-
-    def _get_latest_tag(self, country):
-        try:
-            logger.info("Retrieving latest version, this may take some time...")
-            # get latest 10 tags
-            tags = self._github.get_tags(self._project)
-            #  exclude tags that don't match semver notation
-            semver_tags = [t for t in tags if semver.VersionInfo.isvalid(t.name)][0:9]
-
-            builds = []
-            for tag in semver_tags:
-                for b in self._drone.get_builds_from_tag(tag.name):
-                    if (
-                        b["event"] == "promote"
-                        and b["status"] == "success"
-                        and b["deploy_to"] == f"deploy-{country}-production"
-                        and "params" in b
-                        and "DRONE_TAG" in b["params"]
-                    ):
-                        builds.append(b)
-
-            if not builds:
-                # may end up here if there isn't any successfull build for latest 10 tags
-                logger.error(
-                    "An error has occurred retrieving current version during rollback.\nPlease ask #team-platform-operations for help."
-                )
-                sys.exit(255)
-
-            # get latest build using build number as key
-            current_build = max(builds, key=lambda x: x["number"])
-
-            if not current_build:
-                logger.error(
-                    f"Unable to determine current version for country {country}"
-                )
-                sys.exit(255)
-
-            v = semver.VersionInfo.parse(current_build["params"]["DRONE_TAG"])
-            logger.info(f"Current version for country {country}: {v}")
-
-            return v
-        except Exception as e:
-            print(e)
-            logger.error(
-                "An error has occurred retrieving current version during batch job.\nPlease ask #team-platform-operations for help."
-            )
-            sys.exit(255)
-
-    def _ask_confirm(self, country, command, message):
-        logger.info(
-            textwrap.dedent(
-                f"""
-                You're about to run a batch job on project {self._project}, for env {self._environment}, in country {country}.
-                {message}
-                Command: {command}
-                """
-            )
-        )
-
-        if not prompt_utils.ask_confirm("Do you confirm?", default=False):
-            sys.exit(0)
-
-
-def _parse_available_countries(drone):
-    pipelines = drone.parse_yaml()
-
-    if pipelines is None:
-        logger.error("The file .drone.yml was not found. Unable to continue.")
-        sys.exit(1)
-
-    countries = []
-    REGEX = re.compile(r"deploy-([a-z]+)-.*")
-    for pipeline in pipelines:
-        if "name" in pipeline:
-            c = REGEX.findall(pipeline["name"])
-            if len(c) > 0 and c[0] is not None and c[0] not in countries:
-                countries.append(c[0])
-
-    return countries
-
-
-def _has_job_pipeline(drone, country, environment):
-    pipelines = drone.parse_yaml()
-
-    if pipelines is None:
-        logger.error("The file .drone.yml was not found. Unable to continue.")
-        sys.exit(1)
-
-    job_pipeline = f"job-{country}-{environment}"
-    for pipeline in pipelines:
-        if "name" in pipeline and pipeline["name"] == job_pipeline:
-            return True
-
-    return False