Source for specific tables

CHANGELOG.md

@@ -1,3 +1,10 @@
+# dbt-codegen v0.4.1
+
+## Features
+- Add optional `table_names` arg to `generate_source` (#50 @rahulj51)
+
+
+
 # dbt-codegen v0.4.0
 
 ## Breaking changes

README.md

@@ -21,6 +21,7 @@ which you can then paste into a schema file.
 * `schema_name` (required): The schema name that contains your source data
 * `database_name` (optional, default=target.database): The database that your
 source data is in.
+* `table_names` (optional, default=None): A list of tables that you want to generate the source definitions for.
 * `generate_columns` (optional, default=False): Whether you want to add the
 column names to your source definition.
 * `include_descriptions` (optional, default=False): Whether you want to add 
@@ -43,7 +44,7 @@ or
 
 ```
 # for multiple arguments, use the dict syntax
-$ dbt run-operation generate_source --args '{"schema_name": "jaffle_shop", "database_name": "raw"}'
+$ dbt run-operation generate_source --args '{"schema_name": "jaffle_shop", "database_name": "raw", "table_names":"[table_1, table_2]"}'
 ```
 
 2. The YAML for the source will be logged to the command line

integration_tests/tests/test_generate_source_some_tables.sql

@@ -0,0 +1,28 @@
+{% set raw_schema = generate_schema_name('raw_data') %}
+
+-- test all args
+{% set actual_source_yaml = codegen.generate_source(
+    schema_name=raw_schema,
+    database_name=target.database,
+    table_names=['data__a_relation'],
+    generate_columns=True,
+    include_descriptions=True
+) %}
+
+
+{% set expected_source_yaml %}
+version: 2
+
+sources:
+  - name: {{ raw_schema | trim }}
+    tables:
+      - name: data__a_relation
+        columns:
+          - name: col_a
+            description: ""
+          - name: col_b
+            description: ""
+
+{% endset %}
+
+{{ assert_equal (actual_source_yaml | trim, expected_source_yaml | trim) }}

macros/generate_source.sql

@@ -14,7 +14,7 @@
 
 
 ---
-{% macro generate_source(schema_name, database_name=target.database, generate_columns=False, include_descriptions=False) %}
+{% macro generate_source(schema_name, database_name=target.database, table_names=None, generate_columns=False, include_descriptions=False) %}
 
 {% set sources_yaml=[] %}
 
@@ -29,7 +29,11 @@
 
 {% do sources_yaml.append('    tables:') %}
 
+{% if table_names is none %}
 {% set tables=codegen.get_tables_in_schema(schema_name, database_name) %}
+{% else %}
+{% set tables=table_names %}
+{% endif %}
 
 {% for table in tables %}
     {% do sources_yaml.append('      - name: ' ~ table | lower ) %}