tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-06 19:25:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(screen-recorder): save using webp format

Browse Source
This commit is contained in:
arkohut 2024-08-21 17:39:52 +08:00
parent b53d816b17
commit e115e00702
2 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
screen-recorder/record.py
View File

@ -82,12 +82,12 @@ def take_screenshot(
["screencapture", "-C", "-x", "-D", str(display + 1), temp_filename]
)
# 压缩图像为 JPEG 并添加元数据
# 压缩图像为 WebP 并添加元数据
with Image.open(temp_filename) as img:
img = img.convert("RGB")
jpeg_filename = os.path.join(
webp_filename = os.path.join(
os.path.join(base_dir, date),
f"screenshot-{timestamp}-of-{screen_name}.jpg",
f"screenshot-{timestamp}-of-{screen_name}.webp",
)
# 计算当前截图的哈希值
@ -124,15 +124,15 @@ def take_screenshot(
}
# 使用 write_image_metadata 函数写入元数据
img.save(jpeg_filename, format="JPEG", quality=85)
write_image_metadata(jpeg_filename, metadata)
img.save(webp_filename, format="WebP", quality=85)
write_image_metadata(webp_filename, metadata)
save_screen_sequences(base_dir, screen_sequences, date)
# 删除临时 PNG 文件
os.remove(temp_filename)
# 添加 JPEG 文件到截图列表
screenshots.append(jpeg_filename)
# 添加 WebP 文件到截图列表
screenshots.append(webp_filename)
# 记录成功的截图
worklog.write(f"{timestamp} - {screen_name} - Saved\n")

8
screen-recorder/utils.py
View File

@ -17,6 +17,8 @@ def write_image_metadata(image_path, metadata):
metadata_info = PngInfo()
metadata_info.add_text("Description", json.dumps(metadata))
img.save(image_path, "PNG", pnginfo=metadata_info)
elif image_path.lower().endswith(".webp"):
img.save(image_path, "WebP", quality=85, metadata=json.dumps(metadata))
else:
print(f"Skipping unsupported file format: {image_path}")
@ -39,6 +41,12 @@ def get_image_metadata(image_path):
return json.loads(existing_description)
except json.JSONDecodeError:
return {}
elif image_path.lower().endswith(".webp"):
existing_metadata = img.info.get("metadata", "{}")
try:
return json.loads(existing_metadata)
except json.JSONDecodeError:
return {}
else:
print(f"Unsupported file format: {image_path}")
return None