MirroredRepos/thefuck
1
0
Fork 0
mirror of https://github.com/nvbn/thefuck.git synced 2024-10-05 18:31:10 +01:00
Code Issues Releases Wiki Activity

#682: Measure time spent on log reading

Browse Source
This commit is contained in:
Vladimir Iakovlev 2017-09-09 06:07:08 +02:00
parent 4875d75a64
commit 22aa9b701d
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
thefuck/output_readers/read_log.py
View File

@ -7,8 +7,7 @@ except ImportError:
import six
import pyte
from ..exceptions import ScriptNotInLog
from ..logs import warn, debug
from .. import const
from .. import const, logs
def _group_by_calls(log):
@ -68,28 +67,30 @@ def get_output(script):
"""
if six.PY2:
warn('Experimental instant mode is Python 3+ only')
logs.warn('Experimental instant mode is Python 3+ only')
return None
if 'THEFUCK_OUTPUT_LOG' not in os.environ:
warn("Output log isn't specified")
logs.warn("Output log isn't specified")
return None
if const.USER_COMMAND_MARK not in os.environ.get('PS1', ''):
warn("PS1 doesn't contain user command mark, please ensure "
"that PS1 is not changed after The Fuck alias initialization")
logs.warn(
"PS1 doesn't contain user command mark, please ensure "
"that PS1 is not changed after The Fuck alias initialization")
return None
try:
with open(os.environ['THEFUCK_OUTPUT_LOG'], 'rb') as log_file:
with logs.debug_time(u'Read output from log'), \
open(os.environ['THEFUCK_OUTPUT_LOG'], 'rb') as log_file:
_skip_old_lines(log_file)
lines = _get_output_lines(script, log_file)
output = '\n'.join(lines).strip()
debug(u'Received output: {}'.format(output))
logs.debug(u'Received output: {}'.format(output))
return output
except OSError:
warn("Can't read output log")
logs.warn("Can't read output log")
return None
except ScriptNotInLog:
warn("Script not found in output log")
logs.warn("Script not found in output log")
return None