修复pre hook的引入导致旧配置无法使用的兼容问题

SohuTech · May 16, 2018 · caa59c9 · caa59c9
1 parent a5a8df9
commit caa59c9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/essay/tasks/deploy.py b/essay/tasks/deploy.py
@@ -24,8 +24,8 @@ def deploy(version, venv_dir, profile):
 
     virtualenv.ensure(venv_dir)
 
-    pre_hook = env.DEPLOY_PRE_HOOK
-    post_hook = env.DEPLOY_POST_HOOK
+    pre_hook = getattr(env, 'DEPLOY_PRE_HOOK', None)
+    post_hook = getattr(env, 'DEPLOY_POST_HOOK', None)
 
     with virtualenv.activate(venv_dir):
         if callable(pre_hook):