added get categories

HiWay-Media · Oct 18, 2023 · a24f8d2 · a24f8d2
1 parent 33f8199
commit a24f8d2
Showing 1 changed file with 24 additions and 7 deletions.
diff --git a/compress/compress.py b/compress/compress.py
@@ -53,13 +53,30 @@ def __init__(self, api_key :str , customer_name : str):
     # */
     def get_s3_space(self):
         return self.req.post(TNGRM_BASE_URL + S3_SPACE, 
-          headers={
-            "Content-Type": "application/json",
-          },
-          body = json.dumps({
-            "api_key": self.api_key,
-            "client_id": self.client_id,
-          }),
+            headers={
+                "Content-Type": "application/json",
+            },
+            body = json.dumps({
+                "api_key": self.api_key,
+                "client_id": self.client_id,
+            }),
+        )
+    #
+    # /**
+    # *
+    # * @param {string} apikey
+    # * @param {string} customer
+    # * @returns list of categories of the customer
+    # */
+    def get_categories(self):
+        return self.req.post(TNGRM_BASE_URL + GET_CATEGORIES, 
+            headers={
+                "Content-Type": "application/json",
+            },
+            body = json.dumps(({
+                "api_key": self.api_key,
+                "client_id": self.client_id,
+            })
         )
     #
 #