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)
53 elif http_code
== 404:
55 return json
.loads(data
.getvalue().decode())
59 return json
.loads(data
.getvalue().decode())
62 def send_cmd(self
, endpoint
='', postfields_dict
=None,
63 formfile
=None, filename
=None,
66 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
68 curl_cmd
.setopt(pycurl
.PUT
, 1)
70 curl_cmd
.setopt(pycurl
.POST
, 1)
71 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
73 if postfields_dict
is not None:
74 jsondata
= json
.dumps(postfields_dict
)
75 curl_cmd
.setopt(pycurl
.POSTFIELDS
, jsondata
)
76 elif formfile
is not None:
82 elif filename
is not None:
83 with
open(filename
, 'r') as stream
:
84 postdata
=stream
.read()
85 curl_cmd
.setopt(pycurl
.POSTFIELDS
, postdata
)
88 http_code
= curl_cmd
.getinfo(pycurl
.HTTP_CODE
)
90 if http_code
not in (200, 201, 202, 204):
91 raise ClientException(data
.getvalue().decode())
92 if postfields_dict
is not None:
94 return json
.loads(data
.getvalue().decode())
96 elif formfile
is not None:
98 return yaml
.safe_load(data
.getvalue().decode())
100 elif filename
is not None:
102 return yaml
.safe_load(data
.getvalue().decode())
106 def post_cmd(self
, endpoint
='', postfields_dict
=None,
107 formfile
=None, filename
=None):
108 return self
.send_cmd(endpoint
=endpoint
,
109 postfields_dict
=postfields_dict
,
114 def put_cmd(self
, endpoint
='', postfields_dict
=None,
115 formfile
=None, filename
=None):
116 return self
.send_cmd(endpoint
=endpoint
,
117 postfields_dict
=postfields_dict
,
122 def get2_cmd(self
, endpoint
):
124 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
125 curl_cmd
.setopt(pycurl
.HTTPGET
, 1)
126 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
129 return data
.getvalue()