Skip to content

Commit

Permalink
Merge pull request frappe#1006 from gavindsouza/cleanup-logger
Browse files Browse the repository at this point in the history
fix: exit with appropriate return code
  • Loading branch information
gavindsouza authored May 21, 2020
2 parents 89dd972 + 0ba5a09 commit b8d3dc3
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 18 deletions.
1 change: 1 addition & 0 deletions bench/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -127,6 +127,7 @@ def get_app(git_url, branch=None, bench_path='.', skip_assets=False, verbose=Fal
install_app(app=app_name, bench_path=bench_path, verbose=verbose, skip_assets=skip_assets)
sys.exit()

print('\n{0}Getting {1}{2}'.format(color.yellow, repo_name, color.nc))
logger.log('Getting app {0}'.format(repo_name))
exec_cmd("git clone {git_url} {branch} {shallow_clone} --origin upstream".format(
git_url=git_url,
Expand Down
6 changes: 4 additions & 2 deletions bench/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,8 +59,10 @@ def cli():
try:
bench_command()
except BaseException as e:
logger.warn("{0} executed with exit code {1}".format(command, getattr(e, "code", None)))
sys.exit(1)
return_code = getattr(e, "code", 0)
if return_code:
logger.warning("{0} executed with exit code {1}".format(command, return_code))
sys.exit(return_code)


def check_uid():
Expand Down
19 changes: 3 additions & 16 deletions bench/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -302,7 +302,7 @@ def exec_cmd(cmd, cwd='.'):
cmd = shlex.split(cmd)
return_code = subprocess.call(cmd, cwd=cwd, universal_newlines=True)
if return_code:
logger.warn("{0} executed with exit code {1}".format(cmd_log, return_code))
logger.warning("{0} executed with exit code {1}".format(cmd_log, return_code))


def which(executable, raise_err = False):
Expand Down Expand Up @@ -435,26 +435,13 @@ def logv(self, message, *args, **kws):
self._log(LOG_LEVEL, message, args, **kws)
logging.Logger.log = logv

class log_filter(object):
def __init__(self, level):
self.__level = level

def filter(self, logRecord):
return logRecord.levelno == self.__level

if os.path.exists(os.path.join(bench_path, 'logs')):
logger = logging.getLogger(bench.PROJECT_NAME)
log_file = os.path.join(bench_path, 'logs', 'bench.log')
formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
hdlr = logging.FileHandler(log_file)
hdlr.setFormatter(formatter)

log_hndlr = logging.StreamHandler(sys.stdout)
log_hndlr.setFormatter(logging.Formatter('%(message)s'))
log_hndlr.addFilter(log_filter(LOG_LEVEL))

logger.addHandler(hdlr)
logger.addHandler(log_hndlr)
logger.setLevel(logging.DEBUG)

return logger
Expand Down Expand Up @@ -985,7 +972,7 @@ def migrate_env(python, backup=False):
logger.log('Clearing Redis DataBase...')
exec_cmd('{redis} FLUSHDB'.format(redis = redis))
except:
logger.warn('Please ensure Redis Connections are running or Daemonized.')
logger.warning('Please ensure Redis Connections are running or Daemonized.')

# Backup venv: restore using `virtualenv --relocatable` if needed
if backup:
Expand Down Expand Up @@ -1015,7 +1002,7 @@ def migrate_env(python, backup=False):
logger.log('Migration Successful to {}'.format(python))
except:
if venv_creation or packages_setup:
logger.warn('Migration Error')
logger.warning('Migration Error')


def is_dist_editable(dist):
Expand Down

0 comments on commit b8d3dc3

Please sign in to comment.