add strip function to remove property in option

Nasdaq · couture-ql · Jun 10, 2022 · Jun 13, 2022 · Jun 15, 2022 · Jun 15, 2022
commit 78cacaecb95ce7581f0487adc07d824eea6914b0
diff --git a/nasdaqdatalink/connection.py b/nasdaqdatalink/connection.py
@@ -14,6 +14,11 @@
     AuthenticationError, ForbiddenError, InvalidRequestError,
     NotFoundError, ServiceUnavailableError)
 
+KW_TO_REMOVE = [
+    'session',
+    'api_config'
+]
+
 
 class Connection:
     @classmethod
@@ -48,9 +53,7 @@ def execute_request(cls, http_verb, url, **options):
 
         api_config = get_config_from_kwargs(options)
 
-        # clean the request payload
-        options.get('params', {}).pop('session', None)
-        options.get('params', {}).pop('api_config', None)
+        cls.options_kw_strip(options)
         try:
             response = session.request(method=http_verb,
                                        url=url,
@@ -126,3 +129,8 @@ def handle_api_error(cls, resp):
         klass = d_klass.get(code_letter, DataLinkError)
 
         raise klass(message, resp.status_code, resp.text, resp.headers, code)
+
+    @classmethod
+    def options_kw_strip(self, options):
+        for kw in KW_TO_REMOVE:
+            options.get('params', {}).pop(kw, None)