Update to use secrets manager

jdgsmallwood · Jul 11, 2024 · 3061a04 · 3061a04
1 parent 657f76a
commit 3061a04
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/lambda_function.py b/src/lambda_function.py
@@ -1,3 +1,4 @@
+import json
 import os
 import traceback
 import xml.etree.ElementTree as ET
@@ -75,7 +76,11 @@ def get_previous_uv(uv: float) -> float:
 
 
 def send_message_to_telegram(message: str) -> None:
-    BOT_TOKEN = os.getenv("BOT_TOKEN")
+    client = boto3.client(
+        service_name="secretsmanager",
+    )
+    secret_value = client.get_secret_value(SecretId="telegram/uv")
+    BOT_TOKEN = json.loads(secret_value["SecretString"])["UV_TELEGRAM_TOKEN"]
     CHAT_ID = os.getenv("CHAT_ID")
     bot = telebot.TeleBot(BOT_TOKEN)
     bot.send_message(CHAT_ID, message)