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

gpe*_*ana 7 python flask flask-sqlalchemy flask-admin

我的用户在SQLAlchemy中建模为:

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

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

我已经可以添加用户和上传照片了(请在https://flask-admin.readthedocs.org/en/latest/quickstart/上解释),但无法找到有关如何合并添加用户和照片上传的任何信息相同的观点.

任何线索?

Gio*_*ova 6

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

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

现在你应该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'))
...
Run Code Online (Sandbox Code Playgroud)

在这里你可以找到以下文档ModelView:链接在这里 我希望这可以帮助某人!

  • `User` 类中的 `pic` 应该是一个带有路径名的字符串,否则如果这成为一个大项目,在数据库中存储大量的 `LargeBinary` 会变得非常昂贵。 (2认同)