osmclient/common/http.py

   1 # Copyright 2017 Sandvine
   2 #
   3 # All Rights Reserved.
   4 #
   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
   8 #
   9 #         http://www.apache.org/licenses/LICENSE-2.0
  10 #
  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
  15 #    under the License.
  16
  17 from io import BytesIO
  18 import pycurl
  19 import json
  20
  21
  22 class Http(object):
  23
  24     def __init__(self, url, user='admin', password='admin'):
  25         self._url = url
  26         self._user = user
  27         self._password = password
  28         self._http_header = None
  29
  30     def set_http_header(self, header):
  31         self._http_header = header
  32
  33     def _get_curl_cmd(self, endpoint):
  34         curl_cmd = pycurl.Curl()
  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)
  38         curl_cmd.setopt(
  39             pycurl.USERPWD,
  40             '{}:{}'.format(
  41                 self._user,
  42                 self._password))
  43         if self._http_header:
  44             curl_cmd.setopt(pycurl.HTTPHEADER, self._http_header)
  45         return curl_cmd
  46
  47     def get_cmd(self, endpoint):
  48
  49         data = BytesIO()
  50         curl_cmd = self._get_curl_cmd(endpoint)
  51         curl_cmd.setopt(pycurl.HTTPGET, 1)
  52         curl_cmd.setopt(pycurl.WRITEFUNCTION, data.write)
  53         curl_cmd.perform()
  54         curl_cmd.close()
  55         if data.getvalue():
  56             return json.loads(data.getvalue().decode())
  57         return None
  58
  59     def delete_cmd(self, endpoint):
  60         data = BytesIO()
  61         curl_cmd = self._get_curl_cmd(endpoint)
  62         curl_cmd.setopt(pycurl.CUSTOMREQUEST, "DELETE")
  63         curl_cmd.setopt(pycurl.WRITEFUNCTION, data.write)
  64         curl_cmd.perform()
  65         curl_cmd.close()
  66         if data.getvalue():
  67             return json.loads(data.getvalue().decode())
  68         return None
  69
  70     def post_cmd(self, endpoint='', postfields_dict=None, formfile=None, ):
  71         data = BytesIO()
  72         curl_cmd = self._get_curl_cmd(endpoint)
  73         curl_cmd.setopt(pycurl.POST, 1)
  74         curl_cmd.setopt(pycurl.WRITEFUNCTION, data.write)
  75
  76         if postfields_dict is not None:
  77             jsondata = json.dumps(postfields_dict)
  78             curl_cmd.setopt(pycurl.POSTFIELDS, jsondata)
  79
  80         if formfile is not None:
  81             curl_cmd.setopt(
  82                 pycurl.HTTPPOST,
  83                 [((formfile[0],
  84                            (pycurl.FORM_FILE,
  85                             formfile[1])))])
  86
  87         curl_cmd.perform()
  88         curl_cmd.close()
  89         if data.getvalue():
  90             return json.loads(data.getvalue().decode())
  91         return None