Flask代码去重最佳实践：如何消除环境检测逻辑冗余？

首先先修正原代码里的两个基础错误：

• 导入语句错误：import flask import Flask 应改为 from flask import Flask
• 缺少os模块导入，需要添加import os

针对代码中重复的on_heroku判断和环境分支逻辑，提供两种实用的改进方案：

方案一：抽取环境检测函数 #

把判断是否运行在Heroku的逻辑抽成独立的可复用函数，避免重复编写相同的环境检测代码，后续修改判断规则时只需维护这一个函数。

改进后的代码示例：

from flask import Flask
import os

# 抽取环境检测逻辑为独立函数
def is_running_on_heroku():
    return 'DYNO' in os.environ

app = Flask(__name__)

@app.route('/')
def index():
    # Some code...
    on_heroku = is_running_on_heroku()
    if on_heroku:
        # 执行Heroku环境下的逻辑
        pass
    else:
        # 执行本地环境下的逻辑
        pass
    return ("SUCCESS", 200)

def open_browser():
    on_heroku = is_running_on_heroku()
    if on_heroku:
        # Heroku环境下无需打开浏览器，执行对应逻辑
        pass
    else:
        # 本地环境下打开浏览器
        import webbrowser
        webbrowser.open('http://localhost:5000')

if __name__ == "__main__":
    open_browser()
    app.run(debug=True)

方案二：封装环境分支逻辑 #

如果两处的环境分支逻辑可以统一处理，可将环境判断和逻辑执行封装成一个通用函数，通过传入不同的回调函数来处理不同场景的环境逻辑，彻底消除重复的if-else结构。

改进后的代码示例：

from flask import Flask
import os

# 封装环境判断与逻辑执行
def handle_environment(heroku_action, local_action):
    on_heroku = 'DYNO' in os.environ
    heroku_action() if on_heroku else local_action()

app = Flask(__name__)

@app.route('/')
def index():
    # Some code...
    # 定义当前场景的Heroku环境逻辑
    def heroku_logic():
        pass
    # 定义当前场景的本地环境逻辑
    def local_logic():
        pass
    handle_environment(heroku_logic, local_logic)
    return ("SUCCESS", 200)

def open_browser():
    # 浏览器场景的Heroku逻辑（空操作）
    def heroku_browser_op():
        pass
    # 浏览器场景的本地逻辑
    def local_browser_op():
        import webbrowser
        webbrowser.open('http://localhost:5000')
    handle_environment(heroku_browser_op, local_browser_op)

if __name__ == "__main__":
    open_browser()
    app.run(debug=True)

补充说明 #

不推荐将on_heroku设为全局变量，因为全局变量会降低代码的可测试性和灵活性，后续扩展多环境时维护成本更高。上述两种方案都遵循了DRY（Don't Repeat Yourself）原则，让代码更简洁、易维护。

内容的提问来源于stack exchange，提问作者livemyaerodream