From 27555eb6877454630a1be4334a57cd5431c499f5 Mon Sep 17 00:00:00 2001 From: J Kishore Kumar Date: Thu, 26 Feb 2015 12:13:15 +0530 Subject: [PATCH] Fix running all sniffs in contrib when atleast one is imported --- fabpolish/__init__.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/fabpolish/__init__.py b/fabpolish/__init__.py index fd423fd..ac8f216 100644 --- a/fabpolish/__init__.py +++ b/fabpolish/__init__.py @@ -3,6 +3,7 @@ from functools import wraps +from fabric.main import list_commands from fabric.api import lcd, local, settings, task, puts, hide from fabric.colors import green @@ -10,7 +11,7 @@ FABFILE_DIR = os.path.abspath(os.path.dirname(fabfile.__file__)) -__version__ = '1.0.0' +__version__ = '1.0.1' def info(text): @@ -77,13 +78,20 @@ def polish(env='dev'): When environment is 'dev', only fast-critical and fast-major sniffs are run. """ + fabric_tasks = list_commands('', 'short') results = list() with settings(warn_only=True): if env == 'ci': - sniffs_to_run = _sniffs + sniffs_to_run = [] + for sniff in _sniffs: + if sniff['function'].name not in fabric_tasks: + continue + sniffs_to_run.append(sniff) elif env == 'dev': sniffs_to_run = [] for sniff in _sniffs: + if sniff['function'].name not in fabric_tasks: + continue if sniff['timing'] != 'fast': continue if sniff['severity'] not in ('critical', 'major'):