shaba/virt-manager
1
0
Fork 0
mirror of https://github.com/virt-manager/virt-manager.git synced 2025-02-04 17:47:19 +03:00
Code Issues Projects Releases Wiki Activity

urlfetcher: Use python style tempfiles

Browse Source 
Simplifies things WRT writing file contents
This commit is contained in:
Cole Robinson 2015-09-18 17:11:37 -04:00
parent cda3e381e6
commit b7beb0edb0
1 changed files with 24 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
virtinst/urlfetcher.py
View File

@ -70,29 +70,27 @@ class _URLFetcher(object):
ret += "/"
return ret + filename
def _saveTemp(self, fileobj, prefix):
def _saveTemp(self, urlobj, prefix):
"""
Save the fileobj contents to a temporary file, and return
the filename
"""
prefix = "virtinst-" + prefix
if "VIRTINST_TEST_SUITE" in os.environ:
fn = os.path.join("/tmp", prefix)
fd = os.open(fn, os.O_RDWR | os.O_CREAT, 0640)
filename = os.path.join("/tmp", prefix)
fileobj = file(filename, "w+b")
else:
(fd, fn) = tempfile.mkstemp(prefix=prefix,
dir=self.scratchdir)
fileobj = tempfile.NamedTemporaryFile(
dir=self.scratchdir, prefix=prefix, delete=False)
filename = fileobj.name
block_size = 16384
try:
while 1:
buff = fileobj.read(block_size)
if not buff:
break
os.write(fd, buff)
finally:
os.close(fd)
return fn
while 1:
buff = urlobj.read(block_size)
if not buff:
break
fileobj.write(buff)
return filename
##############
@ -122,28 +120,21 @@ class _URLFetcher(object):
Grab the passed filename from self.location and save it to
a temporary file, returning the temp filename
"""
url = self._make_full_url(filename)
base = os.path.basename(filename)
logging.debug("Fetching URI: %s", url)
f = None
try:
path = self._make_full_url(filename)
base = os.path.basename(filename)
logging.debug("Fetching URI: %s", path)
try:
f = grabber.urlopen(path,
progress_obj=self.meter,
text=_("Retrieving file %s...") % base)
except Exception, e:
raise ValueError(_("Couldn't acquire file %s: %s") %
(path, str(e)))
tmpname = self._saveTemp(f, prefix=base + ".")
logging.debug("Saved file to " + tmpname)
return tmpname
finally:
if f:
f.close()
urlobj = grabber.urlopen(url,
progress_obj=self.meter,
text=_("Retrieving file %s...") % base)
except Exception, e:
raise ValueError(_("Couldn't acquire file %s: %s") %
(url, str(e)))
tmpname = self._saveTemp(urlobj, prefix=base + ".")
logging.debug("Saved file to " + tmpname)
return tmpname
class _HTTPURLFetcher(_URLFetcher):