From cba62894edb473ba480ebc0d94a1a378136529c9 Mon Sep 17 00:00:00 2001 From: Leo Kim Date: Wed, 16 Feb 2022 19:47:41 -0500 Subject: [PATCH] custom query support --- duneanalytics/duneanalytics.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/duneanalytics/duneanalytics.py b/duneanalytics/duneanalytics.py index cc3a7cf0..6b1dc060 100644 --- a/duneanalytics/duneanalytics.py +++ b/duneanalytics/duneanalytics.py @@ -84,14 +84,21 @@ def fetch_auth_token(self): else: print(response.text) - def query_result_id(self, query_id): + def query_result_id(self, query_id, parameters=[]): """ Fetch the query result id for a query :param query_id: provide the query_id + :param parameters: (optional) list of parameter objects to customize the query + ex. [{"type": "data_type", "key": "key_name", "value": "value"}] :return: """ - query_data = {"operationName": "GetResult", "variables": {"query_id": query_id}, + if parameters: + query_variables = {"query_id": query_id, "parameters": parameters} + else: + query_variables = {"query_id": query_id} + + query_data = {"operationName": "GetResult", "variables": query_variables, "query": "query GetResult($query_id: Int!, $parameters: [Parameter!]) " "{\n get_result(query_id: $query_id, parameters: $parameters) " "{\n job_id\n result_id\n __typename\n }\n}\n"