1 # Copyright 2018 Telefonica
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
9 # http://www.apache.org/licenses/LICENSE-2.0
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
17 from io
import BytesIO
21 from osmclient
.common
import http
22 from osmclient
.common
.exceptions
import ClientException
24 class Http(http
.Http
):
26 def __init__(self
, url
, user
='admin', password
='admin'):
29 self
._password
= password
30 self
._http
_header
= None
32 def _get_curl_cmd(self
, endpoint
):
33 curl_cmd
= pycurl
.Curl()
34 #print self._url + endpoint
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)
39 curl_cmd
.setopt(pycurl
.HTTPHEADER
, self
._http
_header
)
42 def delete_cmd(self
, endpoint
):
44 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
45 curl_cmd
.setopt(pycurl
.CUSTOMREQUEST
, "DELETE")
46 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
48 http_code
= curl_cmd
.getinfo(pycurl
.HTTP_CODE
)
49 #print 'HTTP_CODE: {}'.format(http_code)
51 # TODO 202 accepted should be returned somehow
53 return http_code
, data
.getvalue().decode()
55 return http_code
, None
57 def send_cmd(self
, endpoint
='', postfields_dict
=None,
58 formfile
=None, filename
=None,
59 put_method
=False, patch_method
=False):
61 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
63 curl_cmd
.setopt(pycurl
.PUT
, 1)
65 curl_cmd
.setopt(pycurl
.CUSTOMREQUEST
, "PATCH")
67 curl_cmd
.setopt(pycurl
.POST
, 1)
68 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
70 if postfields_dict
is not None:
71 jsondata
= json
.dumps(postfields_dict
)
72 curl_cmd
.setopt(pycurl
.POSTFIELDS
, jsondata
)
73 elif formfile
is not None:
79 elif filename
is not None:
80 with
open(filename
, 'r') as stream
:
81 postdata
=stream
.read()
82 curl_cmd
.setopt(pycurl
.POSTFIELDS
, postdata
)
85 http_code
= curl_cmd
.getinfo(pycurl
.HTTP_CODE
)
88 return http_code
, data
.getvalue().decode()
90 return http_code
, None
92 def post_cmd(self
, endpoint
='', postfields_dict
=None,
93 formfile
=None, filename
=None):
94 return self
.send_cmd(endpoint
=endpoint
,
95 postfields_dict
=postfields_dict
,
98 put_method
=False, patch_method
=False)
100 def put_cmd(self
, endpoint
='', postfields_dict
=None,
101 formfile
=None, filename
=None):
102 return self
.send_cmd(endpoint
=endpoint
,
103 postfields_dict
=postfields_dict
,
106 put_method
=True, patch_method
=False)
108 def patch_cmd(self
, endpoint
='', postfields_dict
=None,
109 formfile
=None, filename
=None):
110 return self
.send_cmd(endpoint
=endpoint
,
111 postfields_dict
=postfields_dict
,
114 put_method
=False, patch_method
=True)
116 def get2_cmd(self
, endpoint
):
118 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
119 curl_cmd
.setopt(pycurl
.HTTPGET
, 1)
120 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
122 http_code
= curl_cmd
.getinfo(pycurl
.HTTP_CODE
)
125 return http_code
, data
.getvalue().decode()
126 return http_code
, None