suite-py 1.41.3__tar.gz → 1.41.4__tar.gz

{suite_py-1.41.3 → suite_py-1.41.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: suite-py
-Version: 1.41.3
+Version: 1.41.4
 Summary: 
 Author: larrywax, EugenioLaghi, michelangelomo
 Author-email: devops@prima.it
@@ -12,12 +12,10 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: Click (>=7.0)
-Requires-Dist: Flask (==1.1.2)
 Requires-Dist: InquirerPy (>=0.2.0)
 Requires-Dist: Jinja2 (>=2.11,<3.0.0)
 Requires-Dist: PyGithub (>=1.57)
 Requires-Dist: PyYaml (>=5.4)
-Requires-Dist: Werkzeug (==2.0.2)
 Requires-Dist: autoupgrade-prima (>=0.6)
 Requires-Dist: black (>=22.6,<25.0)
 Requires-Dist: boto3 (>=1.17.84)
@@ -27,7 +25,7 @@ Requires-Dist: cryptography (==42.0.5)
 Requires-Dist: halo (>=0.0.28)
 Requires-Dist: inquirer (==3.1.4)
 Requires-Dist: itsdangerous (==2.0.1)
-Requires-Dist: keyring (>=23.9.1,<25.0.0)
+Requires-Dist: keyring (>=23.9.1,<26.0.0)
 Requires-Dist: kubernetes (==29.0.0)
 Requires-Dist: logzero (==1.7.0)
 Requires-Dist: markupsafe (==2.0.1)

{suite_py-1.41.3 → suite_py-1.41.4}/pyproject.toml

@@ -2,16 +2,14 @@
 authors = ["larrywax, EugenioLaghi, michelangelomo <devops@prima.it>"]
 description = ""
 name = "suite-py"
-version = "1.41.3"
+version = "1.41.4"
 
 [tool.poetry.dependencies]
 Click = ">=7.0"
-Flask = "==1.1.2"
 InquirerPy = ">=0.2.0"
 Jinja2 = ">=2.11,<3.0.0"
 PyGithub = ">=1.57"
 PyYaml = ">=5.4"
-Werkzeug = "==2.0.2"
 autoupgrade-prima = ">=0.6"
 black = ">=22.6,<25.0"
 boto3 = ">=1.17.84"
@@ -20,7 +18,7 @@ colorama = ">=0.4.3"
 halo = ">=0.0.28"
 inquirer = "==3.1.4"
 itsdangerous = "==2.0.1"
-keyring = ">=23.9.1,<25.0.0"
+keyring = ">=23.9.1,<26.0.0"
 kubernetes = "==29.0.0"
 logzero = "==1.7.0"
 markupsafe = "==2.0.1"

{suite_py-1.41.3 → suite_py-1.41.4}/suite_py/__version__.py

@@ -1,2 +1,2 @@
 # -*- encoding: utf-8 -*-
-__version__ = "1.41.3"
+__version__ = "1.41.4"

{suite_py-1.41.3 → suite_py-1.41.4}/suite_py/cli.py

@@ -4,27 +4,29 @@
 # for performance reasons, so turn off the lint warning
 # pylint: disable=import-outside-toplevel
 
-from functools import wraps
 import os
 import sys
+from functools import wraps
 from typing import Optional
 
 import click
-from click.exceptions import ClickException
 import pkg_resources
 import requests
 from autoupgrade import Package
+from click.exceptions import ClickException
 
 from suite_py.__version__ import __version__
-from suite_py.lib import logger
-from suite_py.lib import metrics
+from suite_py.commands.context import Context
+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.captainhook_handler import CaptainHook
+from suite_py.lib.handler.okta_handler import Okta
 from suite_py.lib.tokens import Tokens
 
-ALLOW_NO_GIT_SUBCOMMAND = ["login", "qa", "aggregator"]
-ALLOW_NO_HOME_SUBCOMMAND = ["login", "qa", "aggregator"]
+ALLOW_NO_GIT_SUBCOMMAND = ["login", "aggregator"]
+ALLOW_NO_HOME_SUBCOMMAND = ["login", "aggregator"]
 
 
 def maybe_inject_truststore() -> None:
@@ -121,7 +123,6 @@ def main(ctx, project, timeout, verbose):
     config = Config()
 
     logger.setup(verbose)
-    metrics.setup(config)
 
     logger.debug(f"v{__version__}")
     maybe_inject_truststore()
@@ -156,20 +157,30 @@ def main(ctx, project, timeout, verbose):
         f"Do you want to continue on project {os.path.basename(project)}?"
     ):
         return
