omnata-plugin-devkit 0.10.16a108__py3-none-any.whl → 0.10.16a110__py3-none-any.whl

omnata_plugin_devkit/plugin_uploader.py

@@ -20,6 +20,7 @@ import sys
 import typing
 import zipfile
 import copy
+import uuid
 from pathlib import Path
 from typing import Any, Dict, List, Literal, Optional
 
@@ -619,6 +620,28 @@ $${}$$;
 FROM '/streamlit'
 MAIN_FILE = '/plugin_configuration.py';"""
             )
+            
+            setup_script.write("""
+WITH CREATE_TOKEN_FUNCTION AS PROCEDURE()
+RETURNS BOOLEAN
+LANGUAGE JAVASCRIPT
+AS
+$$
+    var uuidStatement = snowflake.createStatement({ sqlText: 
+        `select UUID_STRING();`
+    });
+    var uuidResult = uuidStatement.execute();
+    uuidResult.next();
+    var uuid = uuidResult.getColumnValue(1);
+    var twoDollars='$' + '$';
+    snowflake.createStatement({ sqlText: 
+        `create or replace function PLUGIN.TOKEN() returns string as ${twoDollars}'${uuid}'${twoDollars} `
+    }).execute();
+    return true;
+$$
+CALL CREATE_TOKEN_FUNCTION();
+GRANT USAGE ON FUNCTION PLUGIN.TOKEN() TO application role OMNATA_MANAGEMENT;
+""")
 
             setup_script.write(
                 """GRANT USAGE ON STREAMLIT UI."Plugin Configuration" TO application role OMNATA_MANAGEMENT;\n"""
@@ -653,6 +676,15 @@ MAIN_FILE = '/plugin_configuration.py';"""
                 }
             )
             manifest_file.write(content)
+            custom_manifest_privileges = os.path.join(plugin_directory, "custom_manifest_privileges.yml")
+            if os.path.exists(custom_manifest_privileges):
+                print("Adding manifest privileges from custom_manifest_privileges.yml")
+                with open(
+                    custom_manifest_privileges, "r", encoding="utf-8"
+                ) as custom_manifest_privileges_stream:
+                    setup_script.write(custom_manifest_privileges_stream.read() + "\n")
+
+        
         print("Uploading app manifest to stage")
         self.session.sql(
             (

{omnata_plugin_devkit-0.10.16a108.dist-info → omnata_plugin_devkit-0.10.16a110.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: omnata-plugin-devkit
-Version: 0.10.16a108
+Version: 0.10.16a110
 Summary: 
 Author: James Weakley
 Author-email: james.weakley@omnata.com

{omnata_plugin_devkit-0.10.16a108.dist-info → omnata_plugin_devkit-0.10.16a110.dist-info}/RECORD

@@ -45,7 +45,7 @@ omnata_plugin_devkit/plugin_template/icon.svg,sha256=xEXOqutQhh8JaKU3xgyx9Weq1Kw
 omnata_plugin_devkit/plugin_template/plugin.py,sha256=VI2J64tRvrwuUbCHaB7SCO-QMisyeO1qTlamr9me_iM,15465
 omnata_plugin_devkit/plugin_template/plugin_development.ipynb,sha256=8_rkQnwWKe31MQp2rbM4ySjtET3FlAQcJVW8ix5Bh50,2912
 omnata_plugin_devkit/plugin_template/requirements.txt,sha256=nWMMGhIC_HIZHWTImksSLNU9mIAh0HlQHFVAS0rUKnA,30
-omnata_plugin_devkit/plugin_uploader.py,sha256=2Nl1UK5o5VwHWmcQtz5u2LaMC836ZSsK8VOYz5aIr5E,41053
+omnata_plugin_devkit/plugin_uploader.py,sha256=r30KQO-8LEb5PXcqN_fVoE1IxWbybfGa3uUHuhgli3Q,42247
 omnata_plugin_devkit/snowcli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omnata_plugin_devkit/snowcli/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omnata_plugin_devkit/snowcli/cli/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -70,7 +70,7 @@ omnata_plugin_devkit/snowcli/cli/templates/environment.yml.jinja,sha256=LRT00o5J
 omnata_plugin_devkit/streamlit/plugin_configuration.py,sha256=k89uspeQ_t5_Wbb_aXRL-fvbypXtax_r40oxxqB6fKc,100172
 omnata_plugin_devkit/test_step_definitions.py,sha256=Und60RpAn0YcJn-i1FpzUzVHRPnDB5QqO6YnZm2l6aw,53205
 omnata_plugin_devkit/utils.py,sha256=pFUltWEcb7Wxq0TPUdjvNPZG0AXHVrU-fZ874Q-EgUU,1378
-omnata_plugin_devkit-0.10.16a108.dist-info/LICENSE,sha256=rGaMQG3R3F5-JGDp_-rlMKpDIkg5n0SI4kctTk8eZSI,56
-omnata_plugin_devkit-0.10.16a108.dist-info/METADATA,sha256=mStInXbSRHO_gzjVOD0C0c6peo6FvkSi6YaKzkY8sZ8,1192
-omnata_plugin_devkit-0.10.16a108.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-omnata_plugin_devkit-0.10.16a108.dist-info/RECORD,,
+omnata_plugin_devkit-0.10.16a110.dist-info/LICENSE,sha256=rGaMQG3R3F5-JGDp_-rlMKpDIkg5n0SI4kctTk8eZSI,56
+omnata_plugin_devkit-0.10.16a110.dist-info/METADATA,sha256=QR7n6l7Gi3oQIkpjHVlHWKQcT6J43dp_6K-qBrLG_Hk,1192
+omnata_plugin_devkit-0.10.16a110.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+omnata_plugin_devkit-0.10.16a110.dist-info/RECORD,,