omnata-plugin-devkit 0.10.15a116__tar.gz → 0.10.15a118__tar.gz

{omnata_plugin_devkit-0.10.15a116 → omnata_plugin_devkit-0.10.15a118}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: omnata-plugin-devkit
-Version: 0.10.15a116
+Version: 0.10.15a118
 Summary: 
 Author: James Weakley
 Author-email: james.weakley@omnata.com

{omnata_plugin_devkit-0.10.15a116 → omnata_plugin_devkit-0.10.15a118}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-devkit"
-version = "0.10.15-a116"
+version = "0.10.15-a118"
 description = ""
 authors = ["James Weakley <james.weakley@omnata.com>"]
 readme = "README.md"

omnata_plugin_devkit-0.10.15a118/src/omnata_plugin_devkit/jinja_templates/FETCH_CONNECTIONS.sql.jinja

@@ -0,0 +1,26 @@
+CREATE or replace PROCEDURE PLUGIN.FETCH_CONNECTIONS()
+RETURNS TABLE()
+LANGUAGE SQL
+as
+DECLARE
+
+CONNECTIONS resultset;
+SYNC_ENGINE_DATABASE_NAME varchar;
+PROC_NAME varchar;
+PLUGIN_REGISTRATION_EXCEPTION EXCEPTION (-20001, 'Plugin is not registered to a sync engine instance');
+c1 CURSOR FOR select min(APPLICATION_NAME) from DATA.OMNATA_REGISTRATION;
+BEGIN
+
+OPEN c1;
+FETCH c1 INTO SYNC_ENGINE_DATABASE_NAME;
+IF (:SYNC_ENGINE_DATABASE_NAME = null) THEN
+   RAISE PLUGIN_REGISTRATION_EXCEPTION;
+END IF;
+
+select CONCAT(:SYNC_ENGINE_DATABASE_NAME,'.API.PLUGIN_FETCH_CONNECTIONS') into :PROC_NAME;
+
+CONNECTIONS:= (call IDENTIFIER(:PROC_NAME)(current_database(),PLUGIN.TOKEN()));
+
+RETURN TABLE(CONNECTIONS);
+
+END;

omnata_plugin_devkit-0.10.15a118/src/omnata_plugin_devkit/jinja_templates/FETCH_SYNC_BRANCHES.sql.jinja

@@ -0,0 +1,26 @@
+CREATE or replace PROCEDURE PLUGIN.FETCH_SYNC_BRANCHES(SYNC_ID numeric)
+RETURNS TABLE()
+LANGUAGE SQL
+as
+DECLARE
+
+SYNC_BRANCHES resultset;
+SYNC_ENGINE_DATABASE_NAME varchar;
+PROC_NAME varchar;
+PLUGIN_REGISTRATION_EXCEPTION EXCEPTION (-20001, 'Plugin is not registered to a sync engine instance');
+c1 CURSOR FOR select min(APPLICATION_NAME) from DATA.OMNATA_REGISTRATION;
+BEGIN
+
+OPEN c1;
+FETCH c1 INTO SYNC_ENGINE_DATABASE_NAME;
+IF (:SYNC_ENGINE_DATABASE_NAME = null) THEN
+   RAISE PLUGIN_REGISTRATION_EXCEPTION;
+END IF;
+
+select CONCAT(:SYNC_ENGINE_DATABASE_NAME,'.API.PLUGIN_FETCH_SYNC_BRANCHES') into :PROC_NAME;
+
+SYNC_BRANCHES:= (call IDENTIFIER(:PROC_NAME)(current_database(),PLUGIN.TOKEN(),:SYNC_ID));
+
+RETURN TABLE(SYNC_BRANCHES);
+
+END;

{omnata_plugin_devkit-0.10.15a116 → omnata_plugin_devkit-0.10.15a118}/src/omnata_plugin_devkit/plugin_uploader.py

@@ -469,7 +469,9 @@ class PluginUploader:
                 "CONFIGURE_APIS",
                 "UPDATE_API_CONFIGURATION",
                 "PENDING_API_CONFIGURATION",
+                "FETCH_CONNECTIONS",
                 "FETCH_SYNCS",
+                "FETCH_SYNC_BRANCHES",
                 "TEST_CALLBACK",
                 "LIST_STAGES",
                 "CREATE_GENERIC_SECRET_OBJECT",
@@ -575,9 +577,14 @@ $${}$$;
                 os.chdir(direct_imports_path)
                 # upload all files in the directory to the stage
                 for path in sorted(Path(".").rglob("*")):
-                    print(f"Uploading {path} to stage")
+                    # don't try to upload directories, only files
+                    if path.is_dir():
+                        continue
+                    # figure out the path
+                    relative_path = str(f"/{path.parent}") if path.parent != Path(".") else ""
+                    print(f"Uploading {path} to stage under {relative_path}")
                     self.session.sql(
-                        f"put file://{path} @{database_name}.{schema_name}.{stage_name} AUTO_COMPRESS=FALSE OVERWRITE=TRUE"
+                        f"put file://{path} @{database_name}.{schema_name}.{stage_name}{relative_path} AUTO_COMPRESS=FALSE OVERWRITE=TRUE"
                     ).collect()
                 os.chdir(cwd)