From 7355e8bc56b1f042602756f9f47287cf78839e84 Mon Sep 17 00:00:00 2001 From: Sourcery AI <> Date: Tue, 20 Apr 2021 01:54:44 +0000 Subject: [PATCH] 'Refactored by Sourcery' --- serpapi/serp_api_client.py | 8 +++----- testwrapper.py | 2 +- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/serpapi/serp_api_client.py b/serpapi/serp_api_client.py index 2502c2d..2d4f09d 100644 --- a/serpapi/serp_api_client.py +++ b/serpapi/serp_api_client.py @@ -28,9 +28,8 @@ def construct_url(self, path = "/search"): self.params_dict['source'] = 'python' if self.SERP_API_KEY: self.params_dict['serp_api_key'] = self.SERP_API_KEY - if self.engine: - if not 'engine' in self.params_dict: - self.params_dict['engine'] = self.engine + if self.engine and 'engine' not in self.params_dict: + self.params_dict['engine'] = self.engine return self.BACKEND + path, self.params_dict def get_results(self, path = '/search'): @@ -94,8 +93,7 @@ def get_location(self, q, limit = 5): Returns: Dict: Location matching q """ - self.params_dict = {} - self.params_dict["output"] = "json" + self.params_dict = {"output": "json"} self.params_dict["q"] = q self.params_dict["limit"] = limit return json.loads(self.get_results('/locations.json')) diff --git a/testwrapper.py b/testwrapper.py index 8201848..10ace44 100644 --- a/testwrapper.py +++ b/testwrapper.py @@ -17,7 +17,7 @@ def test_query(): if len(sys.argv) == 5: params['country'] = sys.argv[4] if len(sys.argv) == 6: - params['num_results'] = sys.argv[5] if sys.argv[5] else None + params['num_results'] = sys.argv[5] or None # print query, location query_instance = QUERY(None) return query_instance.retrieve_html(params)