File size: 2,968 Bytes
0191eab |
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 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 |
import datetime
import os
from pathlib import Path
import random
import sys
import ffmpeg
from rich.console import Console
import msg
from src.logger import setup_logger
console = Console()
logger = setup_logger()
class KeepDir:
def __init__(self):
self.original_dir = os.getcwd()
def __enter__(self):
return self
def __exit__(self, exc_type, exc_val, exc_tb):
os.chdir(self.original_dir)
def chdir(self, path):
os.chdir(path)
def rich_print(text, style: str = ""):
console.print(text, style=style)
def random_background(folder: str = "background") -> str:
"""
Returns the filename of a random file in the specified folder.
Args:
folder(str): The folder containing the files.
Returns:
str: The filename of a randomly selected file in the folder.
"""
directory = Path(folder).absolute()
if not directory.exists():
directory.mkdir()
with KeepDir() as keep_dir:
keep_dir.chdir(folder)
files = os.listdir(".")
random_file = random.choice(files)
return Path(random_file).absolute()
def get_info(filename: str, kind: str):
global probe
try:
probe = ffmpeg.probe(filename)
except ffmpeg.Error as e:
console.log(f"{msg.ERROR}{e.stderr}")
logger.exception(e.stderr)
sys.exit(1)
if kind == 'video':
global video_stream
# Extract
for stream in probe['streams']:
if stream['codec_type'] == 'video':
video_stream = stream
break
duration = float(video_stream['duration'])
width = int(video_stream['width'])
height = int(video_stream['height'])
return {'width': width, 'height': height, 'duration': duration}
elif kind == 'audio':
global audio_stream
# Extract
for stream in probe['streams']:
if stream['codec_type'] == 'audio':
audio_stream = stream
break
duration = float(audio_stream['duration'])
return {'duration': duration}
def convert_time(time_in_seconds):
"""
Converts time in seconds to a string in the format "hh:mm:ss.mmm".
Args:
time_in_seconds (float): The time in seconds to be converted.
Returns:
str: The time in the format "hh:mm:ss.mmm".
"""
hours = int(time_in_seconds // 3600)
minutes = int((time_in_seconds % 3600) // 60)
seconds = int(time_in_seconds % 60)
milliseconds = int((time_in_seconds - int(time_in_seconds)) * 1000)
return f"{hours:02d}:{minutes:02d}:{seconds:02d}.{milliseconds:03d}"
def rgb_to_bgr(rgb: str) -> str:
"""
Converts a color from RGB to BGR.
Args:
rgb (str): The color in RGB format.
Returns:
str: The color in BGR format.
Example:
>>> rgb_to_bgr("FFF000")
"000FFF"
"""
return rgb[4:6] + rgb[2:4] + rgb[0:2]
|