From 5526cc8b1687ce4823318c383966dd75b8df5ec0 Mon Sep 17 00:00:00 2001 From: Marek Isalski Date: Wed, 20 Jul 2022 11:23:34 +0100 Subject: [PATCH] refactor requests --- src/sipcentric/__init__.py | 37 ++++++++++++++----------------------- 1 file changed, 14 insertions(+), 23 deletions(-) diff --git a/src/sipcentric/__init__.py b/src/sipcentric/__init__.py index c23d1c4..209601f 100644 --- a/src/sipcentric/__init__.py +++ b/src/sipcentric/__init__.py @@ -52,33 +52,24 @@ class API(object): url = self.base + uri auth = requests.auth.HTTPBasicAuth(self.username, self.password) + kwargs = dict(auth=auth, verify=True, timeout=3.0) + if params: + kwargs.update(params=params) + if method == "GET": - if params: - r = requests.get(url, auth=auth, params=params, verify=True, timeout=3.0) - else: - r = requests.get(url, auth=auth, verify=True, timeout=3.0) + fn = requests.get elif method == "POST": + fn = requests.post headers = {"content-type": "application/json"} - if params: - r = requests.post( - url, - auth=auth, - headers=headers, - data=json.dumps(data), - params=params, - verify=True, - timeout=3.000, - ) - else: - r = requests.post( - url, - auth=auth, - headers=headers, - data=json.dumps(data), - verify=True, - timeout=3.000, - ) + kwargs.update(headers=headers,data=json.dumps(data)) + + elif method == "PUT": + fn = requests.post + headers = {"content-type": "application/json"} + kwargs.update(headers=headers,data=json.dumps(data)) + + r = fn(url, **kwargs) if (r.status_code == 200) or (r.status_code == 201): try: