initial commit

mattermost.py

@@ -0,0 +1,43 @@
+from datetime import datetime, timedelta, timezone
+from dotenv import load_dotenv
+import os
+import requests
+
+load_dotenv()
+
+
+class Mattermost:
+    def __init__(self, user_id="me"):
+        host = os.getenv("MATTERMOST_SERVER_URL", "http://localhost")
+        access_token = os.getenv("MATTERMOST_ACCESS_TOKEN", "")
+        self.user_id = user_id
+        self.url = f"{host}/api/v4/users/{self.user_id}/status/custom"
+        self.headers = {
+            "Authorization": f"Bearer {access_token}",
+            "Content-Type": "application/json",
+        }
+
+    def set_status(self, status, emoji, expires_at=None):
+        data = {
+            "emoji": emoji,
+            "text": status,
+            "expires_at": expires_at,
+        }
+        response = requests.put(self.url, headers=self.headers, json=data)
+        if response.status_code != 200:
+            print(f"Failed to set Mattermost status: {response.content}")
+            raise Exception(f"Failed to set Mattermost status: {response.content}")
+
+    def set_now_playing(self, track, artist, duration):
+        expires_at = (datetime.now(timezone.utc) + timedelta(seconds=duration)).astimezone()
+        status = f"{track} - {artist}"
+        print(f"Setting Mattermost status to {status} until {expires_at}")
+        self.set_status(status, "headphones", expires_at.isoformat())
+
+
+    def get_status(self):
+        response = requests.get(self.url, headers=self.headers)
+        if response.status_code != 200:
+            print(f"Failed to get Mattermost status: {response.content}")
+            raise Exception(f"Failed to get Mattermost status: {response.content}")
+        return response.json()