generated from dellevin/template
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
pin-tu 图片浏览工具函数
|
|
"""
|
|
import os
|
|
import re
|
|
|
|
|
|
def atoi(text):
|
|
"""辅助函数：将文本中的数字部分转换为整数，用于排序"""
|
|
return int(text) if text.isdigit() else text
|
|
|
|
|
|
def natural_keys(text):
|
|
"""自然排序键生成器"""
|
|
return [atoi(c) for c in re.split(r'(\d+)', text)]
|
|
|
|
|
|
def find_media_in_folder(folder_path):
|
|
"""在指定文件夹中查找所有图片、视频和其他文件"""
|
|
allowed_image_exts = {'.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp'}
|
|
allowed_video_exts = {'.mp4', '.avi', '.mov', '.wmv', '.flv', '.webm', '.m4v', '.mkv', '.mpg', '.mpeg', '.3gp', '.3g2'}
|
|
|
|
try:
|
|
files = os.listdir(folder_path)
|
|
except PermissionError:
|
|
return [], [], []
|
|
|
|
image_files = []
|
|
video_files = []
|
|
other_files = []
|
|
for f in files:
|
|
if not os.path.isfile(os.path.join(folder_path, f)):
|
|
continue
|
|
ext = os.path.splitext(f)[1].lower()
|
|
if ext in allowed_image_exts:
|
|
image_files.append(f)
|
|
elif ext in allowed_video_exts:
|
|
video_files.append(f)
|
|
else:
|
|
other_files.append(f)
|
|
|
|
return sorted(image_files, key=natural_keys), sorted(video_files, key=natural_keys), sorted(other_files, key=natural_keys)
|
|
|
|
|
|
def get_subfolders(folder_path):
|
|
"""获取指定文件夹下的所有子文件夹"""
|
|
try:
|
|
items = os.listdir(folder_path)
|
|
except PermissionError:
|
|
return []
|
|
subfolders = [item for item in items if os.path.isdir(os.path.join(folder_path, item))]
|
|
return sorted(subfolders, key=natural_keys)
|
|
|
|
|
|
def build_breadcrumbs(subpath, base_url='/pin-tu/browse'):
|
|
"""构建面包屑导航数据
|
|
base_url 示例: /pin-tu/browse/RDovRGF0YQ
|
|
返回 URL 格式: {base_url}?p={累计子路径}
|
|
"""
|
|
path_parts = [p for p in subpath.split('/') if p]
|
|
breadcrumbs = [{'name': 'Home', 'url': base_url}]
|
|
cumulative = ''
|
|
for part in path_parts:
|
|
cumulative += ('/' if cumulative else '') + part
|
|
breadcrumbs.append({'name': part, 'url': f'{base_url}?p={cumulative}'})
|
|
return breadcrumbs
|
|
|
|
|
|
def is_safe_path(root, target):
|
|
"""安全检查：防止路径穿越"""
|
|
return os.path.normpath(target).startswith(os.path.normpath(root))
|