Preserve quotes round trip (#33)

dbt_invoke/internal/_utils.py

@@ -83,6 +83,7 @@ def parse_yaml(location):
     :return: The contents of the yaml file
     """
     yaml = YAML(typ="rt")
+    yaml.preserve_quotes = True
     with open(location, 'r') as stream:
         try:
             parsed_yaml = yaml.load(stream)
@@ -101,6 +102,7 @@ def write_yaml(location, data, mode='w'):
     :return: None
     """
     yaml = YAML(typ="rt")
+    yaml.preserve_quotes = True
     try:
         with open(location, mode) as stream:
             yaml.dump(data, stream)

dbt_invoke/internal/_version.py

@@ -1 +1 @@
-__version__ = '0.2.2'
+__version__ = '0.2.3'

tests/test_properties.py

@@ -211,6 +211,13 @@ def test_multiline_quotes(self):
             expected_file="customers_multiline_quotes_expected.yml",
         )
 
+    def test_preserve_quotes(self):
+        self.edit_update_compare(
+            "customers_preserve_quotes.yml",
+            target_model="customers",
+            expected_file="customers_preserve_quotes_expected.yml",
+        )
+
     def test_long_string(self):
         self.edit_update_compare(
             "customers_long_string.yml",

tests/test_property_files/customers_preserve_quotes.yml

@@ -0,0 +1,11 @@
+version: 2
+models:
+- name: customers
+  description: A description
+  config:
+    labels: {'contains_pii': 'no', 'created_by': 'somebody'}
+  columns:
+  - name: customer_id
+    description: ''
+  - name: created_at
+    description: ''

tests/test_property_files/customers_preserve_quotes_expected.yml

@@ -0,0 +1,11 @@
+version: 2
+models:
+- name: customers
+  description: A description
+  config:
+    labels: {'contains_pii': 'no', 'created_by': 'somebody'}
+  columns:
+  - name: customer_id
+    description: ''
+  - name: created_at
+    description: ''