Question

我想用SQL.factory（）上传文件我只想保留原始文件名我的代码目前是

form = SQLFORM.factory(
    Field('file_name', requires=IS_NOT_EMPTY()),
    Field('file', 'upload',uploadfolder=upload_folder))
if form.accepts(request.vars, session):  #.process().accepted:
    response.flash = u'File uploaded'
    session.your_name = form.vars.file_name
    session.filename = request.vars.file 
elif form.errors:
    response.flash = 'form has errors'
return dict(form=form)

我猜session.filename = request.vars.file是你设置文件名的地方。为什么我得到自动生成的文件名no_data.smth.23u8o8274823zu4i2.smth

谢谢

Answer 1

首先，request.vars.file是一个Python cgi.FieldStorage对象，因此session.filename = request.vars.file会导致错误。 request.vars.file.file是实际的文件对象，request.vars.file.filename是上传文件的原始名称。

通过上传字段上传文件时，web2py会自动生成“table_name.field_name.random_id.b16encoded_original_filename.extension”形式的新名称。这样做是为了防止目录遍历攻击并启用下载机制（需要知道表和字段名称）。对于SQLFORM.factory，没有数据库表名，因此默认为表名为“no_table”。

您显示的代码实际上不应生成像'no_data.smth.23u8o8274823zu4i2.smth'这样的文件名。该文件名意味着您已明确告知SQLFORM.factory使用表名“no_data”（通过其table_name参数）并且上传字段名称为“smth”。（上面的代码会生成一个以'no_table.file'开头的文件名。）

注意，web2py自动获取上传文件的原始名称，并将其编码（使用b16encode）到新文件名中（然后在使用内置下载机制时解码原始文件名）。 form.vars.file.filename中也提供了原始文件名。因此，您根本不需要用户输入文件名。但是，如果要使用户能够输入可能与实际文件名不同的文件名，然后使用用户输入的文件名，则可以在创建表单之前添加以下内容：

if 'file' in request.vars and request.vars.file_name:
    request.vars.file.filename = request.vars.file_name

这会将上传文件的文件名重新分配给用户输入的值，然后web2py会将用户输入的文件名编码为新文件名。但请注意，web2py依赖于文件扩展名来在下载时适当地设置HTTP标头，因此您可能需要添加一些逻辑来获取原始文件扩展名，以防用户输入错误。

Answer 2

如果您只是重命名文件，这将破坏下载机制。此外，有时您可能希望使用与原始名称不同的名称保存文件。我们假设您有以下模型：

db.define_table("files",
Field("name", unique=True),
Field("file", "upload"))

您需要使用自定义存储和检索功能扩展上传字段：

Field("file", "upload", custom_store=store_file, custom_retrieve=retrieve_file)

这些功能只是从固定的上传目录中写入/读取文件：

import os
import shutil

def store_file(file, filename=None, path=None):
    path = "applications/app_name/uploads"
    if not os.path.exists(path):
         os.makedirs(path)
    pathfilename = os.path.join(path, filename)
    dest_file = open(pathfilename, 'wb')
    try:
            shutil.copyfileobj(file, dest_file)
    finally:
            dest_file.close()
    return filename

def retrieve_file(filename, path=None):
    path = "applications/app_name/uploads"
    return (filename, open(os.path.join(path, filename), 'rb'))

现在在控制器中，您需要在完成数据库插入/更新之前修改form.vars并设置文件名。如果您想保留上传文件的原始名称，则无需这样做。

def validate(form):
    # set the uploaded file name equal to a name given in the form
    if form.vars.file is not None:
        form.vars.file.filename = form.vars.name

你还需要定义一个函数来下载文件，因为build in response.download不会工作：

import contenttype as c

def download():
    if not request.args:
        raise HTTP(404)
    name = request.args[-1]
    field = db["files"]["file"]
    try:
        (filename, file) = field.retrieve(name)
    except IOError:
        raise HTTP(404)
    response.headers["Content-Type"] = c.contenttype(name)
    response.headers["Content-Disposition"] = "attachment; filename=%s" % name
    stream = response.stream(file, chunk_size=64*1024, request=request)
    raise HTTP(200, stream, **response.headers)

要连接点，您需要构建表单。在下面的示例中，我使用了新的网格机制，这种机制比旧的学校形式更好（但尚未在书中记录）。

upload = lambda filename: URL("download", args=[filename])

def index():
    grid = SQLFORM.grid(db.files, onvalidation=validate, upload=upload)
    return {"grid":grid}

如果您不想要网格的所有功能，则等效的控制器代码为：

def index():
    if len(request.args):
        form=SQLFORM(db.files, request.args[0], upload=URL("download"))
    else:
        form=SQLFORM(db.files, upload=URL("download"))

    if form.process(onvalidation=validate).accepted:
        response.flash = "files updated"

    return {"form":form}

Answer 3

所以我做到了:) 这是我的代码

import os
upload_folder ='C:\\Python27\\web2py'
sl = "\\"
path = upload_folder + sl

def display_form():

     form = SQLFORM.factory(
        Field('file_name', requires=IS_NOT_EMPTY()),
        Field('file', 'upload',uploadfolder=upload_folder))


     if form.accepts(request.vars, session):  #.process().accepted:
        session.file_name= form.vars.file_name
        coded_name = form.vars.file 
        orig_name = request.vars.file.filename
        os.rename(path + coded_name, path + orig_name)
        response.flash = u'datoteka naložena'

    elif form.errors:
        response.flash = 'form has errors'
    return dict(form=form)

我知道它可能不是最好的解决方案，但由于它有效，我喜欢：）

谢谢你，安东尼

web2py上传原始文件名

3 个答案: