xCAT/confluent
2
0
Fork 0
mirror of https://github.com/xcat2/confluent.git synced 2024-11-25 11:01:09 +00:00
Code Issues Projects Releases Wiki Activity

Add binary output to apiclient

Browse Source 
It is convenient to use apiclient as
a downloader, it's shorter and to the point
than curl.
This commit is contained in:
Jarrod Johnson 2020-06-19 07:49:05 -04:00
parent e7b93dba50
commit 17ec74dcf8
2 changed files with 36 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
confluent_osdeploy/esxi7/initramfs/opt/confluent/bin/apiclient
View File

@ -1,5 +1,8 @@
#!/usr/bin/python
import http.client as client
try:
import http.client as client
except ImportError:
import httplib as client
import socket
import ssl
import sys
@ -37,20 +40,30 @@ class HTTPSClient(client.HTTPConnection, object):
ctx.check_hostname = True
self.sock = ctx.wrap_socket(psock, server_hostname=host)
def grab_url(self, url, data=None):
def grab_url(self, url, data=None, returnrsp=False):
if data:
method = 'POST'
else:
method = 'GET'
self.request(method, url, headers=self.stdheaders)
rsp = self.getresponse()
body = rsp.read()
if rsp.status >= 200 and rsp.status < 300:
return body
raise Exception(body)
if returnrsp:
return rsp
else:
return rsp.read()
raise Exception(rsp.read())
if __name__ == '__main__':
data = None
if len(sys.argv) == 4:
with open(sys.argv[3], 'wb') as outf:
reader = HTTPSClient().grab_url(sys.argv[1], data, returnrsp=True)
chunk = reader.read(16384)
while chunk:
outf.write(chunk)
chunk = reader.read(16384)
sys.exit(0)
if len(sys.argv) == 3:
data = open(sys.argv[2]).read()
print(HTTPSClient().grab_url(sys.argv[1], data).decode())

23
confluent_osdeploy/rhvh4/initramfs/opt/confluent/bin/apiclient
View File

@ -1,5 +1,8 @@
#!/usr/bin/python
import httplib as client
try:
import http.client as client
except ImportError:
import httplib as client
import socket
import ssl
import sys
@ -37,20 +40,30 @@ class HTTPSClient(client.HTTPConnection, object):
ctx.check_hostname = True
self.sock = ctx.wrap_socket(psock, server_hostname=host)
def grab_url(self, url, data=None):
def grab_url(self, url, data=None, returnrsp=False):
if data:
method = 'POST'
else:
method = 'GET'
self.request(method, url, headers=self.stdheaders)
rsp = self.getresponse()
body = rsp.read()
if rsp.status >= 200 and rsp.status < 300:
return body
raise Exception(body)
if returnrsp:
return rsp
else:
return rsp.read()
raise Exception(rsp.read())
if __name__ == '__main__':
data = None
if len(sys.argv) == 4:
with open(sys.argv[3], 'wb') as outf:
reader = HTTPSClient().grab_url(sys.argv[1], data, returnrsp=True)
chunk = reader.read(16384)
while chunk:
outf.write(chunk)
chunk = reader.read(16384)
sys.exit(0)
if len(sys.argv) == 3:
data = open(sys.argv[2]).read()
print(HTTPSClient().grab_url(sys.argv[1], data).decode())