first commit

jellyfin_accounts/jf_api.py

@@ -0,0 +1,82 @@
+#!/usr/bin/env python3
+import requests
+
+
+class Error(Exception):
+    pass
+
+class Jellyfin:
+    class UserExistsError(Error):
+        pass
+    class UserNotFoundError(Error):
+        pass
+    class AuthenticationError(Error):
+        pass
+    class AuthenticationRequiredError(Error):
+        pass
+    def __init__(self, server, client, version, device, deviceId):
+        self.server = server
+        self.client = client
+        self.version = version
+        self.device = device
+        self.deviceId = deviceId
+        self.useragent = f"{self.client}/{self.version}"
+        self.auth = "MediaBrowser "
+        self.auth += f"Client={self.client}, "
+        self.auth += f"Device={self.device}, "
+        self.auth += f"DeviceId={self.deviceId}, "
+        self.auth += f"Version={self.version}"
+        self.header = {
+           "Accept": "application/json",
+           "Content-type": "application/x-www-form-urlencoded; charset=UTF-8",
+           "X-Application": f"{self.client}/{self.version}",
+           "Accept-Charset": "UTF-8,*",
+           "Accept-encoding": "gzip",
+           "User-Agent": self.useragent,
+           "X-Emby-Authorization": self.auth
+        }
+    def getUsers(self, username="all"):
+        response = requests.get(self.server+"/emby/Users/Public").json()
+        if username == "all":
+            return response
+        else:
+            match = False
+            for user in response:
+                if user['Name'] == username:
+                    match = True
+                    return user
+            if not match:
+                raise self.UserNotFoundError
+    def authenticate(self, username, password):
+        self.username = username
+        self.password = password
+        response = requests.post(self.server+"/emby/Users/AuthenticateByName",
+                                 headers=self.header,
+                                 params={'Username': self.username,
+                                         'Pw': self.password})
+        if response.status_code == 200:
+            json = response.json()
+            self.userId = json['User']['Id']
+            self.accessToken = json['AccessToken']
+            self.auth += f", Token={self.accessToken}"
+            self.header['X-Emby-Authorization'] = self.auth
+            return True
+        else:
+            raise self.AuthenticationError
+    def setPolicy(self, userId, policy):
+        return requests.post(self.server+"/Users/"+userId+"/Policy",
+                             headers=self.header,
+                             params=policy)
+    def newUser(self, username, password):
+        for user in self.getUsers():
+            if user['Name'] == username:
+                raise self.UserExistsError
+        response = requests.post(self.server+"/emby/Users/New",
+                                 headers=self.header,
+                                 params={'Name': username,
+                                         'Password': password})
+        if response.status_code == 401:
+            raise self.AuthenticationRequiredError
+        return response
+
+# template user's policies should be copied to each new account.