ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2025-04-14 22:26:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

control.py: minor improvements

Browse Source 
refs ticket:1737
This commit is contained in:
Brian Warner 2015-04-28 11:03:12 -07:00
parent a823362252
commit 341e263e03
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/allmydata/control.py
View File

@ -61,7 +61,8 @@ class ControlServer(Referenceable, service.Service):
return self.parent.debug_wait_for_client_connections(num_clients)
def remote_upload_random_data_from_file(self, size, convergence):
filename = tempfile.NamedTemporaryFile(delete=False).name
tempdir = tempfile.mkdtemp()
filename = os.path.join(tempdir, "data")
f = open(filename, "wb")
block = "a" * 8192
while size > 0:
@ -75,6 +76,7 @@ class ControlServer(Referenceable, service.Service):
d.addCallback(lambda results: results.get_uri())
def _done(uri):
os.remove(filename)
os.rmdir(tempdir)
return uri
d.addCallback(_done)
return d