Fix requests

master
Sven Slootweg 12 years ago
parent 8a62287f5f
commit a89b36c34a

@ -57,45 +57,30 @@ class Filesystem:
def standard_request(self, destination, data=None): def standard_request(self, destination, data=None):
return self.do_json_request("/%s?t=json" % destination, data) return self.do_json_request("/%s?t=json" % destination, data)
def do_request(self, destination, data=None): def Directory(self, uri, data=None):
if data is not None: if data is None:
post_data = urllib.urlencode(data) data = requests.get("%s/uri/%s?t=json" % (self.url, urllib.quote(uri))).json
try:
if data is None: if data is None:
return urllib.urlopen("%s%s" % (self.url, destination)).read() raise FilesystemException("Could not reach the WAPI or did not receive a valid response.")
else:
return urllib.urlopen("%s%s" % (self.url, destination), post_data).read()
except urllib.URLError:
raise FilesystemException("The WAPI could not be reached.")
def do_json_request(self, destination, data=None):
results = self.do_request(destination, data)
try:
return json.loads(results)
except ValueError:
raise FilesystemException("Corrupted data was received from the WAPI")
def do_put_request(self, destination, data, headers):
request = urllib2.Request("%s%s" % (self.url, destination), data, headers)
return urllib2.urlopen(request).read()
def Directory(self, uri, data=None):
if data == None:
data = self.standard_request("uri/%s" % urllib.quote(uri))
return Directory(self, uri, data) return Directory(self, uri, data)
def File(self, uri, data=None): def File(self, uri, data=None):
if data == None: if data is None:
data = self.standard_request("uri/%s" % urllib.quote(uri)) data = requests.get("%s/uri/%s?t=json" % (self.url, urllib.quote(uri))).json
if data is None:
raise FilesystemException("Could not reach the WAPI or did not receive a valid response.")
return File(self, uri, data) return File(self, uri, data)
def Object(self, uri, data=None): def Object(self, uri, data=None):
if data == None: if data is None:
data = self.standard_request("uri/%s" % urllib.quote(uri)) data = requests.get("%s/uri/%s?t=json" % (self.url, urllib.quote(uri))).json
if data is None:
raise FilesystemException("Could not reach the WAPI or did not receive a valid response.")
if "filenode" in data: if "filenode" in data:
return self.File(uri, data) return self.File(uri, data)
@ -133,7 +118,7 @@ class Directory:
self.uri = uri self.uri = uri
if data is None: if data is None:
data = requests.get("%s/uri/%s" % (this.filesystem.url, urllib.quote(uri))).json data = requests.get("%s/uri/%s?t=json" % (self.filesystem.url, urllib.quote(uri))).json
if data is None: if data is None:
raise FilesystemException("Could not reach the WAPI or did not receive a valid response.") raise FilesystemException("Could not reach the WAPI or did not receive a valid response.")
@ -212,8 +197,11 @@ class File:
self.filesystem = filesystem self.filesystem = filesystem
self.uri = uri self.uri = uri
if data == None: if data is None:
data = self.filesystem.standard_request("uri/%s" % urllib.quote(uri)) data = requests.get("%s/uri/%s?t=json" % (self.filesystem.url, urllib.quote(uri))).json
if data is None:
raise FilesystemException("Could not reach the WAPI or did not receive a valid response.")
if "filenode" in data: if "filenode" in data:
details = data[1] details = data[1]

@ -11,4 +11,4 @@ import pytahoe
#print fs.upload("test.py") #print fs.upload("test.py")
fs = pytahoe.Filesystem("http://tahoe.ccnmtl.columbia.edu/") fs = pytahoe.Filesystem("http://tahoe.ccnmtl.columbia.edu/")
fs.Directory("URI:DIR2:cbk47f5lybaj5qh6bm6eedezwe:m525plntx47u44xvf44r6rliec3gp6yeyio7olndibtke75zb6fa").mount("test") print fs.Directory("URI:DIR2:cbk47f5lybaj5qh6bm6eedezwe:m525plntx47u44xvf44r6rliec3gp6yeyio7olndibtke75zb6fa").children

Loading…
Cancel
Save