1 # Copyright 2017 Sandvine
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
23 def __init__(self
, url
, user
="admin", password
="admin"):
26 self
._password
= password
27 self
._http
_header
= None
29 def set_http_header(self
, header
):
30 self
._http
_header
= header
32 def _get_curl_cmd(self
, endpoint
):
33 curl_cmd
= pycurl
.Curl()
34 curl_cmd
.setopt(pycurl
.URL
, self
._url
+ endpoint
)
35 curl_cmd
.setopt(pycurl
.SSL_VERIFYPEER
, 0)
36 curl_cmd
.setopt(pycurl
.SSL_VERIFYHOST
, 0)
37 curl_cmd
.setopt(pycurl
.USERPWD
, "{}:{}".format(self
._user
, self
._password
))
39 curl_cmd
.setopt(pycurl
.HTTPHEADER
, self
._http
_header
)
42 def get_cmd(self
, endpoint
):
45 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
46 curl_cmd
.setopt(pycurl
.HTTPGET
, 1)
47 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
49 "Request METHOD: {} URL: {}".format("GET", self
._url
+ endpoint
)
52 http_code
= curl_cmd
.getinfo(pycurl
.HTTP_CODE
)
53 self
._logger
.info("Response HTTPCODE: {}".format(http_code
))
57 "Response DATA: {}".format(json
.loads(data
.getvalue().decode()))
59 return json
.loads(data
.getvalue().decode())
62 def delete_cmd(self
, endpoint
):
64 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
65 curl_cmd
.setopt(pycurl
.CUSTOMREQUEST
, "DELETE")
66 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
68 "Request METHOD: {} URL: {}".format("DELETE", self
._url
+ endpoint
)
71 http_code
= curl_cmd
.getinfo(pycurl
.HTTP_CODE
)
72 self
._logger
.info("Response HTTPCODE: {}".format(http_code
))
76 "Response DATA: {}".format(json
.loads(data
.getvalue().decode()))
78 return json
.loads(data
.getvalue().decode())
88 curl_cmd
= self
._get
_curl
_cmd
(endpoint
)
89 curl_cmd
.setopt(pycurl
.POST
, 1)
90 curl_cmd
.setopt(pycurl
.WRITEFUNCTION
, data
.write
)
92 if postfields_dict
is not None:
93 jsondata
= json
.dumps(postfields_dict
)
94 curl_cmd
.setopt(pycurl
.POSTFIELDS
, jsondata
)
96 if formfile
is not None:
98 pycurl
.HTTPPOST
, [((formfile
[0], (pycurl
.FORM_FILE
, formfile
[1])))]
102 "Request METHOD: {} URL: {}".format("POST", self
._url
+ endpoint
)
105 http_code
= curl_cmd
.getinfo(pycurl
.HTTP_CODE
)
106 self
._logger
.info("Response HTTPCODE: {}".format(http_code
))
110 "Response DATA: {}".format(json
.loads(data
.getvalue().decode()))
112 return json
.loads(data
.getvalue().decode())