Доступ к переменным конфигурации из внешнего файла

Question

Я создаю веб-приложение Flask и реализую функцию, с помощью которой пользователи могут загружать свои собственные изображения. У меня есть три файла, над которыми я работаю для реализации этой функции:

• models.py: файл, в котором я поддерживаю структуру и функции базы данных, которые можно вызывать для объектов в базе данных.

• rout.py: файл, который обрабатывает ответы сайта на HTTP-запросы на различные URL-адреса.

• __ init__.py: где установлены все мои переменные конфигурации.

У меня в models.py есть функция avatar, которая возвращает аватар пользователя. Если они не загрузили фотографию, она создает аватар с использованием хеш-генератора gravatars MD5. Я хочу изменить эту функцию так, чтобы она сначала смотрела в моем файле загрузок, чтобы увидеть, загрузил ли пользователь фотографию.

Я настроил функцию загрузки так, что фотографии пользователей будут сохраняться в папке с форматом. У меня есть переменная конфигурации UPLOAD_FOLDER, в которой хранится абсолютный путь к этой папке. Однако проблема в том, что переменные конфигурации недоступны из файла моделей. Я попытался создать локальный контекст, как описано в docs , но у меня ошибка KeyError: UPLOAD_FOLDER.

models.py

from flask import Flask, current_app
from hashlib import md5
class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key = True)
    username = db.Column(db.String(64), index = True, unique = True)
    email = db.Column(db.String(128), index = True, unique = True, nullable = False)
    password_hash = db.Column(db.String(128), nullable = False)
    is_admin = db.Column(db.Boolean(), default = False)
    polls = db.relationship("Poll", backref = "author", lazy = "dynamic")
    votes = db.relationship("Votes", backref = "voter", lazy = "dynamic")

   def avatar(self, size):
      app = Flask(__name__)
      with app.app_context():
      UPLOAD_FOLDER = current_app.config["UPLOAD_FOLDER"]
      if(UPLOAD_FOLDER):
         print("hooray")
      digest = md5(self.email.lower().encode("utf-8")).hexdigest()
      return(("https://www.gravatar.com/avatar/{}?d=retro&s={}").format(digest, size))

__ __ INIT. Ру

import os 

dirname = os.path.dirname(os.path.abspath(__file__))
UPLOAD_FOLDER = dirname + "/static/user-images/"
ALLOWED_FILES = ["png", "jpeg", "jpg"]

app = Flask(__name__)
app.config.from_object(Config)
app.config["UPLOAD_FOLDER"] = UPLOAD_FOLDER
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
app.config["ALLOWED_FILES"] = ALLOWED_FILES

routes.py

import os 
from werkzeug.utils import secure_filename
from flask_login import current_user

def allowed_file(file):
    return(file.filename.split(".")[1].lower() in app.config["ALLOWED_FILES"])

@app.route("/upload", methods = ["POST"])
@login_required
def upload_file():
    if(request.method == "POST"):
        if("file" not in request.files):
            print(request.files)
            flash("No file part")
            return(redirect("upload"))
        file = request.files["file"]
        if(not file.filename):
            flash("No file uploaded!")
            return(redirect("upload"))
        if(file and allowed_file(file)):
            extension = file.filename.split(".")[1]
            filename = secure_filename(str(current_user.id) + "." + extension)
            file.save(os.path.join(app.config["UPLOAD_FOLDER"], filename))
            flash("Files successfully uploaded")
            return(redirect(url_for("user", username = current_user.username)))

Answer 1

Вместо того, чтобы помещать переменные конфигурации в контекст приложения, почему бы вам просто не поместить их в файл конфигурации и импортировать их, где это необходимо?

Создайте файл с именем config.py

config.py

import os

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

UPLOAD_FOLDER = dirname + "/static/user-images/"
ALLOWED_FILES = ["png", "jpeg", "jpg"]
MAX_CONTENT_LENGTH = 16 * 1024 * 1024

Затем в rout.py импортируйте эти переменные конфигурации и используйте их, замените app.config["UPLOAD_FOLDER"] на UPLOAD_FOLDER

routes.py

import os 
from werkzeug.utils import secure_filename
from flask_login import current_user
from config import UPLOAD_FOLDER

def allowed_file(file):
    return(file.filename.split(".")[1].lower() in app.config["ALLOWED_FILES"])

@app.route("/upload", methods = ["POST"])
@login_required
def upload_file():
    if(request.method == "POST"):
        if("file" not in request.files):
            print(request.files)
            flash("No file part")
            return(redirect("upload"))
        file = request.files["file"]
        if(not file.filename):
            flash("No file uploaded!")
            return(redirect("upload"))
        if(file and allowed_file(file)):
            extension = file.filename.split(".")[1]
            filename = secure_filename(str(current_user.id) + "." + extension)
            file.save(os.path.join(UPLOAD_FOLDER, filename))
            flash("Files successfully uploaded")
            return(redirect(url_for("user", username = current_user.username)))

Сделайте то же самое в models.py, замените current_app.config["UPLOAD_FOLDER"] на UPLOAD_FOLDER.

models.py

from flask import Flask, current_app
from hashlib import md5
from config import UPLOAD_FOLDER
class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key = True)
    username = db.Column(db.String(64), index = True, unique = True)
    email = db.Column(db.String(128), index = True, unique = True, nullable = False)
    password_hash = db.Column(db.String(128), nullable = False)
    is_admin = db.Column(db.Boolean(), default = False)
    polls = db.relationship("Poll", backref = "author", lazy = "dynamic")
    votes = db.relationship("Votes", backref = "voter", lazy = "dynamic")

   def avatar(self, size):
      app = Flask(__name__)
      with app.app_context():
          upload_folder = UPLOAD_FOLDER  # please don't make normal variables UPPERCASE
          if(upload_folder):
             print("hooray")
          digest = md5(self.email.lower().encode("utf-8")).hexdigest()
          return(("https://www.gravatar.com/avatar/{}?d=retro&s={}").format(digest, size))