comparison uploader/handlers.py @ 15:1ee374416987

handle more gracefully when no file is uploaded
author Jeff Hammel <jhammel@mozilla.com>
date Tue, 12 Jul 2011 09:18:29 -0700
parents 916d45d4f921
children d15f85eb2ab9
comparison
equal deleted inserted replaced
14:916d45d4f921 15:1ee374416987
51 fout.write(fin.file.read()) 51 fout.write(fin.file.read())
52 fout.close() 52 fout.close()
53 53
54 def __call__(self): 54 def __call__(self):
55 fin = self.request.POST['file'] 55 fin = self.request.POST['file']
56 _path = fin.filename.replace('..', '_') 56 try:
57 _path = fin.filename.replace('..', '_')
58 except AttributeError: # no file uploaded
59 return self.redirect(self.link('/'))
57 _path = _path.replace(os.path.sep, '_') 60 _path = _path.replace(os.path.sep, '_')
58 _path = os.path.join(self.app.directory, _path) 61 _path = os.path.join(self.app.directory, _path)
59 self.write(fin, _path) 62 self.write(fin, _path)
60 return self.redirect(self.link('/?uploaded=' + fin.filename)) 63 return self.redirect(self.link('/?uploaded=' + fin.filename))
61 64