首页 文章

Flask - 不能使用其他文件中的Flask和Flask-mail实例

提问于
浏览
1

我目前正在使用Flask构建应用程序 . 我正在努力访问Flask实例'app'以及Flask-mail实例'mail' .

以下是我的项目的样子:

└── my-project
    ├── application
    │   ├── __init__.py
    │   ├── admin
    │   │   ├── __init__.py
    │   │   ├── forms.py
    │   │   └── views.py
    │   ├── auth
    │   │   ├── __init__.py
    │   │   ├── forms.py
    │   │   └── views.py
    │   │   └── token.py
    │   │   └── email.py
    │   ├── home
    │   │   ├── __init__.py
    │   │   └── views.py
    │   ├── models.py
    │   ├── static
    │   └── templates
    │       └──....
    │
    ├── config.py
    ├── instance
    │   └── config.py
    ├── migrations
    │   ├── README
    │   ├── alembic.ini
    │   ├── env.py
    │   ├── script.py.mako
    │   └── versions
    │       └── a1a1d8b30202_.py
    ├── requirements.txt
    └── run.py

Flask实例是在run.py中使用create_app函数创建的(来自

run.py

import os

from application import create_app

config_name = os.getenv('FLASK_CONFIG')
app = create_app(config_name)

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

application / __ init__.py

# third-party imports
from flask import Flask, render_template
from flask_sqlalchemy import SQLAlchemy
from flask_login import LoginManager
from flask_migrate import Migrate
from flask_bootstrap import Bootstrap
from flask_mail import Mail
import stripe


# local imports
from config import app_config

# db variable initialization
db = SQLAlchemy()
login_manager = LoginManager()
LoginManager.user_loader

def create_app(config_name):
    app = Flask(__name__, instance_relative_config=True)
    app.config.from_object(app_config[config_name])
    app.config.from_pyfile('config.py')


    Bootstrap(app)
    db.init_app(app)
    login_manager.init_app(app)
    mail = Mail(app)
    migrate = Migrate(app,db)

    from application import models

    from .admin import admin as admin_blueprint
    app.register_blueprint(admin_blueprint, url_prefix='/admin')
    #the rest of the blueprint import goes here


    return app

我想要做的是使用'app'和'mail' . 例如,在application / auth / token中:

application / auth / token.py

from itsdangerous import URLSafeTimedSerializer

from . import auth

def generate_confirmation_token(email):
    serializer = URLSafeTimedSerializer(app.config['SECRET_KEY'])
    return serializer.dumps(email, salt=app.config['SECURITY_PASSWORD_SALT'])

def confirm_token(token, expiration = 600):
    serializer = URLSafeTimedSerializer(app.config['SECRET_KEY'])
    try:
        email = serializer.loads(
            token,
            salt=app.config['SECURITY_PASSWORD_SALT'],
            max_age=expiration
        )
    except:
        return False
    return email

或者在application / auth / email.py中:

application / auth / email.py

from flask_mail import Message

from . import auth

def send_mail(to, subject, template):
    msg = Message(
        subject,
        recipients=[to],
        html=template,
        sender=app.config['MAIL_DEFAULT_SENDER']
    )
    mail.send(msg)

我在application / aut / views.py中需要这两个函数

application / auth / views.py

from flask import flash, redirect, render_template, url_for, request
from flask_login import login_required, login_user, logout_user
from werkzeug.security import check_password_hash
import datetime

from . import auth
from forms import LoginForm, RegistrationForm
from .. import db
from ..models import User

@auth.route('/register', methods=['GET', 'POST'])
def register():
    """
    Handle requests to the /register route
    Add a user to the database through the registration form
    """
    form = RegistrationForm()
    form.id = 'form_signup' 
    if form.validate_on_submit():
        user = User(email=form.email.data,
                    #username=form.username.data,
                    first_name=form.first_name.data,
                    last_name=form.last_name.data,
                    password=form.password.data,
                    registered_on=datetime.datetime.now(),
                    confirmed=False,
                    premium=False)

        # add employee to the database
        db.session.add(user)
        db.session.commit()
        flash("We've just sent you an email confirmation. Please activate you account to completly finish your registration", 'succes')

        token = generate_confirmation_token(user.email)
        confirm_url = url_for('auth.confirm_email', token=token, _external=True)
        html = render_template('auth/activate.html', confirm_url=confirm_url)
        subject = "Please confirm your email"
        send_email(user.email, subject, html)

        login_user(user)

        flash('A confirmation email has been sent via email.', 'success')

        # redirect to the login page
        #return redirect(url_for('auth.login'))
        return redirect(url_for('home.homepage'))

    # load registration template
    return render_template('auth/register.html', form=form, title='Register')

@auth.route('/confirm/<token>')
@login_required
def confirm_email(token):
    try:
        email = confirm_token(token)
    except:
        flash('The confirmation link is invalid or has expired.', 'danger')
    user = User.query.filter_by(email=email).first_or_404()
    if user.confirmed:
        flash('Account already confirmed. Please login.', 'succes')
    else:
        user.confirmed =True
        user.confirmed_on = datetime.datetime.now()
        db.session.add(user)
        db.session.commit()
        flash("You've confirmed your account. Thanks!", 'succes')
    return redirect(url_for('auth.login'))

我得到的是一个错误'未定义全局名称应用'或'未定义全局名称邮件' . 我试图从应用程序导入应用程序导入变量,它返回一个导入错误'无法导入应用程序'

感谢您的支持

2 回答

  • 3

    由于您使用的是应用程序工厂,因此需要在 Flask-Mail 类上使用 .init_app 方法,就像使用 Flask-SQLAlchemy 类一样 . from application import app 将无效,因为在 run.py 中调用 create_app 函数之前,您永远不会初始化烧瓶应用程序的实例

    application/init.py

    from flask_mail import Mail
    
    mail = Mail()
    
    
    def create_app(config_lvl):
        # stuff
    
        mail.init_app(app)
        # more stuff
        return app
    

    您也可以使用 current_app 来引用应用程序实例而不是实例本身,只要您在其中使用的代码块正在烧瓶应用程序中运行 . Here是一个更深入的解释 .

    application/auth/email.py

    from application import mail  # you can now import the Mail() object
    from flask_mail import Message
    from flask import current_app  # use this to reference current application context
    
    def send_email(to, subject, template):
        msg = Message(
            subject,
            recipients=[to],
            html=template,
            sender=current_app.config['MAIL_DEFAULT_SENDER']
        )
        mail.send(msg)
    

    application/auth/token.py

    from itsdangerous import URLSafeTimedSerializer
    from flask import current_app
    
    def generate_confirmation_token(email):
        serializer = URLSafeTimedSerializer(current_app.config['SECRET_KEY'])
        return serializer.dumps(email, salt=current_app.config['SECURITY_PASSWORD_SALT'])
    
    def confirm_token(token, expiration = 600):
        serializer = URLSafeTimedSerializer(current_app.config['SECRET_KEY'])
        try:
            email = serializer.loads(
                token,
                salt=current_app.config['SECURITY_PASSWORD_SALT'],
                max_age=expiration
            )
        except:
            return False
        return email
    

    还应注意,除了 views.py 之外, auth 蓝图下的任何模块都不需要 from . import auth

    EDIT

    附注:您不必将用户添加到会话中,因为它是在路由中先前查询时添加的 . 我自己也没有意识到这一点 .

    @auth.route('/confirm/<token>')
    @login_required
    def confirm_email(token):
        try:
            email = confirm_token(token)
        except:
            flash('The confirmation link is invalid or has expired.', 'danger')
        user = User.query.filter_by(email=email).first_or_404()
        if user.confirmed:
            flash('Account already confirmed. Please login.', 'success')
        else:
            user.confirmed = True
            user.confirmed_on = datetime.datetime.now()
          #  db.session.add(user) # can remove this
            db.session.commit()
            flash("You've confirmed your account. Thanks!", 'success')
        return redirect(url_for('auth.login'))
    
  • 0

    根据http://flask.pocoo.org/docs/0.12/patterns/appfactories/,您必须使用current_app(来自flask import current_app)来访问您的应用程序中的Flask实例 . 然后你可以使用你的配置变量:例如 current_app.config['DEBUG'] .

    使用应用程序工厂时要考虑的另一点是,在'create_app()'函数之外创建实例类实例 . 然后你必须使用.init_app方法:

    以下是一个例子:

    from flask_mail import Mail
    
    mail= Mail()
    
    def create_app():
        mail.init_app(app)
    

相关问题