我正在尝试在 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
属性创建默认值吗?