Flask 蓝图将对象传递给另一个文件

debugcn 发表于 Dev

密封条

我正在使用 python selenium 制作一个 web 服务，当它被调用时会执行一些对页面的指令，但问题是使用烧瓶蓝图将“驱动程序”类传递到另一个页面。

主文件

from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.options import Options as chrome_options
from classes.send_message import send_message
from flask import Flask
from flask_restful import Resource, Api
from flask import request

app = Flask(__name__)
api = Api(app)

app.register_blueprint(send_message)

if __name__ == '__main__':

    logging.info('START APP')

    options = webdriver.ChromeOptions() 
    options.add_argument("user-data-dir=C:\\myuserpath........")
    driver = webdriver.Chrome(executable_path="chromedriver.exe", chrome_options=options)

    driver.get("https://example.com")

    app.run(host='192.168.50.1', port='11111',debug=False)

发送消息.py

from flask import Blueprint
#from flask import current_app as app    
send_message = Blueprint('send_message', __name__)

@send_message.route("/SendPostMessage")
def sendMessage():
    side_panel = driver.find_element_by_id('menutop') # error here!!!

错误：

  File "C:\python36\lib\site-packages\flask\app.py", line 1598, in dispatch_request
    return self.view_functions[rule.endpoint](**req.view_args)
  File "C:\rararara\classes\send_message.py", line 8, in sendMessage
    side_panel = driver.find_element_by_id('menutop')
NameError: name 'driver' is not defined

塔马斯·赫格杜斯

您可以使用 app 对象作为上下文：

...
driver.get("https://example.com")
app.driver = driver
app.run(host='192.168.50.1', port='11111',debug=False)

然后

from flask import current_app as app
...
@send_message.route("/SendPostMessage")
def sendMessage():
    side_panel = app.driver.find_element_by_id('menutop')

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。