File size: 2,606 Bytes
5fa5566 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import time
from pathlib import Path
from contextlib import contextmanager
@contextmanager
def time_task(message_start=None, end=' ', message="⏱ Took"):
if message_start:
print(message_start, end=end, flush=True)
start_time = time.time()
yield
end_time = time.time()
elapsed_time = end_time - start_time
formatted_elapsed_time = format_time(elapsed_time)
print(f"{message} {formatted_elapsed_time}", flush=True)
def format_time(elapsed_time):
hours, rem = divmod(elapsed_time, 3600)
minutes, seconds = divmod(rem, 60)
# Check and format the non-zero time units
parts = []
if hours:
parts.append(f"{int(hours)}h")
if minutes:
parts.append(f"{int(minutes)}m")
if seconds or not parts:
# Add seconds if it's the only non-zero unit or if all units are zero
parts.append(f"{int(seconds)}s")
return ' '.join(parts)
def time_func(func):
def wrapper(*args, **kwargs):
start_time = time.time()
result = func(*args, **kwargs)
end_time = time.time()
elapsed_time = end_time - start_time
formatted_elapsed_time = format_time(elapsed_time)
print(f"Execution time of {func.__name__}: {formatted_elapsed_time}")
return result
return wrapper
def check_other_extensions(file_path, extensions_to_check):
"""
Check the existence of files with the same name but different extensions
in the same folder.
Parameters:
- file_path (str): The path of the file to check.
- extensions_to_check (list): List of extensions to check.
Returns:
- list: List of existing file paths with different extensions.
"""
file_path = Path(file_path)
folder = file_path.parent
base_name = file_path.stem
matching_files = [
folder / (base_name + ext)
for ext in extensions_to_check
if (folder / (base_name + ext)).exists()
]
return matching_files
video_extensions = {".mp4", ".webm", ".mkv", ".avi", ".mov", ".wmv", ".flv", ".vob", ".mts", ".m2ts", ".ts", ".yuv", ".mpg", ".mp2", ".mpeg", ".mpe", ".mpv", ".m2v", ".m4v", ".3gp", ".3g2", ".nsv", ".mts"}
audio_extensions = {".aa", ".aac", ".aax", ".act", ".aiff", ".alac", ".amr", ".ape", ".au", ".awb", ".dss", ".dvf", ".flac", ".gsm", ".iklax", ".ivs", ".m4a", ".m4b", ".m4p", ".mpga", ".mmf", ".mp3", ".mpc", ".msv", ".nmf", ".ogg", ".oga", ".mogg", ".opus", ".ra", ".rm", ".raw", ".rf64", ".sln", ".tta", ".voc", ".vox", ".wav", ".wma", ".wv", ".webm", ".8svx"} |