Make PASSWORD optional if private_key_file is specified

Snowflake-Labs · Dec 8, 2023 · 8eb66f8 · 8eb66f8
1 parent ca05756
commit 8eb66f8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/django_snowflake/base.py b/django_snowflake/base.py
@@ -90,6 +90,7 @@ class DatabaseWrapper(BaseDatabaseWrapper):
     introspection_class = DatabaseIntrospection
     ops_class = DatabaseOperations
 
+    password_not_required_options = ('private_key', 'private_key_file', 'authenticator')
     settings_is_missing = "settings.DATABASES is missing '%s' for 'django_snowflake'."
 
     def get_connection_params(self):
@@ -111,7 +112,7 @@ def get_connection_params(self):
 
         if settings_dict['PASSWORD']:
             conn_params['password'] = settings_dict['PASSWORD']
-        elif 'private_key' not in conn_params and 'authenticator' not in conn_params:
+        elif all(x not in conn_params for x in self.password_not_required_options):
             raise ImproperlyConfigured(self.settings_is_missing % 'PASSWORD')
 
         if settings_dict.get('ACCOUNT'):