mirror of
https://github.com/tahoe-lafs/tahoe-lafs.git
synced 2024-12-29 01:08:54 +00:00
6ed3a23c62
It doesn't exit properly afterward, and it might not do the best things with non-success responses from the server. (See tahoe-put-web2ish.py for an example of better response handling.)
76 lines
2.0 KiB
Python
76 lines
2.0 KiB
Python
#!/usr/bin/env python
|
|
|
|
import re, socket, sys
|
|
|
|
SERVERURL_RE=re.compile("http://([^:]*)(:([1-9][0-9]*))?")
|
|
|
|
def put(serverurl, vdrive, vdrive_fname, local_fname, verbosity):
|
|
"""
|
|
@param verbosity: 0, 1, or 2, meaning quiet, verbose, or very verbose
|
|
|
|
@return: a Deferred which eventually fires with the exit code
|
|
"""
|
|
mo = SERVERURL_RE.match(serverurl)
|
|
if not mo:
|
|
raise ValueError("serverurl is required to look like \"http://HOSTNAMEORADDR:PORT\"")
|
|
host = mo.group(1)
|
|
port = int(mo.group(3))
|
|
|
|
url = "/vdrive/" + vdrive + "/"
|
|
if vdrive_fname:
|
|
url += vdrive_fname
|
|
|
|
if local_fname is None or local_fname == "-":
|
|
infileobj = sys.stdin
|
|
else:
|
|
infileobj = open(local_fname, "rb")
|
|
|
|
so = socket.socket()
|
|
so.connect((host, port,))
|
|
|
|
CHUNKSIZE=2**16
|
|
data = "PUT %s HTTP/1.1\r\nHostname: %s\r\n\r\n" % (url, host,)
|
|
while data:
|
|
try:
|
|
sent = so.send(data)
|
|
except Exception, le:
|
|
print "got socket error: %s" % (le,)
|
|
return -1
|
|
|
|
if sent == len(data):
|
|
data = infileobj.read(CHUNKSIZE)
|
|
else:
|
|
data = data[sent:]
|
|
|
|
respbuf = []
|
|
data = so.recv(CHUNKSIZE)
|
|
while data:
|
|
print "debuggery 1 okay now we've got some more data: %r" % (data,)
|
|
respbuf.append(data)
|
|
data = so.recv(CHUNKSIZE)
|
|
|
|
so.shutdown(socket.SHUT_WR)
|
|
data = so.recv(CHUNKSIZE)
|
|
while data:
|
|
print "debuggery 2 okay now we've got some more data: %r" % (data,)
|
|
respbuf.append(data)
|
|
data = so.recv(CHUNKSIZE)
|
|
|
|
def main():
|
|
import optparse
|
|
parser = optparse.OptionParser()
|
|
parser.add_option("-d", "--vdrive", dest="vdrive", default="global")
|
|
parser.add_option("-s", "--server", dest="server", default="http://tahoebs1.allmydata.com:8011")
|
|
|
|
(options, args) = parser.parse_args()
|
|
|
|
local_file = args[0]
|
|
vdrive_file = None
|
|
if len(args) > 1:
|
|
vdrive_file = args[1]
|
|
|
|
put(options.server, options.vdrive, vdrive_file, local_file)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|