code refactoring

This commit is contained in:
sHa
2024-04-08 09:06:26 +03:00
parent 13475a7db4
commit f2a4978cf2
13 changed files with 269 additions and 183 deletions

19
connectors/mattermost.py Normal file
View File

@@ -0,0 +1,19 @@
import requests
class MattermostConnector:
def __init__(self, connection_config) -> None:
self.url = connection_config.get("url", "https://mattermost.com")
self.token = connection_config.get("token", "")
self.connect(connection_config.get("user_id", "me"))
def connect(self, user_id):
self.headers = {
"Authorization": f"Bearer {self.token}",
"Content-Type": "application/json",
}
self.url = f"{self.url}/api/v4/users/{user_id}/status/custom"
def send(self, data):
response = requests.put(self.url, headers=self.headers, json=data)
if response.status_code != 200:
raise Exception(f"Failed to set Mattermost status: {response.content!r}")