Adicionando misc com notebooks (#12)

misc/colab-scripts/client_graphql.py

@@ -0,0 +1,70 @@
+from gql import Client, gql
+from gql.transport.requests import RequestsHTTPTransport
+from google.colab import userdata
+from typing import Dict
+from utils import extract_id_from_response
+
+
+class MakeClient:
+    def __init__(self):
+        self.graphql_url = "https://backend.basedosdados.org/graphql"
+        self.query = self.make_client()
+
+    def make_client(self, headers: Dict[str, str] = None) -> Client:
+        transport = RequestsHTTPTransport(
+            url=self.graphql_url, headers=headers, use_json=True
+        )
+
+        return Client(transport=transport, fetch_schema_from_transport=False)
+
+    def mutation(self) -> Client:
+        query = """
+      mutation ($email: String!, $password: String!) {
+          tokenAuth(email: $email, password: $password) {
+              token
+            }
+        }
+      """
+
+        variables = {
+            "email": userdata.get("email"),
+            "password": userdata.get("senha"),
+        }
+
+        response = self.query.execute(gql(query), variable_values=variables)
+
+        token = response["tokenAuth"]["token"]
+
+        header_for_mutation_query = {"Authorization": f"Bearer {token}"}
+
+        return self.make_client(header_for_mutation_query)
+
+    def query_mutation(
+        self, mutation_class: str, input_values: dict, only_id: bool = False
+    ) -> str | dict:
+        client = self.mutation()
+
+        query = f"""
+                mutation($input:CreateUpdate{mutation_class}Input!){{
+                    CreateUpdate{mutation_class}(input: $input){{
+                    errors {{
+                        field,
+                        messages
+                    }},
+                    {mutation_class.lower()} {{
+                        id
+                    }}
+                }}
+                }}
+              """
+
+        variables = {"input": input_values}
+
+        mutation_response = client.execute(
+            gql(query), variable_values=variables
+        )
+
+        if only_id:
+            return extract_id_from_response(mutation_response, mutation_class)
+
+        return mutation_response

misc/colab-scripts/utils.py

@@ -0,0 +1,10 @@
+def clear_id(string: str, place: str) -> str:
+    return string.replace(place, "")
+
+
+def extract_id_from_response(
+    mutation_response: dict, mutation_class: str
+) -> str:
+    key_mutation = f"CreateUpdate{mutation_class}"
+    id_response = mutation_response[key_mutation][mutation_class.lower()]["id"]
+    return clear_id(id_response, f"{mutation_class}Node:")