78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
|
|
||
|
|
import os
|
||
|
|
import subprocess
|
||
|
|
import logging
|
||
|
|
|
||
|
|
# Configure logging
|
||
|
|
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
||
|
|
logger = logging.getLogger("VideoConverter")
|
||
|
|
|
||
|
|
def get_codec(filepath):
|
||
|
|
try:
|
||
|
|
cmd = [
|
||
|
|
"ffprobe",
|
||
|
|
"-v", "error",
|
||
|
|
"-show_entries", "stream=codec_name",
|
||
|
|
"-of", "default=noprint_wrappers=1:nokey=1",
|
||
|
|
filepath
|
||
|
|
]
|
||
|
|
result = subprocess.run(cmd, capture_output=True, text=True, check=True)
|
||
|
|
return result.stdout.strip().split('\n')
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Error checking codec for {filepath}: {e}")
|
||
|
|
return []
|
||
|
|
|
||
|
|
def convert_to_h264(filepath):
|
||
|
|
try:
|
||
|
|
directory = os.path.dirname(filepath)
|
||
|
|
filename = os.path.basename(filepath)
|
||
|
|
name, ext = os.path.splitext(filename)
|
||
|
|
temp_filepath = os.path.join(directory, f"{name}_temp{ext}")
|
||
|
|
|
||
|
|
logger.info(f"Converting {filename} to H.264...")
|
||
|
|
|
||
|
|
cmd = [
|
||
|
|
"ffmpeg",
|
||
|
|
"-i", filepath,
|
||
|
|
"-c:v", "libx264",
|
||
|
|
"-c:a", "copy",
|
||
|
|
"-y",
|
||
|
|
temp_filepath
|
||
|
|
]
|
||
|
|
|
||
|
|
subprocess.run(cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
||
|
|
|
||
|
|
# Replace original file
|
||
|
|
os.remove(filepath)
|
||
|
|
os.rename(temp_filepath, filepath)
|
||
|
|
logger.info(f"Successfully converted: {filename}")
|
||
|
|
return True
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Failed to convert {filepath}: {e}")
|
||
|
|
if os.path.exists(temp_filepath):
|
||
|
|
os.remove(temp_filepath)
|
||
|
|
return False
|
||
|
|
|
||
|
|
def main():
|
||
|
|
directory = r"d:\dsWork\aiData\DouYin\DownloadedVideos"
|
||
|
|
if not os.path.exists(directory):
|
||
|
|
logger.error(f"Directory not found: {directory}")
|
||
|
|
return
|
||
|
|
|
||
|
|
files = [f for f in os.listdir(directory) if f.endswith(".mp4")]
|
||
|
|
total = len(files)
|
||
|
|
logger.info(f"Found {total} video files.")
|
||
|
|
|
||
|
|
for i, filename in enumerate(files, 1):
|
||
|
|
filepath = os.path.join(directory, filename)
|
||
|
|
codecs = get_codec(filepath)
|
||
|
|
|
||
|
|
if "hevc" in codecs:
|
||
|
|
logger.info(f"[{i}/{total}] HEVC detected: {filename}")
|
||
|
|
convert_to_h264(filepath)
|
||
|
|
else:
|
||
|
|
logger.info(f"[{i}/{total}] Skipping (already compatible or unknown): {filename} ({codecs})")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|