Python 如何基于heroku部署在django中运行不同的代码

我正在尝试设置python/django项目。我在win7本地工作，希望推送到heroku。理想情况下，我会在虚拟化环境中工作，以匹配本地和部署环境，但有时这是不可能的。最有效的运行方式是什么

if (in heroku):
    do this
else:
   do that

---

定义一个环境变量。在Heroku的例子中，它看起来像：

heroku config:set WORKING_ENVIRONMENT=HEROKU

然后在代码中检查此变量：

if os.environ['WORKING_ENVIRONMENT'] == 'HEROKU':
    pass
else:
    pass