2

我正在尝试在 SQLAlchemy 中创建一个自定义字段,以管理 Flask 应用程序中上传的文件。我基本上是在尝试实现类似于 Django 的 FileField 的东西。

所以我想做的是,给定一个模型:

class TestModel(Base):
    __tablename__ = "testmodel"
    id = Column(Integer(), primary_key=True)
    profile_photo = Column(FileField(upload_set, 100))

我希望能够做类似的事情:

m = TestModel()
m.profile_photo.save(request.files["profile_photo"])
models.db.session.add(m)
models.db.session.commit()

我所拥有的是基于内置 String 类型的,并且像这样工作:

class FileField(types.TypeDecorator):
    impl = String

    def __init__(self, upload_set, *args, **kwargs):
        super(FileField, self).__init__(*args, **kwargs)
        self.upload_set = upload_set

    def process_bind_param(self, value, dialect):
        if value is not None:
            return value.filename
        return value

    def process_result_value(self, value, dialect):
        return FileNameString(self.upload_set, value)


class FileNameString(object):

    def __init__(self, upload_set, filename):
        self.upload_set = upload_set
        self.filename = filename

    def save(self, req_file):
        self.filename = self.upload_set.save(req_file)
        return self.filename

    @property
    def url(self):
        return self.upload_set.url(self.filename)

这在模型存在于数据库中时有效,但在模型是从头开始创建的profile_photo属性时无效None。理想情况下,我希望这种类型的默认值是空白FileField,尽管我还没有弄清楚它会是什么样子。问题是我无法弄清楚如何为属性创建默认值,而不必在任何使用FileField.

那么,知道如何为profile_photo属性创建默认值吗?

4

0 回答 0