fixup! REDSHIFT: standardize credential usage

spotify · Mar 22, 2017 · 615d8ff · 615d8ff
1 parent d8e406d
commit 615d8ff
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/luigi/contrib/redshift.py b/luigi/contrib/redshift.py
@@ -21,11 +21,6 @@
 import time
 import os
 
-try:
-    from ConfigParser import NoSectionError
-except:
-    from configparser import NoSectionError
-
 import luigi
 from luigi import postgres
 from luigi.contrib import rdbms
@@ -95,10 +90,7 @@ def aws_session_token(self):
     def _get_configuration_attribute(self, attribute):
         config = luigi.configuration.get_config()
 
-        try:
-            value = config.get(self.configuration_section, attribute)
-        except NoSectionError:
-            value = None
+        value = config.get(self.configuration_section, attribute, default=None)
 
         if not value:
             value = os.environ.get(attribute.upper(), None)