-
-    ctx.ensure_object(dict)
-    ctx.obj["project"] = os.path.basename(project)
     if timeout:
         config.user["captainhook_timeout"] = timeout
-    ctx.obj["config"] = config
-    ctx.obj["tokens"] = Tokens()
+
+    project = os.path.basename(project)
+    tokens = Tokens()
+    okta = Okta(config, tokens)
+    captainhook = CaptainHook(config, okta)
+
+    ctx.obj = Context(
+        project=project,
+        tokens=tokens,
+        okta=okta,
+        config=config,
+        captainhook=captainhook,
+    )
+
+    ctx.obj.call(metrics.setup)
 
     # Skip chdir if not needed
     if (
         ctx.invoked_subcommand not in ALLOW_NO_GIT_SUBCOMMAND
         or ctx.invoked_subcommand not in ALLOW_NO_HOME_SUBCOMMAND
     ):
-        os.chdir(os.path.join(config.user["projects_home"], ctx.obj["project"]))
+        os.chdir(os.path.join(config.user["projects_home"], ctx.obj.project))
 
 
 @main.result_callback()
@@ -188,12 +199,10 @@ def cleanup(_obj, _, **_kwargs):
 )
 @click.pass_obj
 @catch_exceptions
-def bump(obj, project: Optional[str] = None, version: Optional[str] = None):
+def bump(obj: Context, project: Optional[str] = None, version: Optional[str] = None):
     from suite_py.commands.bump import Bump
 
-    Bump(obj["project"], obj["config"], obj["tokens"]).run(
-        project=project, version=version
-    )
+    obj.call(Bump).run(project=project, version=version)
 
 
 @main.command(
@@ -205,7 +214,7 @@ def bump(obj, project: Optional[str] = None, version: Optional[str] = None):
 def cli_create_branch(obj, card):
     from suite_py.commands.create_branch import CreateBranch
 
-    CreateBranch(obj["project"], card, obj["config"], obj["tokens"]).run()
+    obj.call(CreateBranch, card=card).run()
 
 
 @main.command("lock", help="Lock project on staging or prod")
@@ -217,7 +226,7 @@ def cli_create_branch(obj, card):
 def cli_lock_project(obj, environment):
     from suite_py.commands.project_lock import ProjectLock
 
-    ProjectLock(obj["project"], environment, "lock", obj["config"], obj["tokens"]).run()
+    obj.call(ProjectLock, environment=environment, action="lock").run()
 
 
 @main.command("unlock", help="Unlock project on staging or prod")
@@ -229,9 +238,7 @@ def cli_lock_project(obj, environment):
 def cli_unlock_project(obj, environment):
     from suite_py.commands.project_lock import ProjectLock
 
-    ProjectLock(
-        obj["project"], environment, "unlock", obj["config"], obj["tokens"]
-    ).run()
+    obj.call(ProjectLock, environment=environment, action="unlock").run()
 
 
 @main.command("open-pr", help="Open a PR on GitHub")
@@ -239,8 +246,10 @@ def cli_unlock_project(obj, environment):
 @catch_exceptions
 def cli_open_pr(obj):
     from suite_py.commands.open_pr import OpenPR
+    from suite_py.commands.ask_review import AskReview
 
-    OpenPR(obj["project"], obj["config"], obj["tokens"]).run()
+    ask_review = obj.call(AskReview)
+    obj.call(OpenPR, ask_review=ask_review).run()
 
 
 @main.command("ask-review", help="Requests a PR review")
@@ -249,7 +258,7 @@ def cli_open_pr(obj):
 def cli_ask_review(obj):
     from suite_py.commands.ask_review import AskReview
 
-    AskReview(obj["project"], obj["config"], obj["tokens"]).run()
+    obj.call(AskReview).run()
 
 
 @main.command(
@@ -260,7 +269,7 @@ def cli_ask_review(obj):
 def cli_merge_pr(obj):
     from suite_py.commands.merge_pr import MergePR
 
-    MergePR(obj["project"], obj["config"], obj["tokens"]).run()
+    obj.call(MergePR).run()
 
 
 @main.group("release", help="Manage releases")
@@ -281,13 +290,7 @@ def release():
 def cli_release_create(obj, deploy):
     from suite_py.commands.release import Release
 
-    Release(
-        "create",
-        obj["project"],
-        obj["config"],
-        obj["tokens"],
-        flags={"deploy": deploy},
-    ).run()
+    obj.call(Release, action="create", flags={"deploy": deploy}).run()
 
 
 @release.command("deploy", help="Deploy a github release with Drone CI")
@@ -296,7 +299,7 @@ def cli_release_create(obj, deploy):
 def cli_release_deploy(obj):
     from suite_py.commands.release import Release
 
-    Release("deploy", obj["project"], obj["config"], obj["tokens"]).run()
+    obj.call(Release, action="deploy").run()
 
 
 @release.command("rollback", help="Rollback a deployment")
@@ -305,7 +308,7 @@ def cli_release_deploy(obj):
 def cli_release_rollback(obj):
     from suite_py.commands.release import Release
 
-    Release("rollback", obj["project"], obj["config"], obj["tokens"]).run()
+    obj.call(Release, "rollback").run()
 
 
 @main.command("deploy", help="Deploy master branch in production")
@@ -314,7 +317,7 @@ def cli_release_rollback(obj):
 def cli_deploy(obj):
     from suite_py.commands.deploy import Deploy
 
-    Deploy(obj["project"], obj["config"], obj["tokens"]).run()
+    obj.call(Deploy).run()
 
 
 @main.group("docker", help="Manage docker images")
@@ -328,7 +331,7 @@ def docker():
 def cli_docker_release(obj):
     from suite_py.commands.docker import Docker
 
-    Docker("release", obj["project"], obj["config"], obj["tokens"]).run()
+    obj.call(Docker, action="release").run()
 
 
 @docker.command("versions", help="List all available versions of specific image")
@@ -337,7 +340,7 @@ def cli_docker_release(obj):
 def cli_docker_versions(obj):
     from suite_py.commands.docker import Docker
 
-    Docker("versions", obj["project"], obj["config"], obj["tokens"]).run()
+    obj.call(Docker, action="versions").run()
 
 
 @main.command("status", help="Current status of a project")
@@ -346,7 +349,7 @@ def cli_docker_versions(obj):
 def cli_status(obj):
     from suite_py.commands.status import Status
 
-    Status(obj["project"], obj["config"]).run()
+    obj.call(Status, action="versions").run()
 
 
 @main.command("check", help="Verify authorisations for third party services")
@@ -355,7 +358,7 @@ def cli_status(obj):
 def cli_check(obj):
     from suite_py.commands.check import Check
 
-    Check(obj["config"], obj["tokens"]).run()
+    obj.call(Check).run()
 
 
 @main.command("id", help="Get the ID of the hosts where the task is running")
@@ -365,7 +368,7 @@ def cli_check(obj):
 def cli_id(obj, environment):
     from suite_py.commands.id import ID
 
-    ID(obj["project"], obj["config"], environment).run()
+    obj.call(ID, environment=environment).run()
 
 
 @main.command("ip", help="Get the IP addresses of the hosts where the task is running")
@@ -375,7 +378,7 @@ def cli_id(obj, environment):
 def cli_ip(obj, environment):
     from suite_py.commands.ip import IP
 
-    IP(obj["project"], obj["config"], environment).run()
+    obj.call(IP, environment=environment).run()
 
 
 @main.command("generator", help="Generate different files from templates")
@@ -384,7 +387,7 @@ def cli_ip(obj, environment):
 def cli_generator(obj):
     from suite_py.commands.generator import Generator
 
-    Generator(obj["project"], obj["config"], obj["tokens"]).run()
+    obj.call(Generator).run()
 
 
 @main.group(
@@ -415,7 +418,7 @@ def aggregator(ctx, show_list, change):
 def cli_aggregator_list(obj):
     from suite_py.commands.aggregator import Aggregator
 
-    Aggregator(obj["config"], "list").run()
+    obj.call(Aggregator, action="list").run()
 
 
 @aggregator.command("change", help="Change aggregator record")
@@ -424,7 +427,7 @@ def cli_aggregator_list(obj):
 def cli_aggregator_change(obj):
     from suite_py.commands.aggregator import Aggregator
 
-    Aggregator(obj["config"], "change").run()
+    obj.call(Aggregator, action="list").run()
 
 
 @main.command("login", help="manage login against Auth0")
@@ -433,144 +436,7 @@ def cli_aggregator_change(obj):
 def login(obj):
     from suite_py.commands.login import Login
 
-    Login(obj["config"]).run()
-
-
-@main.group("qa", help="Manage QA envs")
-def qa():
-    pass
-
-
-@qa.command("update-quota", help="Update quota in QA for a user")
-@click.pass_obj
-@catch_exceptions
-def cli_qa_update_quota(obj):
-    from suite_py.commands.qa import QA
-
-    QA("update-quota", obj["project"], obj["config"], obj["tokens"]).run()
-
-
-@qa.command("list", help="List QA envs for user: all to show qa of all users.")
-@click.option("-u", "--user", "user", required=False)
-@click.option("-s", "--status", "status", multiple=True, type=str)
-@click.option("-c", "--card", "card", type=str)
-@click.pass_obj
-@catch_exceptions
-def cli_qa_list(obj, user, status, card):
-    from suite_py.commands.qa import QA
-
-    QA(
-        "list",
-        obj["project"],
-        obj["config"],
-        obj["tokens"],
-        {"user": user, "status": status, "card": card},
-    ).run()
-
-
-@qa.command("create", help="Create QA env")
-@click.argument("microservices", nargs=-1, required=True)
-@click.pass_obj
-@catch_exceptions
-def cli_qa_create(obj, microservices):
-    from suite_py.commands.qa import QA
-
-    QA(
-        "create",
-        obj["project"],
-        obj["config"],
-        obj["tokens"],
-        {"services": microservices},
-    ).run()
-
-
-@qa.command("update", help="Update QA env")
-@click.argument("qa_hash", required=True)
-@click.argument("microservices", nargs=-1, required=True)
-@click.pass_obj
-@catch_exceptions
-def cli_qa_update(obj, qa_hash, microservices):
-    from suite_py.commands.qa import QA
-
-    QA(
-        "update",
-        obj["project"],
-        obj["config"],
-        obj["tokens"],
-        {"hash": qa_hash, "services": microservices},
-    ).run()
-
-
-@qa.command("delete", help="Delete QA env")
-@click.argument("qa_hashes", nargs=-1, required=True)
-@click.option("--force", is_flag=True, default=False, help="Force QA deletion")
-@click.pass_obj
-@catch_exceptions
-def cli_qa_delete(obj, qa_hashes, force):
-    from suite_py.commands.qa import QA
-
-    QA(
-        "delete",
-        obj["project"],
-        obj["config"],
-        obj["tokens"],
-        {"hashes": qa_hashes, "force": force},
-    ).run()
-
-
-@qa.command("freeze", help="Freeze QA env")
-@click.argument("qa_hash", required=True)
-@click.pass_obj
-@catch_exceptions
-def cli_qa_freeze(obj, qa_hash):
-    from suite_py.commands.qa import QA
-
-    QA("freeze", obj["project"], obj["config"], obj["tokens"], {"hash": qa_hash}).run()
-
-
-@qa.command("unfreeze", help="Unfreeze QA env")
-@click.argument("qa_hash", required=True)
-@click.pass_obj
-@catch_exceptions
-def cli_qa_unfreeze(obj, qa_hash):
-    from suite_py.commands.qa import QA
-
-    QA(
-        "unfreeze", obj["project"], obj["config"], obj["tokens"], {"hash": qa_hash}
-    ).run()
-
-
-@qa.command("check", help="Check QA conf")
-@click.pass_obj
-@catch_exceptions
-def cli_qa_check(obj):
-    from suite_py.commands.qa import QA
-
-    QA("check", obj["project"], obj["config"], obj["tokens"]).run()
-
-
-@qa.command("describe", help="Describe QA environment")
-@click.argument("qa_hash", required=True)
-@click.option("--json", is_flag=True, default=False, help="Get response as JSON")
-@click.pass_obj
-@catch_exceptions
-def cli_qa_describe(obj, qa_hash, json):
-    from suite_py.commands.qa import QA
-
-    flags = {"hash": qa_hash, "json": json}
-    QA("describe", obj["project"], obj["config"], obj["tokens"], flags).run()
-
-
-@qa.command(
-    "toggle-maintenance",
-    help="Toggle maintenance mode (requires 'manage:maintenance' permission)",
-)
-@click.pass_obj
-@catch_exceptions
-def cli_qa_toggle_maintenance(obj):
-    from suite_py.commands.qa import QA
-
-    QA("toggle-maintenance", obj["project"], obj["config"], obj["tokens"]).run()
+    obj.call(Login).run()
 
 
 @main.group(
@@ -588,7 +454,9 @@ def secret():
 def cli_secret_create(obj, base_profile, secret_file):
     from suite_py.commands.secret import Secret
 
-    Secret(obj["project"], obj["config"], "create", base_profile, secret_file).run()
+    obj.call(
+        Secret, action="create", base_profile=base_profile, secret_file=secret_file
+    ).run()
 
 
 @secret.command("grant", help="Grant permissions to an existing secret")
@@ -599,7 +467,9 @@ def cli_secret_create(obj, base_profile, secret_file):
 def cli_secret_grant(obj, base_profile, secret_file):
     from suite_py.commands.secret import Secret
 
-    Secret(obj["project"], obj["config"], "grant", base_profile, secret_file).run()
+    obj.call(
+        Secret, action="grant", base_profile=base_profile, secret_file=secret_file
+    ).run()
 
 
 @main.command("batch-job", help="Run batch job on kube")
@@ -625,13 +495,11 @@ def cli_secret_grant(obj, base_profile, secret_file):
 def cli_run_batch_job(obj, environment, cpu_request, memory_request):
     from suite_py.commands.batch_job import BatchJob
 
-    BatchJob(
-        obj["project"],
-        obj["config"],
-        obj["tokens"],
-        environment,
-        cpu_request,
-        memory_request,
+    obj.call(
+        BatchJob,
+        environment=environment,
+        cpu_request=cpu_request,
+        memory_request=memory_request,
     ).run()
 
 
@@ -641,4 +509,4 @@ def cli_run_batch_job(obj, environment, cpu_request, memory_request):
 def cli_set_token(obj):
     from suite_py.commands.set_token import SetToken
 
-    SetToken(obj["tokens"]).run()
+    obj.call(SetToken).run()

{suite_py-1.41.3 → suite_py-1.41.4}/suite_py/commands/aggregator.py

@@ -7,15 +7,13 @@ from kubernetes import client, config
 from rich.console import Console
 from rich.table import Table
 
-from suite_py.lib import logger
-from suite_py.lib import metrics
+from suite_py.lib import logger, metrics
 from suite_py.lib.handler import prompt_utils
-from suite_py.lib.handler.captainhook_handler import CaptainHook
 
 
 class Aggregator:
-    def __init__(self, config, command):
-        self._captainhook = CaptainHook(config)
+    def __init__(self, captainhook, command):
+        self._captainhook = captainhook
         self._command = command
 
     @metrics.command("aggregator")

{suite_py-1.41.3 → suite_py-1.41.4}/suite_py/commands/ask_review.py

@@ -1,9 +1,7 @@
 # -*- coding: utf-8 -*-
 import sys
 
-from suite_py.lib import logger
-from suite_py.lib import metrics
-from suite_py.lib.handler.captainhook_handler import CaptainHook
+from suite_py.lib import logger, metrics
 from suite_py.lib.handler.frequent_reviewers_handler import FrequentReviewersHandler
 from suite_py.lib.handler.git_handler import GitHandler
 from suite_py.lib.handler.github_handler import GithubHandler
@@ -12,11 +10,11 @@ from suite_py.lib.handler.youtrack_handler import YoutrackHandler
 
 
 class AskReview:
-    def __init__(self, project, config, tokens):
+    def __init__(self, project, captainhook, config, tokens):
         self._project = project
         self._config = config
         self._youtrack = YoutrackHandler(config, tokens)
-        self._captainhook = CaptainHook(config)
+        self._captainhook = captainhook
         self._git = GitHandler(project, config)
         self._github = GithubHandler(tokens)
         self._frequent_reviewers = FrequentReviewersHandler(config)

{suite_py-1.41.3 → suite_py-1.41.4}/suite_py/commands/batch_job.py

@@ -5,8 +5,7 @@ import textwrap
 
 import semver
 
-from suite_py.lib import logger
-from suite_py.lib import metrics
+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

{suite_py-1.41.3 → suite_py-1.41.4}/suite_py/commands/bump.py

@@ -8,8 +8,7 @@ from typing import Any, Dict, List, Optional, cast
 
 from yaml import safe_load
 
-from suite_py.lib import logger
-from suite_py.lib import metrics
+from suite_py.lib import logger, metrics
 from suite_py.lib.handler.changelog_handler import ChangelogHandler
 from suite_py.lib.handler.git_handler import GitHandler
 from suite_py.lib.handler.github_handler import GithubHandler

{suite_py-1.41.3 → suite_py-1.41.4}/suite_py/commands/check.py

@@ -1,9 +1,7 @@
 # -*- coding: utf-8 -*-
 
-from suite_py.lib import logger
-from suite_py.lib import metrics
+from suite_py.lib import logger, metrics
 from suite_py.lib.handler import prompt_utils
-from suite_py.lib.handler.captainhook_handler import CaptainHook
 from suite_py.lib.handler.drone_handler import DroneHandler
 from suite_py.lib.handler.github_handler import GithubHandler
 from suite_py.lib.handler.youtrack_handler import YoutrackHandler
@@ -11,13 +9,13 @@ from suite_py.lib.symbol import CHECKMARK, CROSSMARK
 
 
 class Check:
-    def __init__(self, config, tokens):
+    def __init__(self, captainhook, 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(config)
+        self._captainhook = captainhook
         self._drone = DroneHandler(config, tokens)
 
         self._checks = [

suite_py-1.41.4/suite_py/commands/context.py

@@ -0,0 +1,26 @@
+import dataclasses
+from inspect import signature
+
+from suite_py.lib.config import Config
+from suite_py.lib.handler.captainhook_handler import CaptainHook
+from suite_py.lib.handler.okta_handler import Okta
+from suite_py.lib.tokens import Tokens
+
+
+@dataclasses.dataclass
+class Context:
+    project: str
+    config: Config
+    captainhook: CaptainHook
+    tokens: Tokens
+    okta: Okta
+
+    # Call the function to_call with kwargs, injecting fields from self as default arguments
+    def call(self, to_call, **kwargs):
+        provided = dataclasses.asdict(self)
+        needed = signature(to_call).parameters.keys()
+        provided = {k: provided[k] for k in needed if k in provided}
+
+        kwargs = provided | kwargs
+
+        return to_call(**kwargs)

{suite_py-1.41.3 → suite_py-1.41.4}/suite_py/commands/create_branch.py

@@ -4,8 +4,7 @@ import sys
 
 import requests
 
-from suite_py.lib import logger
-from suite_py.lib import metrics
+from suite_py.lib import logger, metrics
 from suite_py.lib.handler import prompt_utils
 from suite_py.lib.handler.git_handler import GitHandler, is_branch_name_valid
 from suite_py.lib.handler.youtrack_handler import YoutrackHandler

{suite_py-1.41.3 → suite_py-1.41.4}/suite_py/commands/deploy.py

@@ -4,11 +4,9 @@ import sys
 
 from suite_py.commands import common
 from suite_py.commands.release import _parse_available_countries
-from suite_py.lib import logger
-from suite_py.lib import metrics
+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.captainhook_handler import CaptainHook
 from suite_py.lib.handler.changelog_handler import ChangelogHandler
 from suite_py.lib.handler.drone_handler import DroneHandler
 from suite_py.lib.handler.git_handler import GitHandler
@@ -19,11 +17,11 @@ from suite_py.lib.handler.youtrack_handler import YoutrackHandler
 
 class Deploy:
     # pylint: disable=too-many-instance-attributes
-    def __init__(self, project, config, tokens):
+    def __init__(self, project, captainhook, config, tokens):
         self._project = project
         self._config = config
         self._youtrack = YoutrackHandler(config, tokens)
-        self._captainhook = CaptainHook(config)
+        self._captainhook = captainhook
         self._changelog_handler = ChangelogHandler()
         self._github = GithubHandler(tokens)
         self._repo = self._github.get_repo(project)