From 35da55ab4591e49bf84bee6850ed6fe4a81c044c Mon Sep 17 00:00:00 2001 From: iLLiCiTiT Date: Mon, 11 Oct 2021 12:07:46 +0200 Subject: [PATCH] added send/get job methods in module --- .../hosts_job_server/module.py | 20 ++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) diff --git a/openpype/modules/default_modules/hosts_job_server/module.py b/openpype/modules/default_modules/hosts_job_server/module.py index 7f2a17c5c7..af62f5d9a7 100644 --- a/openpype/modules/default_modules/hosts_job_server/module.py +++ b/openpype/modules/default_modules/hosts_job_server/module.py @@ -5,9 +5,27 @@ class HostsJobServer(OpenPypeModule): name = "hosts_job_server" def initialize(self, modules_settings): - self._server_url = modules_settings.get("server_url") + server_url = modules_settings.get("server_url") + while server_url.endswith("/"): + server_url = server_url[:-1] + self._server_url = server_url self.enabled = True + def send_job(self, host_name, job_data): + import requests + + job_data = job_data or {} + job_data["host_name"] = host_name + api_path = "{}/api/jobs".format(self._server_url) + job_id = requests.post(api_path, data=job_data) + return job_id + + def get_job_status(self, job_id): + import requests + + api_path = "{}/api/jobs/{}".format(self._server_url, job_id) + return requests.get(api_path) + def start_server(self, port=None, host=None): from .job_server import main