53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
import os
|
||
import glob
|
||
|
||
|
||
def get_latest_n_segments(camera_root_dir: str, n: int) -> list:
|
||
"""
|
||
获取最新的n个TS分片
|
||
|
||
逻辑：
|
||
1. 获取index_code文件夹下所有时间戳文件夹
|
||
2. 按时间戳名称降序排序（最新的在前）
|
||
3. 从最新的时间戳文件夹开始获取分片
|
||
4. 如果分片数不足n，继续从上一个时间戳文件夹获取
|
||
5. 返回最新的n个分片路径list（按时间顺序，最旧的在前）
|
||
"""
|
||
if not os.path.exists(camera_root_dir):
|
||
return []
|
||
|
||
# 获取所有时间戳文件夹并排序（字符串排序即时间排序）
|
||
timestamp_folders = []
|
||
for folder_name in os.listdir(camera_root_dir):
|
||
folder_path = os.path.join(camera_root_dir, folder_name)
|
||
if os.path.isdir(folder_path):
|
||
timestamp_folders.append(folder_name)
|
||
|
||
if not timestamp_folders:
|
||
return []
|
||
|
||
# 降序排序，最新的在前
|
||
timestamp_folders.sort(reverse=True)
|
||
|
||
# 收集分片
|
||
all_segments = []
|
||
for ts_folder in timestamp_folders:
|
||
ts_folder_path = os.path.join(camera_root_dir, ts_folder)
|
||
pattern = os.path.join(ts_folder_path, "segment_*.ts")
|
||
segment_files = glob.glob(pattern)
|
||
|
||
# 按分片序号排序
|
||
segment_files.sort(key=lambda x: int(os.path.basename(x).split('_')[-1].split('.')[0]))
|
||
|
||
all_segments.extend(segment_files)
|
||
|
||
# 已经收集够了
|
||
if len(all_segments) >= n:
|
||
break
|
||
|
||
# 返回最新的n个（取最后n个，因为最新的在后面）
|
||
if len(all_segments) >= n:
|
||
return all_segments[-n:]
|
||
else:
|
||
return all_segments
|