Updated .gitignore: .idea
[osm/osmclient.git] / osmclient / common / http.py
1 # Copyright 2017 Sandvine
2 #
3 # All Rights Reserved.
4 #
5 # Licensed under the Apache License, Version 2.0 (the "License"); you may
6 # not use this file except in compliance with the License. You may obtain
7 # a copy of the License at
8 #
9 # http://www.apache.org/licenses/LICENSE-2.0
10 #
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
13 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
14 # License for the specific language governing permissions and limitations
15 # under the License.
16
17 from io import BytesIO
18 import pycurl
19 import json
20
21
22 class Http(object):
23
24 def __init__(self, url, user='admin', password='admin'):
25 self._url = url
26 self._user = user
27 self._password = password
28 self._http_header = None
29
30 def set_http_header(self, header):
31 self._http_header = header
32
33 def _get_curl_cmd(self, endpoint):
34 curl_cmd = pycurl.Curl()
35 curl_cmd.setopt(pycurl.URL, self._url + endpoint)
36 curl_cmd.setopt(pycurl.SSL_VERIFYPEER, 0)
37 curl_cmd.setopt(pycurl.SSL_VERIFYHOST, 0)
38 curl_cmd.setopt(
39 pycurl.USERPWD,
40 '{}:{}'.format(
41 self._user,
42 self._password))
43 if self._http_header:
44 curl_cmd.setopt(pycurl.HTTPHEADER, self._http_header)
45 return curl_cmd
46
47 def get_cmd(self, endpoint):
48
49 data = BytesIO()
50 curl_cmd = self._get_curl_cmd(endpoint)
51 curl_cmd.setopt(pycurl.HTTPGET, 1)
52 curl_cmd.setopt(pycurl.WRITEFUNCTION, data.write)
53 curl_cmd.perform()
54 curl_cmd.close()
55 if data.getvalue():
56 return json.loads(data.getvalue().decode())
57 return None
58
59 def delete_cmd(self, endpoint):
60 data = BytesIO()
61 curl_cmd = self._get_curl_cmd(endpoint)
62 curl_cmd.setopt(pycurl.CUSTOMREQUEST, "DELETE")
63 curl_cmd.setopt(pycurl.WRITEFUNCTION, data.write)
64 curl_cmd.perform()
65 curl_cmd.close()
66 if data.getvalue():
67 return json.loads(data.getvalue().decode())
68 return None
69
70 def post_cmd(self, endpoint='', postfields_dict=None, formfile=None, ):
71 data = BytesIO()
72 curl_cmd = self._get_curl_cmd(endpoint)
73 curl_cmd.setopt(pycurl.POST, 1)
74 curl_cmd.setopt(pycurl.WRITEFUNCTION, data.write)
75
76 if postfields_dict is not None:
77 jsondata = json.dumps(postfields_dict)
78 curl_cmd.setopt(pycurl.POSTFIELDS, jsondata)
79
80 if formfile is not None:
81 curl_cmd.setopt(
82 pycurl.HTTPPOST,
83 [((formfile[0],
84 (pycurl.FORM_FILE,
85 formfile[1])))])
86
87 curl_cmd.perform()
88 curl_cmd.close()
89 if data.getvalue():
90 return json.loads(data.getvalue().decode())
91 return None