Handle missing entries in fixed_headers

hotgluexyz · May 20, 2021 · 52b2f7f · 52b2f7f
1 parent 12b871f
commit 52b2f7f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 from setuptools import setup
 
 setup(name='target-csv',
-      version='0.3.2',
+      version='0.3.3',
       description='Singer.io target for writing CSV files',
       author='Stitch',
       url='https://singer.io',

diff --git a/target_csv.py b/target_csv.py
@@ -68,7 +68,7 @@ def persist_messages(delimiter, quotechar, messages, destination_path, fixed_hea
             # flattened_record = flatten(o['record'])
             flattened_record = o['record']
 
-            if fixed_headers is not None:
+            if fixed_headers is not None and o['stream'] in fixed_headers:
                 if o['stream'] not in headers:
                     headers[o['stream']] = fixed_headers[o['stream']]
             else: