Added support for BigQuery relation renaming

CHANGELOG.md

@@ -5,6 +5,7 @@
 - Added a "docs" field to macros, with a "show" subfield to allow for hiding macros from the documentation site ([#2430](https://github.com/fishtown-analytics/dbt/issues/2430))
 - Added intersection syntax for model selector ([#2167](https://github.com/fishtown-analytics/dbt/issues/2167), [#2417](https://github.com/fishtown-analytics/dbt/pull/2417))
 - Extends model selection syntax with at most n-th parent/children `dbt run --models 3+m1+2` ([#2052](https://github.com/fishtown-analytics/dbt/issues/2052), [#2485](https://github.com/fishtown-analytics/dbt/pull/2485))
+- Added support for renaming BigQuery relations ([#2520](https://github.com/fishtown-analytics/dbt/issues/2520), [#2521](https://github.com/fishtown-analytics/dbt/pull/2521))
 
 ### Fixes
 - Fixed an error in create_adapter_plugins.py script when -dependency arg not passed ([#2507](https://github.com/fishtown-analytics/dbt/issues/2507), [#2508](https://github.com/fishtown-analytics/dbt/pull/2508))
@@ -14,7 +15,8 @@ Contributors:
  - [@raalsky](https://github.com/Raalsky) ([#2417](https://github.com/fishtown-analytics/dbt/pull/2417), [#2485](https://github.com/fishtown-analytics/dbt/pull/2485))
  - [@alf-mindshift](https://github.com/alf-mindshift) ([#2431](https://github.com/fishtown-analytics/dbt/pull/2431))
  - [@scarrucciu](https://github.com/scarrucciu) ([#2508](https://github.com/fishtown-analytics/dbt/pull/2508))
-- [@southpolemonkey](https://github.com/southpolemonkey)([#2511](https://github.com/fishtown-analytics/dbt/issues/2511))
+ - [@southpolemonkey](https://github.com/southpolemonkey)([#2511](https://github.com/fishtown-analytics/dbt/issues/2511))
+ - [@azhard](https://github.com/azhard) ([#2521](https://github.com/fishtown-analytics/dbt/pull/2521)
 
 ## dbt 0.17.0 (June 08, 2020)
 

plugins/bigquery/dbt/adapters/bigquery/impl.py

@@ -143,9 +143,30 @@ def truncate_relation(self, relation: BigQueryRelation) -> None:
     def rename_relation(
         self, from_relation: BigQueryRelation, to_relation: BigQueryRelation
     ) -> None:
-        raise dbt.exceptions.NotImplementedException(
-            '`rename_relation` is not implemented for this adapter!'
-        )
+
+        conn = self.connections.get_thread_connection()
+        client = conn.handle
+
+        from_table_ref = self.connections.table_ref(from_relation.database,
+                                                    from_relation.schema,
+                                                    from_relation.identifier,
+                                                    conn)
+        from_table = client.get_table(from_table_ref)
+        if from_table.table_type == "VIEW" or \
+                from_relation.type == RelationType.View or \
+                to_relation.type == RelationType.View:
+            raise dbt.exceptions.RuntimeException(
+                'Renaming of views is not currently supported in BigQuery'
+            )
+
+        to_table_ref = self.connections.table_ref(to_relation.database,
+                                                  to_relation.schema,
+                                                  to_relation.identifier,
+                                                  conn)
+
+        self.cache_renamed(from_relation, to_relation)
+        client.copy_table(from_table_ref, to_table_ref)
+        client.delete_table(from_table_ref)
 
     @available
     def list_schemas(self, database: str) -> List[str]:

plugins/bigquery/dbt/include/bigquery/macros/adapters.sql

@@ -123,3 +123,7 @@
 {% macro bigquery__alter_column_comment(relation, column_dict) -%}
   {% do adapter.update_column_descriptions(relation, column_dict) %}
 {% endmacro %}
+
+{% macro bigquery__rename_relation(from_relation, to_relation) -%}
+  {% do adapter.rename_relation(from_relation, to_relation) %}
+{% endmacro %}

test/integration/054_adapter_methods_test/macros/rename_named_relation.sql

@@ -0,0 +1,6 @@
+-- Macro to rename a relation
+{% macro rename_named_relation(from_name, to_name) %}
+{%- set from_relation = adapter.get_relation(database=target.database, schema=target.schema, identifier=from_name) -%}
+{%- set to_relation = adapter.get_relation(database=target.database, schema=target.schema, identifier=to_name) -%}
+{% do adapter.rename_relation(from_relation, to_relation) %}
+{% endmacro %}

test/integration/054_adapter_methods_test/models/renamed_model.sql

@@ -0,0 +1 @@
+select * from {{ source('test_source', 'renamed_seed') }}

test/integration/054_adapter_methods_test/models/sources.yml

@@ -0,0 +1,6 @@
+version: 2
+sources:
+  - name: test_source
+    schema: "{{ target.schema }}"
+    tables:
+      - name: renamed_seed

test/integration/054_adapter_methods_test/seed.sql

@@ -0,0 +1,111 @@
+create table {schema}.seed (
+	id BIGSERIAL PRIMARY KEY,
+	first_name VARCHAR(50),
+	last_name VARCHAR(50),
+	email VARCHAR(50),
+	gender VARCHAR(50),
+	ip_address VARCHAR(20)
+);
+
+
+insert into {schema}.seed (first_name, last_name, email, gender, ip_address) values
+('Jack', 'Hunter', 'jhunter0@pbs.org', 'Male', '59.80.20.168'),
+('Kathryn', 'Walker', 'kwalker1@ezinearticles.com', 'Female', '194.121.179.35'),
+('Gerald', 'Ryan', 'gryan2@com.com', 'Male', '11.3.212.243'),
+('Bonnie', 'Spencer', 'bspencer3@ameblo.jp', 'Female', '216.32.196.175'),
+('Harold', 'Taylor', 'htaylor4@people.com.cn', 'Male', '253.10.246.136'),
+('Jacqueline', 'Griffin', 'jgriffin5@t.co', 'Female', '16.13.192.220'),
+('Wanda', 'Arnold', 'warnold6@google.nl', 'Female', '232.116.150.64'),
+('Craig', 'Ortiz', 'cortiz7@sciencedaily.com', 'Male', '199.126.106.13'),
+('Gary', 'Day', 'gday8@nih.gov', 'Male', '35.81.68.186'),
+('Rose', 'Wright', 'rwright9@yahoo.co.jp', 'Female', '236.82.178.100'),
+('Raymond', 'Kelley', 'rkelleya@fc2.com', 'Male', '213.65.166.67'),
+('Gerald', 'Robinson', 'grobinsonb@disqus.com', 'Male', '72.232.194.193'),
+('Mildred', 'Martinez', 'mmartinezc@samsung.com', 'Female', '198.29.112.5'),
+('Dennis', 'Arnold', 'darnoldd@google.com', 'Male', '86.96.3.250'),
+('Judy', 'Gray', 'jgraye@opensource.org', 'Female', '79.218.162.245'),
+('Theresa', 'Garza', 'tgarzaf@epa.gov', 'Female', '21.59.100.54'),
+('Gerald', 'Robertson', 'grobertsong@csmonitor.com', 'Male', '131.134.82.96'),
+('Philip', 'Hernandez', 'phernandezh@adobe.com', 'Male', '254.196.137.72'),
+('Julia', 'Gonzalez', 'jgonzalezi@cam.ac.uk', 'Female', '84.240.227.174'),
+('Andrew', 'Davis', 'adavisj@patch.com', 'Male', '9.255.67.25'),
+('Kimberly', 'Harper', 'kharperk@foxnews.com', 'Female', '198.208.120.253'),
+('Mark', 'Martin', 'mmartinl@marketwatch.com', 'Male', '233.138.182.153'),
+('Cynthia', 'Ruiz', 'cruizm@google.fr', 'Female', '18.178.187.201'),
+('Samuel', 'Carroll', 'scarrolln@youtu.be', 'Male', '128.113.96.122'),
+('Jennifer', 'Larson', 'jlarsono@vinaora.com', 'Female', '98.234.85.95'),
+('Ashley', 'Perry', 'aperryp@rakuten.co.jp', 'Female', '247.173.114.52'),
+('Howard', 'Rodriguez', 'hrodriguezq@shutterfly.com', 'Male', '231.188.95.26'),
+('Amy', 'Brooks', 'abrooksr@theatlantic.com', 'Female', '141.199.174.118'),
+('Louise', 'Warren', 'lwarrens@adobe.com', 'Female', '96.105.158.28'),
+('Tina', 'Watson', 'twatsont@myspace.com', 'Female', '251.142.118.177'),
+('Janice', 'Kelley', 'jkelleyu@creativecommons.org', 'Female', '239.167.34.233'),
+('Terry', 'Mccoy', 'tmccoyv@bravesites.com', 'Male', '117.201.183.203'),
+('Jeffrey', 'Morgan', 'jmorganw@surveymonkey.com', 'Male', '78.101.78.149'),
+('Louis', 'Harvey', 'lharveyx@sina.com.cn', 'Male', '51.50.0.167'),
+('Philip', 'Miller', 'pmillery@samsung.com', 'Male', '103.255.222.110'),
+('Willie', 'Marshall', 'wmarshallz@ow.ly', 'Male', '149.219.91.68'),
+('Patrick', 'Lopez', 'plopez10@redcross.org', 'Male', '250.136.229.89'),
+('Adam', 'Jenkins', 'ajenkins11@harvard.edu', 'Male', '7.36.112.81'),
+('Benjamin', 'Cruz', 'bcruz12@linkedin.com', 'Male', '32.38.98.15'),
+('Ruby', 'Hawkins', 'rhawkins13@gmpg.org', 'Female', '135.171.129.255'),
+('Carlos', 'Barnes', 'cbarnes14@a8.net', 'Male', '240.197.85.140'),
+('Ruby', 'Griffin', 'rgriffin15@bravesites.com', 'Female', '19.29.135.24'),
+('Sean', 'Mason', 'smason16@icq.com', 'Male', '159.219.155.249'),
+('Anthony', 'Payne', 'apayne17@utexas.edu', 'Male', '235.168.199.218'),
+('Steve', 'Cruz', 'scruz18@pcworld.com', 'Male', '238.201.81.198'),
+('Anthony', 'Garcia', 'agarcia19@flavors.me', 'Male', '25.85.10.18'),
+('Doris', 'Lopez', 'dlopez1a@sphinn.com', 'Female', '245.218.51.238'),
+('Susan', 'Nichols', 'snichols1b@freewebs.com', 'Female', '199.99.9.61'),
+('Wanda', 'Ferguson', 'wferguson1c@yahoo.co.jp', 'Female', '236.241.135.21'),
+('Andrea', 'Pierce', 'apierce1d@google.co.uk', 'Female', '132.40.10.209'),
+('Lawrence', 'Phillips', 'lphillips1e@jugem.jp', 'Male', '72.226.82.87'),
+('Judy', 'Gilbert', 'jgilbert1f@multiply.com', 'Female', '196.250.15.142'),
+('Eric', 'Williams', 'ewilliams1g@joomla.org', 'Male', '222.202.73.126'),
+('Ralph', 'Romero', 'rromero1h@sogou.com', 'Male', '123.184.125.212'),
+('Jean', 'Wilson', 'jwilson1i@ocn.ne.jp', 'Female', '176.106.32.194'),
+('Lori', 'Reynolds', 'lreynolds1j@illinois.edu', 'Female', '114.181.203.22'),
+('Donald', 'Moreno', 'dmoreno1k@bbc.co.uk', 'Male', '233.249.97.60'),
+('Steven', 'Berry', 'sberry1l@eepurl.com', 'Male', '186.193.50.50'),
+('Theresa', 'Shaw', 'tshaw1m@people.com.cn', 'Female', '120.37.71.222'),
+('John', 'Stephens', 'jstephens1n@nationalgeographic.com', 'Male', '191.87.127.115'),
+('Richard', 'Jacobs', 'rjacobs1o@state.tx.us', 'Male', '66.210.83.155'),
+('Andrew', 'Lawson', 'alawson1p@over-blog.com', 'Male', '54.98.36.94'),
+('Peter', 'Morgan', 'pmorgan1q@rambler.ru', 'Male', '14.77.29.106'),
+('Nicole', 'Garrett', 'ngarrett1r@zimbio.com', 'Female', '21.127.74.68'),
+('Joshua', 'Kim', 'jkim1s@edublogs.org', 'Male', '57.255.207.41'),
+('Ralph', 'Roberts', 'rroberts1t@people.com.cn', 'Male', '222.143.131.109'),
+('George', 'Montgomery', 'gmontgomery1u@smugmug.com', 'Male', '76.75.111.77'),
+('Gerald', 'Alvarez', 'galvarez1v@flavors.me', 'Male', '58.157.186.194'),
+('Donald', 'Olson', 'dolson1w@whitehouse.gov', 'Male', '69.65.74.135'),
+('Carlos', 'Morgan', 'cmorgan1x@pbs.org', 'Male', '96.20.140.87'),
+('Aaron', 'Stanley', 'astanley1y@webnode.com', 'Male', '163.119.217.44'),
+('Virginia', 'Long', 'vlong1z@spiegel.de', 'Female', '204.150.194.182'),
+('Robert', 'Berry', 'rberry20@tripadvisor.com', 'Male', '104.19.48.241'),
+('Antonio', 'Brooks', 'abrooks21@unesco.org', 'Male', '210.31.7.24'),
+('Ruby', 'Garcia', 'rgarcia22@ovh.net', 'Female', '233.218.162.214'),
+('Jack', 'Hanson', 'jhanson23@blogtalkradio.com', 'Male', '31.55.46.199'),
+('Kathryn', 'Nelson', 'knelson24@walmart.com', 'Female', '14.189.146.41'),
+('Jason', 'Reed', 'jreed25@printfriendly.com', 'Male', '141.189.89.255'),
+('George', 'Coleman', 'gcoleman26@people.com.cn', 'Male', '81.189.221.144'),
+('Rose', 'King', 'rking27@ucoz.com', 'Female', '212.123.168.231'),
+('Johnny', 'Holmes', 'jholmes28@boston.com', 'Male', '177.3.93.188'),
+('Katherine', 'Gilbert', 'kgilbert29@altervista.org', 'Female', '199.215.169.61'),
+('Joshua', 'Thomas', 'jthomas2a@ustream.tv', 'Male', '0.8.205.30'),
+('Julie', 'Perry', 'jperry2b@opensource.org', 'Female', '60.116.114.192'),
+('Richard', 'Perry', 'rperry2c@oracle.com', 'Male', '181.125.70.232'),
+('Kenneth', 'Ruiz', 'kruiz2d@wikimedia.org', 'Male', '189.105.137.109'),
+('Jose', 'Morgan', 'jmorgan2e@webnode.com', 'Male', '101.134.215.156'),
+('Donald', 'Campbell', 'dcampbell2f@goo.ne.jp', 'Male', '102.120.215.84'),
+('Debra', 'Collins', 'dcollins2g@uol.com.br', 'Female', '90.13.153.235'),
+('Jesse', 'Johnson', 'jjohnson2h@stumbleupon.com', 'Male', '225.178.125.53'),
+('Elizabeth', 'Stone', 'estone2i@histats.com', 'Female', '123.184.126.221'),
+('Angela', 'Rogers', 'arogers2j@goodreads.com', 'Female', '98.104.132.187'),
+('Emily', 'Dixon', 'edixon2k@mlb.com', 'Female', '39.190.75.57'),
+('Albert', 'Scott', 'ascott2l@tinypic.com', 'Male', '40.209.13.189'),
+('Barbara', 'Peterson', 'bpeterson2m@ow.ly', 'Female', '75.249.136.180'),
+('Adam', 'Greene', 'agreene2n@fastcompany.com', 'Male', '184.173.109.144'),
+('Earl', 'Sanders', 'esanders2o@hc360.com', 'Male', '247.34.90.117'),
+('Angela', 'Brooks', 'abrooks2p@mtv.com', 'Female', '10.63.249.126'),
+('Harold', 'Foster', 'hfoster2q@privacy.gov.au', 'Male', '139.214.40.244'),
+('Carl', 'Meyer', 'cmeyer2r@disqus.com', 'Male', '204.117.7.88');

test/integration/054_adapter_methods_test/test_adapter_methods.py

@@ -1,4 +1,5 @@
 from test.integration.base import DBTIntegrationTest, use_profile
+import yaml
 
 
 class TestBaseCaching(DBTIntegrationTest):
@@ -40,3 +41,31 @@ def test_bigquery_adapter_methods(self):
         self.run_dbt(['compile'])  # trigger any compile-time issues
         self.run_dbt()
         self.assertTablesEqual('model', 'expected')
+
+
+class TestRenameRelation(DBTIntegrationTest):
+    @property
+    def schema(self):
+        return "rename_relation_054"
+
+    @property
+    def models(self):
+        return "models"
+
+    @property
+    def project_config(self):
+        return {
+            'config-version': 2,
+            'source-paths': ['models']
+        }
+
+    @use_profile('bigquery')
+    def test_bigquery_adapter_methods(self):
+        self.run_dbt(['compile'])  # trigger any compile-time issues
+        self.run_dbt(['seed'])
+        rename_relation_args = yaml.safe_dump({
+            'from_name': 'seed',
+            'to_name': 'renamed_seed',
+        })
+        self.run_dbt(['run-operation', 'rename_named_relation', '--args', rename_relation_args])
+        self.run_dbt()