Merge pull request #89 from heroku/content_type

set contentType to default value if necessary
heroku · Aug 26, 2020 · 1c1253c · 1c1253c
2 parents 54f437b + 04537ff
commit 1c1253c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/salesforce_bulk/salesforce_bulk.py b/salesforce_bulk/salesforce_bulk.py
@@ -309,7 +309,7 @@ def query(self, job_id, soql, contentType='CSV'):
                 re.search(re.compile(r"from (\w+)", re.I), soql).group(1),
                 "query", contentType=contentType)
 
-        job_content_type = self.job_content_types[job_id]
+        job_content_type = self.job_content_types.get(job_id, contentType)
         http_content_type = job_to_http_content_type[job_content_type]
 
         headers = self.headers(content_type=http_content_type)