bugfix (#19)

* README and priv bugs

* note

* version bump

---------

Co-authored-by: TJ Murphy <[email protected]>

Author: titan-teej

images/github-explainer.png

@@ -3,7 +3,7 @@ RETURNS OBJECT NOT NULL
 LANGUAGE PYTHON
 RUNTIME_VERSION = '3.9'
 PACKAGES = ('snowflake-snowpark-python', 'inflection', 'pyparsing')
-IMPORTS = ('@titan_aws/releases/titan-0.1.5.zip')
+IMPORTS = ('@titan_aws/releases/titan-0.1.6.zip')
 HANDLER = 'titan.spi.install'
 EXECUTE AS CALLER
 CALL install()

scripts/install

@@ -2,7 +2,7 @@
 
 setup(
     name="titan",
-    version="0.1.5",
+    version="0.1.6",
     description="Snowflake infrastructure as code",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",

setup.py

@@ -20,7 +20,7 @@
     "Warehouse",
 ]
 
-__version__ = "0.1.5"
+__version__ = "0.1.6"
 
 LOGO = r"""
     __  _ __          

titan/__init__.py

@@ -226,8 +226,12 @@ def _contains(role, principal, priv):
         if role_grants:
             for principal, grant_list in role_grants.items():
                 for grant in grant_list:
-                    priv = priv_for_principal(parse_URN(principal), grant["priv"])
-                    _add(role, principal, priv)
+                    try:
+                        priv = priv_for_principal(parse_URN(principal), grant["priv"])
+                        _add(role, principal, priv)
+                    except ValueError:
+                        # Priv not recognized by Titan
+                        pass
 
         # Implied privilege grants in the context of our plan
         for action, urn_str, data in plan:
@@ -483,8 +487,6 @@ def generate_manifest(self, session_context: dict = {}):
 
             urns.append(manifest_key)
 
-            print(urn, resource)
-
             for ref in resource.refs:
                 ref_urn = URN.from_resource(account_locator=self._account_locator, resource=ref)
                 refs.append((str(urn), str(ref_urn)))
@@ -586,7 +588,6 @@ def destroy(self, session, manifest=None):
                 try:
                     execute(session, lifecycle.drop_resource(urn, data))
                 except snowflake.connector.errors.ProgrammingError as err:
-                    print("failed")
                     continue
 
     def _add(self, resource: Resource):

titan/blueprint.py

@@ -219,6 +219,7 @@ def drop_future_grant(urn: URN, data: dict, **kwargs):
 
 
 def drop_grant(urn: URN, data: dict, **kwargs):
+    # FIXME
     if data["priv"] == "OWNERSHIP":
         return "select 1"
     return tidy_sql(

titan/lifecycle.py

@@ -110,6 +110,7 @@ class AlertPriv(ParseableEnum):
 
 class DatabasePriv(ParseableEnum):
     ALL = "ALL"
+    APPLYBUDGET = "APPLYBUDGET"
     CREATE_DATABASE_ROLE = "CREATE DATABASE ROLE"
     CREATE_SCHEMA = "CREATE SCHEMA"
     IMPORTED_PRIVILEGES = "IMPORTED PRIVILEGES"
@@ -198,13 +199,18 @@ class RolePriv(ParseableEnum):
 class SchemaPriv(ParseableEnum):
     ALL = "ALL"
     ADD_SEARCH_OPTIMIZATION = "ADD SEARCH OPTIMIZATION"
+    APPLYBUDGET = "APPLYBUDGET"
     CREATE_ALERT = "CREATE ALERT"
+    CREATE_AUTHENTICATION_POLICY = "CREATE AUTHENTICATION POLICY"
     CREATE_DYNAMIC_TABLE = "CREATE DYNAMIC TABLE"
     CREATE_EXTERNAL_TABLE = "CREATE EXTERNAL TABLE"
     CREATE_FILE_FORMAT = "CREATE FILE FORMAT"
     CREATE_FUNCTION = "CREATE FUNCTION"
+    CREATE_HYBRID_TABLE = "CREATE HYBRID TABLE"
+    CREATE_ICEBERG_TABLE = "CREATE ICEBERG TABLE"
     CREATE_MASKING_POLICY = "CREATE MASKING POLICY"
     CREATE_MATERIALIZED_VIEW = "CREATE MATERIALIZED VIEW"
+    CREATE_MODEL = "CREATE MODEL"
     CREATE_NETWORK_RULE = "CREATE NETWORK RULE"
     CREATE_PACKAGES_POLICY = "CREATE PACKAGES POLICY"
     CREATE_PASSWORD_POLICY = "CREATE PASSWORD POLICY"
@@ -218,6 +224,7 @@ class SchemaPriv(ParseableEnum):
     CREATE_SNOWFLAKE_ML_FORECAST = "CREATE SNOWFLAKE.ML.FORECAST"
     CREATE_STAGE = "CREATE STAGE"
     CREATE_STREAM = "CREATE STREAM"
+    CREATE_STREAMLIT = "CREATE STREAMLIT"
     CREATE_TABLE = "CREATE TABLE"
     CREATE_TAG = "CREATE TAG"
     CREATE_TASK = "CREATE TASK"
@@ -256,6 +263,7 @@ class StreamPriv(ParseableEnum):
 
 class TablePriv(ParseableEnum):
     ALL = "ALL"
+    APPLYBUDGET = "APPLYBUDGET"
     DELETE = "DELETE"
     INSERT = "INSERT"
     OWNERSHIP = "OWNERSHIP"
@@ -294,6 +302,7 @@ class ViewPriv(ParseableEnum):
 
 class WarehousePriv(ParseableEnum):
     ALL = "ALL"
+    APPLYBUDGET = "APPLYBUDGET"
     MODIFY = "MODIFY"
     MONITOR = "MONITOR"
     OPERATE = "OPERATE"