adql query parser function

cdci_data_analysis/analysis/ivoa_helper.py

@@ -0,0 +1,84 @@
+import antlr4
+from queryparser.adql import ADQLQueryTranslator
+from queryparser.mysql import MySQLQueryProcessor
+from queryparser.exceptions import QuerySyntaxError
+
+from mysql.connector import connect, Error
+
+from ..flask_app.sentry import sentry
+from ..app_logging import app_logging
+
+logger = app_logging.getLogger('ivoa_helper')
+
+
+def parse_adql_query(query):
+    try:
+        # queryparser
+        adt = ADQLQueryTranslator(query)
+        qp = MySQLQueryProcessor()
+        qp.set_query(adt.to_mysql())
+        qp.process_query()
+
+        output_obj = dict(
+            columns=qp.display_columns,
+            tables=qp.tables,
+            rest=qp,
+            mysql_query=qp.query
+        )
+
+    except QuerySyntaxError as qe:
+        logger.error(f'Error parsing ADQL query: {qe}')
+        output_obj = dict(
+            tables=None,
+            columns=None,
+            rest=None,
+            mysql_query=None
+        )
+    return output_obj
+
+
+def run_ivoa_query(query, **kwargs):
+    parsed_query_obj = parse_adql_query(query)
+
+    # TODO use a specific dedicated table and schema to refer to the product_gallery DB ?
+    # tables = parsed_query_obj.get('tables', [])
+    # if len(tables) == 1 and tables[0] == 'product_gallery':
+    logger.info('Performing query on the product_gallery')
+    vo_mysql_pg_host = kwargs.get('vo_mysql_pg_host', None)
+    vo_mysql_pg_user = kwargs.get('vo_mysql_pg_user', None)
+    vo_mysql_pg_password = kwargs.get('vo_mysql_pg_password', None)
+    vo_mysql_pg_db = kwargs.get('vo_mysql_pg_db', None)
+    result_list = run_ivoa_query_from_product_gallery(parsed_query_obj,
+                                                      vo_mysql_pg_host=vo_mysql_pg_host,
+                                                      vo_mysql_pg_user=vo_mysql_pg_user,
+                                                      vo_mysql_pg_password=vo_mysql_pg_password,
+                                                      vo_mysql_pg_db=vo_mysql_pg_db)
+    return result_list
+
+
+def run_ivoa_query_from_product_gallery(parsed_query_obj,
+                                        vo_mysql_pg_host,
+                                        vo_mysql_pg_user,
+                                        vo_mysql_pg_password,
+                                        vo_mysql_pg_db
+                                        ):
+    result_list = []
+
+    try:
+        with connect(
+                host=vo_mysql_pg_host,
+                user=vo_mysql_pg_user,
+                password=vo_mysql_pg_password,
+                database=vo_mysql_pg_db
+        ) as connection:
+            create_db_query = parsed_query_obj.get('mysql_query')
+            with connection.cursor() as cursor:
+                cursor.execute(create_db_query)
+                for db in cursor:
+                    logger.info(db)
+
+    except Error as e:
+        sentry.capture_message(f"Error when connecting to MySQL or performing the query: {str(e)}")
+        logger.error(f"Error when connecting to MySQL or performing the query: {str(e)}")
+
+    return result_list

cdci_data_analysis/config_dir/conf_env.yml.example

@@ -122,4 +122,10 @@ dispatcher:
         # url for the conversion of a given time, in UTC format, to the correspondent REVNUM
         converttime_revnum_service_url: COVERTTIME_REVNUM_SERVICE_URL
 
-
+    # virtual observatory related configurations (eg mysql credentials)
+    vo_options:
+        # mysql credentials
+        vo_mysql_pg_host: MYSQL_PG_HOST
+        vo_mysql_pg_user: MYSQL_PG_USER
+        vo_mysql_pg_password: MYSQL_PG_PASSWORD
+        vo_mysql_pg_db: MYSQL_PG_DB

cdci_data_analysis/configurer.py

@@ -266,6 +266,10 @@ def __init__(self, cfg_dict, origin=None):
                                      disp_dict.get('renku_options', {}).get('renku_gitlab_repository_url', None),
                                      disp_dict.get('renku_options', {}).get('renku_base_project_url', None),
                                      disp_dict.get('renku_options', {}).get('ssh_key_path', None),
+                                     disp_dict.get('vo_options', {}).get('vo_mysql_pg_host', None),
+                                     disp_dict.get('vo_options', {}).get('vo_mysql_pg_user', None),
+                                     disp_dict.get('vo_options', {}).get('vo_mysql_pg_password', None),
+                                     disp_dict.get('vo_options', {}).get('vo_mysql_pg_db', None),
                                      )
 
         # not used?
@@ -344,6 +348,10 @@ def set_conf_dispatcher(self,
                             renku_gitlab_repository_url,
                             renku_base_project_url,
                             renku_gitlab_ssh_key_path,
+                            vo_mysql_pg_host,
+                            vo_mysql_pg_user,
+                            vo_mysql_pg_password,
+                            vo_mysql_pg_db
                             ):
         # Generic to dispatcher
         #print(dispatcher_url, dispatcher_port)
@@ -395,6 +403,10 @@ def set_conf_dispatcher(self,
         self.renku_gitlab_repository_url = renku_gitlab_repository_url
         self.renku_gitlab_ssh_key_path = renku_gitlab_ssh_key_path
         self.renku_base_project_url = renku_base_project_url
+        self.vo_mysql_pg_host = vo_mysql_pg_host
+        self.vo_mysql_pg_user = vo_mysql_pg_user
+        self.vo_mysql_pg_password = vo_mysql_pg_password
+        self.vo_mysql_pg_db = vo_mysql_pg_db
 
     def get_data_serve_conf(self, instr_name):
         if instr_name in self.data_server_conf_dict.keys():

requirements.txt

@@ -29,6 +29,8 @@ GitPython
 nbformat
 sentry-sdk
 pytest-sentry
+sqlparse
+queryparser-python3
 -e git+https://github.com/oda-hub/oda_api.git#egg=oda_api
 
 MarkupSafe==2.0.1

setup.py

@@ -45,6 +45,8 @@
     "black>=22.10.0",
     "bs4",
     "GitPython",
+    "sqlparse",
+    "queryparser-python3",
     "nbformat",
     "giturlparse",
     "sentry-sdk",