adding timer

Author: samuelcolvin

devtools/__init__.py

@@ -2,4 +2,5 @@
 from .ansi import *
 from .debug import *
 from .prettier import *
+from .timer import *
 from .version import VERSION

devtools/ansi.py

@@ -6,7 +6,7 @@
 _ansi_template = '\033[{}m'
 _ansi_re = re.compile('\033\[((?:\d|;)*)([a-zA-Z])')
 
-__all__ = ['sformat', 'sprint']
+__all__ = 'sformat', 'sprint'
 
 
 def isatty(stream=None):

devtools/debug.py

@@ -10,8 +10,9 @@
 
 from .ansi import isatty, sformat
 from .prettier import PrettyFormat, env_true
+from .timer import Timer
 
-__all__ = ['Debug', 'debug']
+__all__ = 'Debug', 'debug'
 CWD = Path('.').resolve()
 
 
@@ -119,6 +120,9 @@ def format(self, *args, **kwargs) -> DebugOutput:
     def breakpoint(self):
         pdb.Pdb(skip=['devtools.*']).set_trace()
 
+    def timer(self, name=None, *, verbose=True, file=None, dp=3) -> Timer:
+        return Timer(name=name, verbose=verbose, file=file, dp=dp)
+
     def _process(self, args, kwargs, func_regex) -> DebugOutput:
         curframe = inspect.currentframe()
         try:

devtools/prettier.py

@@ -15,12 +15,13 @@
 else:
     pyg_lexer, pyg_formatter = PythonLexer(), Terminal256Formatter(style='vim')
 
+__all__ = 'PrettyFormat', 'pformat', 'pprint'
+
 PARENTHESES_LOOKUP = [
     (list, '[', ']'),
     (set, '{', '}'),
     (frozenset, 'frozenset({', '})'),
 ]
-__all__ = ['PrettyFormat', 'pformat', 'pprint']
 DEFAULT_WIDTH = int(os.getenv('PY_DEVTOOLS_WIDTH', 120))