Question

我的用户在SQLAlchemy中建模为：

class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    url_pic = Column(String(50), nullable=False)
    (...)

我想在Flask-Admin中将用户添加到数据库中，这样当我创建用户时，我可以直接上传照片，并且解析目标URL并将其传递给数据库中的url_pic字段。

我已经可以添加用户并上传照片了（在https://flask-admin.readthedocs.org/en/latest/quickstart/处进行了解释），但在同一视图中找不到有关如何合并添加用户和照片上传的任何信息。

有任何线索吗？

Answer 1

您可以将模型修改为如下所示：

class User(Base):
   __tablename__ = 'users'
   id = Column(Integer, primary_key=True)
   url_pic = Column(String(50), nullable=False)
   pic = Column(LargeBinary, nullable=False)
   ...

现在你应该从ModelView继承flask.ext.admin.contrib.sqla。然后，将该子类的实例添加到Admin实例。以下源代码可以帮助您明白这一点。

from flask.ext.admin.contrib.sqla import ModelView
from flask.ext.admin.form.upload import FileUploadField
from wtforms.validators import ValidationError
from flask.ext.admin import Admin
from flask.ext.sqlalchemy import SQLAlchemy
from flask import Flask
import imghdr

app = Flask(__name__)
db = SQLAlchemy(app)

class UserAdminView(ModelView):

   def picture_validation(form, field):
      if field.data:
         filename = field.data.filename
         if filename[-4:] != '.jpg': 
            raise ValidationError('file must be .jpg')
         if imghdr.what(field.data) != 'jpeg':
            raise ValidationError('file must be a valid jpeg image.')
      field.data = field.data.stream.read()
      return True

   form_columns = ['id','url_pic', 'pic']
   column_labels = dict(id='ID', url_pic="Picture's URL", pic='Picture')

   def pic_formatter(view, context, model, name):
       return 'NULL' if len(getattr(model, name)) == 0 else 'a picture'

   column_formatters =  dict(pic=pic_formatter)
   form_overrides = dict(pic= FileUploadField)
   form_args = dict(pic=dict(validators=[picture_validation]))

admin = Admin(app)
admin.add_view(UserAdminView(User, db.session, category='Database Administration'))
...

您可以在此处找到ModelView的文档：link here 我希望这可以帮助别人！

Flask-Admin自动上传并插入数据库

1 个答案: