Add APM #184

Merged
luketainton merged 5 commits from feature/apm into next 2024-04-21 17:17:03 +02:00
8 changed files with 83 additions and 23 deletions
Showing only changes of commit ca7c9a5226 - Show all commits

View File

@ -1,3 +1,6 @@
APP_LIFECYCLE="dev"
SENTRY_ENABLED="False"
SENTRY_DSN=""
ADMIN_EMAIL="" ADMIN_EMAIL=""
ADMIN_FIRST_NAME="" ADMIN_FIRST_NAME=""
BOT_NAME="" BOT_NAME=""

View File

@ -1,6 +1,7 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import logging import logging
import sentry_sdk
from webex_bot.models.command import Command from webex_bot.models.command import Command
from webex_bot.models.response import Response, response_from_adaptive_card from webex_bot.models.response import Response, response_from_adaptive_card
@ -113,6 +114,7 @@ class SubmitTaskCommand(Command):
Submit(title="Cancel", data={"command_keyword": "exit"}), Submit(title="Cancel", data={"command_keyword": "exit"}),
], ],
) )
with sentry_sdk.start_transaction(name="submit_task_command"):
return response_from_adaptive_card(card) return response_from_adaptive_card(card)
@ -147,6 +149,7 @@ class SubmitTaskCallback(Command):
) )
def execute(self, message, attachment_actions, activity) -> str: def execute(self, message, attachment_actions, activity) -> str:
with sentry_sdk.start_transaction(name="submit_task_callback"):
return self.msg return self.msg
@ -158,11 +161,13 @@ class MyTasksCallback(Command):
self.msg: str = "" self.msg: str = ""
def pre_execute(self, message, attachment_actions, activity) -> str: def pre_execute(self, message, attachment_actions, activity) -> str:
with sentry_sdk.start_transaction(name="my_tasks_preexec"):
return "Getting your tasks..." return "Getting your tasks..."
def execute(self, message, attachment_actions, activity) -> str | None: def execute(self, message, attachment_actions, activity) -> str | None:
sender: str = attachment_actions.inputs.get("sender") sender: str = attachment_actions.inputs.get("sender")
result: bool = get_tasks(requestor=sender) result: bool = get_tasks(requestor=sender)
with sentry_sdk.start_transaction(name="my_tasks_exec"):
if not result: if not result:
return "Failed to get tasks. Please try again." return "Failed to get tasks. Please try again."
return return

View File

@ -1,5 +1,8 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import sentry_sdk
from sentry_sdk.integrations.stdlib import StdlibIntegration
from webex_bot.webex_bot import WebexBot from webex_bot.webex_bot import WebexBot
from app.commands.exit import ExitCommand from app.commands.exit import ExitCommand
@ -7,6 +10,16 @@ from app.commands.submit_task import SubmitTaskCommand
from app.utils.config import config from app.utils.config import config
if config.sentry_enabled:
apm = sentry_sdk.init(
dsn=config.sentry_dsn,
enable_tracing=True,
environment=config.environment,
integrations=[StdlibIntegration()],
spotlight=True
)
def create_bot() -> WebexBot: def create_bot() -> WebexBot:
# Create a Bot Object # Create a Bot Object
webex_bot: WebexBot = WebexBot( webex_bot: WebexBot = WebexBot(

16
app/utils/apm.py Normal file
View File

@ -0,0 +1,16 @@
import sentry_sdk
from sentry_sdk.integrations.stdlib import StdlibIntegration
from app.utils.config import config
if config.sentry_enabled:
apm = sentry_sdk.init(
dsn=config.sentry_dsn,
enable_tracing=True,
environment=config.environment,
integrations=[StdlibIntegration()],
spotlight=True
)
else:
apm = None

View File

@ -5,11 +5,28 @@ import os
class Config: class Config:
def __init__(self) -> None: def __init__(self) -> None:
self.__environment: str = os.environ.get("APP_LIFECYCLE", "DEV").upper()
self.__bot_name: str = os.environ["BOT_NAME"] self.__bot_name: str = os.environ["BOT_NAME"]
self.__webex_token: str = os.environ["WEBEX_API_KEY"] self.__webex_token: str = os.environ["WEBEX_API_KEY"]
self.__admin_first_name: str = os.environ["ADMIN_FIRST_NAME"] self.__admin_first_name: str = os.environ["ADMIN_FIRST_NAME"]
self.__admin_emails: list = os.environ["ADMIN_EMAIL"].split(",") self.__admin_emails: list = os.environ["ADMIN_EMAIL"].split(",")
self.__n8n_webhook_url: str = os.environ["N8N_WEBHOOK_URL"] self.__n8n_webhook_url: str = os.environ["N8N_WEBHOOK_URL"]
self.__sentry_dsn: str = os.environ.get("SENTRY_DSN", "")
self.__sentry_enabled: bool = True if (os.environ.get("SENTRY_ENABLED").upper() == "TRUE" and self.__sentry_dsn != "") else False
@property
def environment(self) -> str:
return self.__environment
@property
def sentry_enabled(self) -> bool:
return self.__sentry_enabled
@property
def sentry_dsn(self) -> str:
if not self.__sentry_enabled:
return ""
return self.__sentry_dsn
@property @property
def bot_name(self) -> str: def bot_name(self) -> str:

View File

@ -1,4 +1,5 @@
import requests import requests
import sentry_sdk
from app.utils.config import config from app.utils.config import config
@ -16,16 +17,19 @@ def __n8n_post(data: dict) -> bool:
def submit_task(summary, description, completion_date, requestor) -> bool: def submit_task(summary, description, completion_date, requestor) -> bool:
with sentry_sdk.start_transaction(name="submit_task"):
data: dict = { data: dict = {
"requestor": requestor, "requestor": requestor,
"title": summary, "title": summary,
"description": description, "description": description,
"completiondate": completion_date, "completiondate": completion_date,
} }
return __n8n_post(data=data) _data = __n8n_post(data=data)
return _data
def get_tasks(requestor) -> bool: def get_tasks(requestor) -> bool:
with sentry_sdk.start_transaction(name="get_tasks"):
headers: dict = {"Content-Type": "application/json"} headers: dict = {"Content-Type": "application/json"}
resp: requests.Response = requests.get( resp: requests.Response = requests.get(
url=config.n8n_webhook_url, url=config.n8n_webhook_url,
@ -34,4 +38,5 @@ def get_tasks(requestor) -> bool:
verify=False, verify=False,
params={"requestor": requestor}, params={"requestor": requestor},
) )
return bool(resp.status_code == 200) _data = bool(resp.status_code == 200)
return _data

View File

@ -33,6 +33,7 @@ python-dotenv==1.0.1
PyYAML==6.0.1 PyYAML==6.0.1
requests==2.31.0 requests==2.31.0
requests-toolbelt==1.0.0 requests-toolbelt==1.0.0
sentry-sdk==1.45.0
six==1.16.0 six==1.16.0
toml==0.10.2 toml==0.10.2
tomli==2.0.1 tomli==2.0.1

View File

@ -1,3 +1,3 @@
export $(cat .env | xargs) export $(cat .env | xargs)
python -B -m app.main python -B -m app.main
unset BOT_NAME WEBEX_API_KEY ADMIN_FIRST_NAME ADMIN_EMAIL N8N_WEBHOOK_URL unset APP_LIFECYCLE SENTRY_ENABLED SENTRY_DSN BOT_NAME WEBEX_API_KEY ADMIN_FIRST_NAME ADMIN_EMAIL N8N_WEBHOOK_URL