add use_role option

.gitignore

@@ -2,4 +2,5 @@
 dist/
 *.egg-info/
 build/
-**/__pycache__/
+**/__pycache__/
+sprocketship/example/target

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sprocketship"
-version = "0.4.2"
+version = "0.5.0"
 authors = [
   { name="Nicklaus Roach", email="nicklausroach@gmail.com" },
 ]

sprocketship/cli.py

@@ -35,6 +35,10 @@ def liftoff(dir, show):
             rendered_proc = create_javascript_stored_procedure(
                 **proc, **{"project_dir": dir}
             )
+            if 'use_role' in proc.keys():
+                con.cursor().execute(f"USE ROLE {proc['use_role'].upper()}")
+            else:
+                con.cursor().execute(f"USE ROLE {data['snowflake']['role']}")
             con.cursor().execute(rendered_proc)
             msg = click.style(f"{proc['name']} ", fg="green", bold=True)
             msg += click.style(f"launched into schema ", fg="white", bold=True)
@@ -80,7 +84,7 @@ def build(dir, target):
             rendered_proc = create_javascript_stored_procedure(
                 **proc, **{"project_dir": dir}
             )
-            with open(os.path.join(dir, target, proc["name"] + ".sql"), "+a") as f:
+            with open(os.path.join(dir, target, proc["name"] + ".sql"), "w") as f:
                 f.write(rendered_proc)
             msg = click.style(f"{proc['name']} ", fg="green", bold=True)
             msg += click.style(f"successfully built", fg="white", bold=True)

sprocketship/example/.sprocketship.yml

@@ -11,6 +11,7 @@ procedures:
       database: !env_var SNOWFLAKE_DATABASE
       schema: !env_var SNOWFLAKE_SCHEMA
       language: javascript
+      use_role: data_engineer
       execute_as: owner
       args:
         - name: database_name

sprocketship/templates/javascript.sql

@@ -22,4 +22,4 @@ $$
 
 {{procedure_definition}}
 
-$$
+$$;