| 123456789101112131415161718192021222324252627282930313233343536373839 |
- from functools import wraps
- from time import time
- def is_video(ext: str):
- """
- Returns true if ext exists in
- allowed_exts for video files.
- Args:
- ext:
- Returns:
- """
- allowed_exts = ('.mp4', '.webm', '.ogg', '.avi', '.wmv', '.mkv', '.3gp')
- return any((ext.endswith(x) for x in allowed_exts))
- def tik_tok(func):
- """
- keep track of time for each process.
- Args:
- func:
- Returns:
- """
- @wraps(func)
- def _time_it(*args, **kwargs):
- start = time()
- try:
- return func(*args, **kwargs)
- finally:
- end_ = time()
- print("time: {:.03f}s, fps: {:.03f}".format(end_ - start, 1 / (end_ - start)))
- return _time_it
|
