File size: 2,768 Bytes
0191eab b9adb22 0191eab b9adb22 0191eab b9adb22 0191eab b9adb22 0191eab b9adb22 0191eab b9adb22 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 |
import os
from pathlib import Path
import random
import sys
import ffmpeg
from rich.console import Console
import msg
console = Console()
def rich_print(text, style: str = ""):
console.print(text, style=style)
def random_background(folder: str = "background") -> Path:
"""
Returns the filename of a random file in the specified folder.
Args:
folder(str): The folder containing the files.
Returns:
Path: The absolute path of the random file.
"""
# Get the absolute path of the folder
directory = Path(folder).absolute()
# Create the folder if it does not exist
if not directory.exists():
directory.mkdir()
# Select a random background video for the clip inside the folder
random_file = random.choice(os.listdir(directory))
# Return the absolute path of the random file adding the folder path
# Concat the folder path with the random file name
return directory / random_file
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}")
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]
|