leadguru-jobs 0.691.0__tar.gz → 0.693.0__tar.gz

{leadguru_jobs-0.691.0 → leadguru_jobs-0.693.0}/PKG-INFO

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

{leadguru_jobs-0.691.0 → leadguru_jobs-0.693.0}/leadguru_jobs.egg-info/PKG-INFO

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

{leadguru_jobs-0.691.0 → leadguru_jobs-0.693.0}/leadguru_jobs.egg-info/SOURCES.txt

@@ -24,6 +24,7 @@ lgt_jobs/assets/images/mail.png
 lgt_jobs/jobs/__init__.py
 lgt_jobs/jobs/analytics.py
 lgt_jobs/jobs/bot_stats_update.py
+lgt_jobs/jobs/bots_killer.py
 lgt_jobs/jobs/chat_history.py
 lgt_jobs/jobs/connect_sources.py
 lgt_jobs/jobs/inbox_leads.py
@@ -114,5 +115,6 @@ lgt_jobs/lgt_data/repositories/__init__.py
 lgt_jobs/lgt_data/repositories/post/__init__.py
 lgt_jobs/lgt_data/repositories/post/posts.py
 lgt_jobs/services/__init__.py
+lgt_jobs/services/k8_manager.py
 lgt_jobs/services/web_client.py
 lgt_jobs/templates/new_message.html

{leadguru_jobs-0.691.0 → leadguru_jobs-0.693.0}/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
 ]

leadguru_jobs-0.693.0/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}')

leadguru_jobs-0.693.0/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))