64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
import re
|
||
from datetime import datetime
|
||
from typing import Dict, Tuple
|
||
|
||
# 你 COS 的公共前缀可以做成默认参数，便于不同环境覆盖
|
||
DEFAULT_COS_PREFIX = "https://data-miningc01-1318950322.cos.ap-shanghai-adc.myqcloud.com/"
|
||
|
||
def extract_datetime_from_filename(
|
||
bag_filename: str,
|
||
cos_prefix: str = DEFAULT_COS_PREFIX,
|
||
) -> Tuple[str, Dict[str, str]]:
|
||
"""
|
||
从 .bag 文件名中提取采集时间，并生成对应的多视角视频 URL。
|
||
|
||
期望文件名包含片段：_{YYYYMMDD}-{HHMMSS}_ 或以 .bag 结尾，例如：
|
||
PL162802_event_all_time_event_20250416-193015_0.bag
|
||
PL123456_event_all_time_event_20250828-221018.bag
|
||
xx_20250524-180926_.bag
|
||
xx_20250524-180926_3.bag
|
||
|
||
返回:
|
||
(datetime_full_str, video_urls)
|
||
- datetime_full_str: 'YYYY-MM-DD HH:MM:SS'
|
||
- video_urls: COS 上该 bag 的多视角视频 URL 字典
|
||
|
||
异常:
|
||
ValueError: 当文件名不匹配预期格式时抛出
|
||
"""
|
||
# 提取日期时间
|
||
match = re.search(r"_(\d{8})-(\d{6})(?:_|\.|$)", bag_filename)
|
||
|
||
if not match:
|
||
raise ValueError(f"文件名格式错误，无法提取时间信息：{bag_filename}")
|
||
|
||
date_part, time_part = match.groups()
|
||
year, month, day = date_part[:4], date_part[4:6], date_part[6:8]
|
||
hour, minute, second = time_part[:2], time_part[2:4], time_part[4:6]
|
||
|
||
datetime_full = f"{year}-{month}-{day} {hour}:{minute}:{second}"
|
||
|
||
|
||
print(f'datetime_full:{datetime_full}')
|
||
# 生成视频路径（与现有逻辑一致）
|
||
# 去掉 .bag 后缀，再补上 " - {View}.mp4"
|
||
base = bag_filename[:-4] if bag_filename.endswith(".bag") else bag_filename
|
||
view_map = {
|
||
"wide": "Wide",
|
||
"left": "Left",
|
||
"right": "Right",
|
||
"rear_left": "Rear Left",
|
||
"rear_right": "Rear Right",
|
||
"rear": "Rear",
|
||
}
|
||
video_urls = {}
|
||
for key, label in view_map.items():
|
||
video_final_path = f"{base}- {label}.mp4"
|
||
# 目录层级：momenta/videos/YYYY/MM/DD/{文件名 - {View}.mp4}
|
||
video_key = f"momenta/videos/{year}/{month}/{day}/{video_final_path}"
|
||
video_urls[key] = f"{cos_prefix}{video_key}"
|
||
|
||
print(f'v{video_urls}')
|
||
|
||
return datetime_full, video_urls
|