python实现文件上传功能_python实现文件上传功能

如文件小可以采用这个方式import osdef form(): return

"""\

enctype="multipart/form-data" action="./upload"

method="post">

File:

name="file">

type="submit"

value="Upload">

"""def

upload(req): try: # Windows needs stdio set for binary mode. import

msvcrt msvcrt.setmode (0, os.O_BINARY) # stdin = 0 msvcrt.setmode

(1, os.O_BINARY) # stdout = 1 except ImportError: pass # A nested

FieldStorage instance holds the file fileitem = req.form['file'] #

Test if the file was uploaded if fileitem.filename: # strip leading

path from file name to avoid directory traversal attacks fname =

os.path.basename(fileitem.filename) # build absolute path to files

directory dir_path = os.path.join(os.path.dirname(req.filename),

'files') open(os.path.join(dir_path, fname),

'wb').write(fileitem.file.read()) message = 'The file "%s" was

uploaded successfully' % fname else: message = 'No file was

uploaded' return

"""\

%s

href="./form">Upload another

file

"""

% message如文件大时,可以分块上传,现在是改良方式import osdef form(): return

"""\

enctype="multipart/form-data" action="./upload"

method="post">

File:

name="file">

type="submit"

value="Upload">

"""#

Generator to buffer file chunksdef fbuffer(f, chunk_size=10000):

while True: chunk = f.read(chunk_size) if not chunk: break yield

chunkdef upload(req): try: # Windows needs stdio set for binary

mode. import msvcrt msvcrt.setmode (0, os.O_BINARY) # stdin = 0

msvcrt.setmode (1, os.O_BINARY) # stdout = 1 except ImportError:

pass # A nested FieldStorage instance holds the file fileitem =

req.form['file'] # Test if the file was uploaded if

fileitem.filename: # strip leading path from file name to avoid

directory traversal attacks fname =

os.path.basename(fileitem.filename) # build absolute path to files

directory dir_path = os.path.join(os.path.dirname(req.filename),

'files') f = open(os.path.join(dir_path, fname), 'wb', 10000) #

Read the file in chunks for chunk in fbuffer(fileitem.file):

f.write(chunk) f.close() message = 'The file "%s" was uploaded

successfully' % fname else: message = 'No file was uploaded' return

"""\

%s

href="./form">Upload another

file

"""

% message