quollio-core 0.6.2__tar.gz → 0.6.3__tar.gz

{quollio_core-0.6.2 → quollio_core-0.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quollio-core
-Version: 0.6.2
+Version: 0.6.3
 Summary: Quollio Core
 Author-email: quollio-dev <qt.dev@quollio.com>
 Maintainer-email: RyoAriyama <ryo.arym@gmail.com>, tharuta <35373297+TakumiHaruta@users.noreply.github.com>

{quollio_core-0.6.2 → quollio_core-0.6.3}/quollio_core/__init__.py

@@ -1,4 +1,4 @@
 """Quollio Core"""
 
-__version__ = "0.6.2"
+__version__ = "0.6.3"
 __author__ = "Quollio Technologies, Inc"

quollio_core-0.6.3/quollio_core/dbt_projects/snowflake/macros/materialization/get_imported_databases.sql

@@ -0,0 +1,45 @@
+{% macro get_imported_databases() %}
+    {%- set query %}
+        WITH DATABASES AS (
+            SELECT database_name
+            FROM snowflake.account_usage.databases
+            WHERE type = 'IMPORTED DATABASE'
+            AND database_owner IS NOT NULL
+            AND deleted is null
+        ), ROLE_PERMISSIONS AS (
+            SELECT
+              table_catalog
+              , ARRAY_AGG(grantee_name) grantee_names
+              , ARRAY_CONTAINS('{{ var("query_role") }}'::VARIANT, ARRAY_AGG(grantee_name)) AS is_role_contained
+            FROM
+              SNOWFLAKE.ACCOUNT_USAGE.GRANTS_TO_ROLES
+            WHERE
+              GRANTED_ON = 'DATABASE'
+              AND PRIVILEGE = 'USAGE'
+              AND DELETED_ON IS NULL
+            GROUP BY
+              table_catalog
+        )
+        SELECT
+          database_name
+        FROM
+          DATABASES
+        INNER JOIN
+          ROLE_PERMISSIONS
+        ON
+          DATABASES.database_name = ROLE_PERMISSIONS.table_catalog
+        WHERE
+          ROLE_PERMISSIONS.is_role_contained = TRUE
+    {%- endset %}
+
+    {%- set results = run_query(query) -%}
+    {%- if execute %}
+        {%- set all_databases = results.rows | map(attribute=0) | list %}
+        {{ log("Extracted Databases: " ~ all_databases, info=True) }}
+    {{ return(all_databases) }}
+    {%- else %}
+    {%- set all_databases = [] %}
+    {%- endif %}
+
+    {{ return(all_databases) }}
+{% endmacro %}

quollio_core-0.6.2/quollio_core/dbt_projects/snowflake/macros/materialization/get_imported_databases.sql

@@ -1,20 +0,0 @@
-{% macro get_imported_databases() %}
-    {%- set query %}
-        SELECT database_name
-        FROM snowflake.account_usage.databases
-        WHERE type = 'IMPORTED DATABASE'
-        AND database_owner IS NOT NULL
-        AND deleted is null
-    {%- endset %}
-
-    {%- set results = run_query(query) -%}
-    {%- if execute %}
-        {%- set all_databases = results.rows | map(attribute=0) | list %}
-        {{ log("Extracted Databases: " ~ all_databases, info=True) }}
-    {{ return(all_databases) }}
-    {%- else %}
-    {%- set all_databases = [] %}
-    {%- endif %}
-
-    {{ return(all_databases) }}
-{% endmacro %}