Как я могу изменить имя изменения файла, когда оно загружено с Flask? - PullRequest
Новая
       28

Как я могу изменить имя изменения файла, когда оно загружено с Flask?

0 голосов
/ 17 февраля 2020

Я пытаюсь создать веб-приложение для своей команды на работе, и мы хотим, чтобы оно брало загруженные пользователем фотографии и переименовывало их по мере их загрузки. На данный момент у меня есть каталог на моей машине, где находятся файлы go. Мое приложение принимает 3 ввода: имя, адрес электронной почты и файл для загрузки. Когда пользователь нажимает кнопку «Загрузить», я хочу, чтобы файл отображался в моем каталоге в качестве введенного им адреса электронной почты. Например: 

Name: trent
Email: trent@trent.com
File: photo_of_trent.jpg

Я бы хотел, чтобы загруженный файл отображался в моем каталоге как trent@trent.com.jpg.

Мой код Python / Flask выглядит следующим образом: 

import os
from flask import Flask, render_template
from flask_uploads import UploadSet, configure_uploads, IMAGES, patch_request_class
from flask_wtf import FlaskForm
from flask_wtf.file import FileField, FileRequired, FileAllowed
from wtforms import StringField
from wtforms import SubmitField
from wtforms.validators import DataRequired, Email

basedir = os.path.abspath(os.path.dirname(__file__))

app = Flask(__name__)
app.config['SECRET_KEY'] = 'I have a dream'
app.config['UPLOADED_PHOTOS_DEST'] = os.path.join(basedir, 'uploads') # you'll need to create a folder named uploads

photos = UploadSet('photos', IMAGES)
configure_uploads(app, photos)
patch_request_class(app)  # set maximum file size, default is 16MB


class UploadForm(FlaskForm):
    name = StringField('Name', [DataRequired()])
    email = StringField('Email', [Email(message='Not a valid email address.'), DataRequired()])
    photo = FileField(validators=[FileAllowed(photos, 'Image only!'), FileRequired('File was empty!')])
    submit = SubmitField('Upload')


@app.route('/', methods=['GET', 'POST'])
def upload_file():
    form = UploadForm()
    if form.validate_on_submit():
        filename = photos.save(form.photo.data)
        file_url = photos.url(filename)
    else:
        file_url = None
    return render_template('index.html', form=form, file_url=file_url)


if __name__ == '__main__':
    app.run()

и мой html выглядит следующим образом: 

{% extends 'layout.html' %}

{% block styles %}
    <link rel="stylesheet" href="{{ url_for('static', filename='css/forms.css') }}" rel="stylesheet" type="text/css">
{% endblock %}


{% block content %}
<div class="formwrapper">
  <h2 class="title">Contact</h2>
  <form method="POST" action="/" enctype="multipart/form-data">
      <div class="form-field">{{ form.name.label }} {{ form.name(size=20) }}
      {% if form.name.errors %}
        <ul class="errors">
          {% for error in form.name.errors %}
            <li>{{ error }}</li>
          {% endfor %}
        </ul>
      {% endif %}
      </div>
      <div class="form-field">{{ form.email.label }} {{ form.email }}
      {% if form.email.errors %}
        <ul class="errors">
            {% for error in form.email.errors %}
                <li>{{ error }}</li>
            {% endfor %}
        </ul>
    {% endif %}
      </div>
       {{ form.hidden_tag() }}
     {{ form.photo }}
      {{ form.submit }}

  </form>
</div>
{% endblock %}

Это не супер комплекс, и я вроде как исказил этот код вместе, так как это мой первый существенный опыт работы с Flask. Там, где дело касается html, есть немного дзиндзя. Пока программа загружает только файл с именем файла. Любая помощь приветствуется. Спасибо.

1 Ответ

1 голос
/ 18 февраля 2020

Функция UploadSet.save() принимает параметр name, который должен дать вам то, что вы хотите.

например, 

photos.save(form.photo.data, name=f"{form.email.data}.jpg")

Примечание. Я не уверен, как будет обрабатываться расширение файла в этом случае, поэтому вам может потребоваться проверить / настроить это.

...