Snowflake custom query block

Author: toniopoggi

Inputs/Databases/Snowflake-custom-query/README.md

@@ -0,0 +1,12 @@
+# Snowflake Custom Query
+
+Executes a SQL query on a Snowflake database
+
+## Language
+Python
+
+## Dependencies
+snowflake-connector-python
+
+## Source
+[script.py](https://github.com/visokio/omniscope-custom-blocks/blob/master/Connectors/Snowflake-custom-query/script.py)

Inputs/Databases/Snowflake-custom-query/manifest.json

@@ -0,0 +1,104 @@
+{
+    "@visokiotype": "CustomBlockSchema.CustomBlockManifest",
+    "name": "Snowflake Custom Query",
+    "scriptFilename": "script.py",
+    "language": "PYTHON",
+    "executableVersion": null,
+    "minVersions": [
+        null
+    ],
+    "optionsVersion": 1,
+    "apiVersion": "VERSION_0",
+    "isResourceIntensiveScript": false,
+    "showPartitioning": false,
+    "icon": null,
+    "description": "Executes a SQL query on a Snowflake database.",
+    "category": "Inputs",
+    "subcategory": "Databases",
+    "tags": [
+        "snowflake",
+        "database",
+        "sql"
+    ],
+    "introductoryText": "# Snowflake Custom SQL Executor\n\nUse this block to connect and execute SQL queries on a Snowflake database. Provide the necessary connection details and SQL statement.\n\nThe data is retrieved in full (in memory). For better performance please user the core **Database Input block* (select *Snowflake* in the connection options).",
+    "dependencies": "snowflake-connector-python",
+    "options": [
+        {
+            "name": "account",
+            "title": "Snowflake Account",
+            "description": "The Snowflake account identifier (e.g. 'kkpgiqe-sf91461').",
+            "groupTitle": "Connection",
+            "width": "ONE",
+            "@visokiotype": "CustomBlockSchema.TextCustomBlockPublicOption",
+            "mandatory": true,
+            "defaultValue": null
+        },
+        {
+            "name": "user",
+            "title": "Username",
+            "description": "The username for the Snowflake connection.",
+            "groupTitle": "Connection",
+            "width": "ONE",
+            "@visokiotype": "CustomBlockSchema.TextCustomBlockPublicOption",
+            "mandatory": true,
+            "defaultValue": null
+        },
+        {
+            "name": "password",
+            "title": "Password",
+            "description": "The password for the Snowflake user.",
+            "groupTitle": "Connection",
+            "width": "ONE",
+            "@visokiotype": "CustomBlockSchema.PasswordCustomBlockPublicOption",
+            "mandatory": true
+        },
+        {
+            "name": "warehouse",
+            "title": "Warehouse",
+            "description": "The Snowflake warehouse to use. E.g. COMPUTE_WH",
+            "groupTitle": "Connection",
+            "width": "ONE",
+            "@visokiotype": "CustomBlockSchema.TextCustomBlockPublicOption",
+            "mandatory": true,
+            "defaultValue": null
+        },
+        {
+            "name": "database",
+            "title": "Database",
+            "description": "The Snowflake database name. E.g. SNOWFLAKE_SAMPLE_DATA",
+            "groupTitle": "Connection",
+            "width": "ONE",
+            "@visokiotype": "CustomBlockSchema.TextCustomBlockPublicOption",
+            "mandatory": true,
+            "defaultValue": null
+        },
+        {
+            "name": "custom_sql",
+            "title": "Custom SQL Query",
+            "description": "Custom SQL statement to execute. Use schema.table syntax",
+            "groupTitle": "Query",
+            "width": "THREE",
+            "@visokiotype": "CustomBlockSchema.TextCustomBlockPublicOption",
+            "mandatory": true,
+            "defaultValue": null
+        }
+    ],
+    "blockOutputs": [
+        {
+            "@visokiotype": "CustomBlockSchema.BlockOutputPublicOption",
+            "id": "Output Data",
+            "label": "Query Results",
+            "displayName": "Query Results",
+            "tooltip": null
+        }
+    ],
+    "docker": {
+        "@visokiotype": "CustomBlockSchema.DockerCustomBlockPublicOption",
+        "customBaseImage": null,
+        "useCustomBaseImage": false,
+        "customSystemLibraries": null,
+        "installVisokioRepLibraries": false
+    },
+    "designLock": false,
+    "apiMode": "BATCH"
+}

Inputs/Databases/Snowflake-custom-query/script.py

@@ -0,0 +1,52 @@
+import pandas as pd
+from omniscope.api import OmniscopeApi
+import snowflake.connector
+
+# Initialize Omniscope API
+omniscope_api = OmniscopeApi()
+
+# Retrieve options from the block configuration
+account = omniscope_api.get_option('account')
+user = omniscope_api.get_option('user')
+password = omniscope_api.get_option('password')
+warehouse = omniscope_api.get_option('warehouse')
+database = omniscope_api.get_option('database')
+custom_sql = omniscope_api.get_option('custom_sql')
+
+# Establish a connection to Snowflake
+def connect_to_snowflake():
+    return snowflake.connector.connect(
+        user=user,
+        password=password,
+        account=account,
+        warehouse=warehouse,
+        database=database
+    )
+
+# Fetch data using the SQL query
+def fetch_data_from_snowflake(sql_query):
+    try:
+        # Connect to Snowflake
+        conn = connect_to_snowflake()
+        
+        # Execute the query and fetch data into a DataFrame
+        df = pd.read_sql(sql_query, conn)
+        
+        # Close the connection
+        conn.close()
+        
+        return df
+    except Exception as e:
+        omniscope_api.error(f"Error executing SQL query: {str(e)}")
+        return pd.DataFrame()  # Return an empty dataframe on error
+
+
+# Fetch the data
+output_data = fetch_data_from_snowflake(custom_sql)
+
+# Write the output records to the first output if there is data
+if output_data is not None and not output_data.empty:
+    omniscope_api.write_output_records(output_data, output_number=0)
+
+# Close the API after processing is complete
+omniscope_api.close()