StashDbClient class

pandas/libraries/StashDbClient.py

@@ -0,0 +1,74 @@
+import requests
+import stashapi.log as logger
+
+
+class StashDbClient(object):
+    def __init__(self, endpoint, api_key):
+        self.endpoint = endpoint
+        self.api_key = api_key
+
+    def query_scenes(self, scene_id):
+        query = """
+            query FindScene($scene_id: ID!) {
+                findScene(
+                    id: $scene_id
+                ) {
+                    id
+                    title
+                    details
+                    release_date
+                    urls {
+                        url
+                        site {
+                            name
+                            url
+                        }
+                    }
+                    studio {
+                        id
+                        name
+                        parent {
+                            id
+                            name
+                        }
+                    }
+                    images {
+                        id
+                        url
+                    }
+                    performers {
+                        performer {
+                            id
+                            name
+                        }
+                    }
+                    duration
+                    code
+                    tags {
+                        id
+                        name
+                    }
+                }
+            }
+        """
+        result = self._gql_query(
+            query, {"scene_id": scene_id}
+        )
+        return result
+
+    def _gql_query(self, query, variables=None):
+        headers = {"Content-Type": "application/json"}
+        if self.api_key:
+            headers["Apikey"] = self.api_key
+        response = requests.post(
+            self.endpoint,
+            json={"query": query, "variables": variables},
+            headers=headers,
+        )
+        if response.status_code == 200:
+            return response.json()
+        else:
+            logger.error(
+                f"Query failed with status code {response.status_code}: {response.text}"
+            )
+            return None