leadguru-jobs 0.692.0__py3-none-any.whl → 0.693.0__py3-none-any.whl

{leadguru_jobs-0.692.0.dist-info → leadguru_jobs-0.693.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leadguru_jobs
-Version: 0.692.0
+Version: 0.693.0
 Summary: LGT jobs builds
 Author-email: developer@leadguru.co
 Classifier: Development Status :: 5 - Production/Stable

{leadguru_jobs-0.692.0.dist-info → leadguru_jobs-0.693.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-lgt_jobs/__init__.py,sha256=VepHCuCW4uUqx6Nj5gKdZeJ9IYDSyhlD70FvpnKNmPk,2316
+lgt_jobs/__init__.py,sha256=fQ97ZFeVtbOgODPdB26x7ku45AQPQ8ixeziiCAjNYZM,2451
 lgt_jobs/basejobs.py,sha256=LPEclvY6fIG17YMGy9pRK0Q58n62gkI5W1BTK6R2p-c,1234
 lgt_jobs/env.py,sha256=cRO03GGvstUjBsv3uYO-iakrOvAk_ZWUP_fnmf21iZQ,789
 lgt_jobs/main.py,sha256=cK_nkBtJHnUNDbba1WZotqPtI_6OWxiYQkAgco9OAmE,1539
@@ -14,6 +14,7 @@ lgt_jobs/assets/images/mail.png,sha256=eORzQcAMkFr7DjgtABVhJ_zFuXgO7OXv78lLF4b39
 lgt_jobs/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/jobs/analytics.py,sha256=IIsWt4A1qUw3w-S-8W16uKY1FRVWfXXA41_mu4uCNiM,979
 lgt_jobs/jobs/bot_stats_update.py,sha256=3gq1Ilj_RHddYeSny_238ITs7YIfCf7d4BmfYU8PKlo,8177
+lgt_jobs/jobs/bots_killer.py,sha256=s2f7kal4vAMJd9YaMQ1X3FkCrLWCKNb-bWeI9RZ1Oco,506
 lgt_jobs/jobs/chat_history.py,sha256=rj337uHT2XmyDu3oMPbnCBxSsRn4roxDNS0hsbvoI7I,5040
 lgt_jobs/jobs/connect_sources.py,sha256=vOKu2Q9cmQsj4VMVzOoEJkD1wzE9XurbIChx35rES8I,4957
 lgt_jobs/jobs/inbox_leads.py,sha256=6UhPp9AfSAGb3elFDT6jm7j2HZ9jPOPngsUj-hUKAYw,4867
@@ -104,9 +105,10 @@ lgt_jobs/lgt_data/repositories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 lgt_jobs/lgt_data/repositories/post/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/lgt_data/repositories/post/posts.py,sha256=ZA19L-CYdxLhlapyMQtSJzP3QMUlHq9UEgW3ab86aDs,1024
 lgt_jobs/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lgt_jobs/services/k8_manager.py,sha256=bXpka9psIQ5UJ6QG_e4xolmE_3gtmNrzzZeL03bJ62I,995
 lgt_jobs/services/web_client.py,sha256=oMyWJxwGeIe3f40fPT7xcisjDg3BhA3Ipf8dr1jVT-Y,1549
 lgt_jobs/templates/new_message.html,sha256=dZl8UmdAOOMq4nidvAgMFroSrTV7Pw0RWt2yLp_2idg,6989
-leadguru_jobs-0.692.0.dist-info/METADATA,sha256=MryU6Xz8TH9h3srWm6vchXspthp8KK1nQX51K7CKTvU,1319
-leadguru_jobs-0.692.0.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-leadguru_jobs-0.692.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
-leadguru_jobs-0.692.0.dist-info/RECORD,,
+leadguru_jobs-0.693.0.dist-info/METADATA,sha256=LD6pdCdAEo2oTXSGd3WZk0jH3dgux1qvvXp3ByNuoXU,1319
+leadguru_jobs-0.693.0.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+leadguru_jobs-0.693.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
+leadguru_jobs-0.693.0.dist-info/RECORD,,

{leadguru_jobs-0.692.0.dist-info → leadguru_jobs-0.693.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.2.0)
+Generator: setuptools (75.5.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

lgt_jobs/__init__.py

@@ -15,6 +15,7 @@ from .runner import (BackgroundJobRunner)
 from .simple_job import (SimpleTestJob, SimpleTestJobData)
 from .jobs.load_slack_people import (LoadSlackPeopleJob, LoadSlackPeopleJobData)
 from .jobs.load_slack_users import LoadSlackUsersJob, LoadSlackUsersJobData
+from .jobs.bots_killer import BotsKillerJob, BotsKillerData
 
 jobs_map = {
     "SimpleTestJob": SimpleTestJob,
@@ -27,6 +28,7 @@ jobs_map = {
     "SendMassMessageSlackChannelJob": SendMassMessageSlackChannelJob,
     "ConnectSourceJob": ConnectSourceJob,
     "LoadSlackPeopleJob": LoadSlackPeopleJob,
+    "BotsKillerJob": BotsKillerJob,
     "LoadSlackUsersJob": LoadSlackUsersJob
 }
 __all__ = [
@@ -43,6 +45,7 @@ __all__ = [
     ConnectSourceJob,
     LoadSlackPeopleJob,
     LoadSlackUsersJob,
+    BotsKillerJob,
 
     # module classes
     BackgroundJobRunner,
@@ -60,6 +63,7 @@ __all__ = [
     ConnectSourceJobData,
     LoadSlackPeopleJobData,
     LoadSlackUsersJobData,
+    BotsKillerData,
     # mapping
     jobs_map
 ]

lgt_jobs/jobs/bots_killer.py

@@ -0,0 +1,21 @@
+from abc import ABC
+from pydantic import BaseModel
+from lgt_jobs.basejobs import BaseBackgroundJob, BaseBackgroundJobData
+from lgt_jobs.services.k8_manager import K8Manager
+
+"""
+Kill bots
+"""
+
+
+class BotsKillerData(BaseBackgroundJobData, BaseModel):
+    environment: str
+
+
+class BotsKillerJob(BaseBackgroundJob, ABC):
+    @property
+    def job_data_type(self) -> type:
+        return BotsKillerData
+
+    def exec(self, data: BotsKillerData):
+        K8Manager().delete_bots(f'leadguru-{data.environment}')

lgt_jobs/services/k8_manager.py

@@ -0,0 +1,28 @@
+import json
+import logging as log
+import os
+
+from kubernetes import config
+from kubernetes.client.api.core_v1_api import CoreV1Api
+from kubernetes.client.rest import ApiException
+
+
+class K8Manager:
+
+    def __init__(self):
+        with open(os.environ.get('K8_CONFIG'), 'r') as k8_config:
+            config.load_kube_config_from_dict(json.loads(k8_config.read()))
+
+    @staticmethod
+    def delete_bots(namespace: str):
+        api_instance = CoreV1Api()
+        pod_names_to_delete = ['lgt-slack-bots-v2', 'lgt-discord-bots', 'lgt-bg-worker']
+        pods = api_instance.list_namespaced_pod(namespace)
+        pods_to_delete = [pod.metadata.name for pod in pods.items for name in pod_names_to_delete
+                          if name in pod.metadata.name]
+        for pod in pods_to_delete:
+            try:
+                api_instance.delete_namespaced_pod(pod, namespace)
+                log.info(f'Killed pod: {pod}')
+            except ApiException as e:
+                log.error(str(e))