1
0
Fork 0
mirror of https://github.com/snovvcrash/usbrip.git synced 2024-05-17 20:56:06 +02:00
usbrip/usbrip/lib/utils/debug.py
2020-01-12 21:27:07 +03:00

57 lines
1.4 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""LICENSE
Copyright (C) 2020 Sam Freeside
This file is part of usbrip.
usbrip is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
usbrip is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with usbrip. If not, see <http://www.gnu.org/licenses/>.
"""
__author__ = 'Sam Freeside (@snovvcrash)'
__email__ = 'snovvcrash@protonmail[.]ch'
__site__ = 'https://github.com/snovvcrash/usbrip'
__brief__ = 'Debug utils'
import functools
import time
import usbrip.lib.core.config as cfg
def time_it(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
start = time.time()
result = func(*args, **kwargs)
end = time.time()
print(f'{func.__name__}: {end-start:.3f} seconds')
return result
return wrapper
class time_it_if_debug:
def __init__(self, condition, decorator):
self._condition = cfg.DEBUG
self._decorator = decorator
def __call__(self, func):
if not self._condition:
return func
return self._decorator(func)