From 45405353f60a05a2d92e05d854635de050982637 Mon Sep 17 00:00:00 2001 From: David Marx Date: Sat, 12 May 2018 20:34:12 -0700 Subject: [PATCH] Fixed bugs in _add_nec_args. Created_utc repetition (#16) and incorrect string -> list encapsulation --- psaw/PushshiftAPI.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/psaw/PushshiftAPI.py b/psaw/PushshiftAPI.py index a4dbb82..7c91825 100644 --- a/psaw/PushshiftAPI.py +++ b/psaw/PushshiftAPI.py @@ -73,10 +73,14 @@ class PushshiftAPIMinimal(object): return if 'limit' not in payload: payload['limit'] = self.max_results_per_request - if 'filter' in payload and payload.get('created_utc', None) is None: + if 'filter' in payload: #and payload.get('created_utc', None) is None: if not isinstance(payload['filter'], list): - payload['filter'] = list(payload['filter']) - payload['filter'].append('created_utc') + if isinstance(payload['filter'], str): + payload['filter'] = [payload['filter']] + else: + payload['filter'] = list(payload['filter']) + if 'created_utc' not in payload['filter']: + payload['filter'].append('created_utc') def _get(self, kind, payload): self._add_nec_args(payload)