c89e91b979318bdca0824b9110be922138e8d23a
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 json
.loads(data
.getvalue().decode())
54 elif http_code
== 404:
56 elif http_code
>= 300:
61 def send_cmd(self
, endpoint
='', postfields_dict
=None,
62 formfile
=None, filename
=None,
65 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
67 curl_cmd
.setopt(pycurl
.PUT
, 1)
69 curl_cmd
.setopt(pycurl
.POST
, 1)
70 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
72 if postfields_dict
is not None:
73 jsondata
= json
.dumps(postfields_dict
)
74 curl_cmd
.setopt(pycurl
.POSTFIELDS
, jsondata
)
75 elif formfile
is not None:
81 elif filename
is not None:
82 with
open(filename
, 'r') as stream
:
83 postdata
=stream
.read()
84 curl_cmd
.setopt(pycurl
.POSTFIELDS
, postdata
)
87 http_code
= curl_cmd
.getinfo(pycurl
.HTTP_CODE
)
89 if http_code
not in (200, 201, 202, 204):
90 raise ClientException(data
.getvalue().decode())
91 if postfields_dict
is not None:
93 return json
.loads(data
.getvalue().decode())
95 elif formfile
is not None:
97 return yaml
.safe_load(data
.getvalue().decode())
99 elif filename
is not None:
101 return yaml
.safe_load(data
.getvalue().decode())
105 def post_cmd(self
, endpoint
='', postfields_dict
=None,
106 formfile
=None, filename
=None):
107 return self
.send_cmd(endpoint
=endpoint
,
108 postfields_dict
=postfields_dict
,
113 def put_cmd(self
, endpoint
='', postfields_dict
=None,
114 formfile
=None, filename
=None):
115 return self
.send_cmd(endpoint
=endpoint
,
116 postfields_dict
=postfields_dict
,
121 def get2_cmd(self
, endpoint
):
123 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
124 curl_cmd
.setopt(pycurl
.HTTPGET
, 1)
125 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
128 return data
.getvalue()