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
, json
.loads(data
.getvalue().decode())
55 return http_code
, None
57 def send_cmd(self
, endpoint
='', postfields_dict
=None,
58 formfile
=None, filename
=None,
61 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
63 curl_cmd
.setopt(pycurl
.PUT
, 1)
65 curl_cmd
.setopt(pycurl
.POST
, 1)
66 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
68 if postfields_dict
is not None:
69 jsondata
= json
.dumps(postfields_dict
)
70 curl_cmd
.setopt(pycurl
.POSTFIELDS
, jsondata
)
71 elif formfile
is not None:
77 elif filename
is not None:
78 with
open(filename
, 'r') as stream
:
79 postdata
=stream
.read()
80 curl_cmd
.setopt(pycurl
.POSTFIELDS
, postdata
)
83 http_code
= curl_cmd
.getinfo(pycurl
.HTTP_CODE
)
85 if http_code
not in (200, 201, 202, 204):
86 raise ClientException(data
.getvalue().decode())
87 if postfields_dict
is not None:
89 return json
.loads(data
.getvalue().decode())
91 elif formfile
is not None:
93 return yaml
.safe_load(data
.getvalue().decode())
95 elif filename
is not None:
97 return yaml
.safe_load(data
.getvalue().decode())
101 def post_cmd(self
, endpoint
='', postfields_dict
=None,
102 formfile
=None, filename
=None):
103 return self
.send_cmd(endpoint
=endpoint
,
104 postfields_dict
=postfields_dict
,
109 def put_cmd(self
, endpoint
='', postfields_dict
=None,
110 formfile
=None, filename
=None):
111 return self
.send_cmd(endpoint
=endpoint
,
112 postfields_dict
=postfields_dict
,
117 def get2_cmd(self
, endpoint
):
119 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
120 curl_cmd
.setopt(pycurl
.HTTPGET
, 1)
121 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
124 return data
.getvalue()