1
from webdav3.client import Client
2
import os, time, requests
3

4
def list_files(webdav_url, username, password):
5
  # 创建WebDAV客户端
6
  options = {
7
    'webdav_hostname': webdav_url,
8
    'webdav_login': username,
9
    'webdav_password': password
10
  }
11

12
  client = Client(options)
13
  mulu = []
14
  wenjian = []
15
  q = 1
16
  while q < 15:
17
    try:
18
      # 获取WebDAV服务器上的文件列表
19
      files = client.list()
20
    except:
21
      q += 1
22
      print('连接失败，1秒后重试...')
23
      time.sleep(1)
24
    else:
25
      if q > 1:
26
        print('重连成功...')
27
      break
28

29
  if q >= 15:
30
    print(f"连接 {webdav_url} 失败，已达到最大重试次数")
31
    return [], []
32

33
  for file in files[1:]:
34
    if file[-1] == '/':
35
      mulu.append(file)
36
    else:
37
      wenjian.append(file)
38
  return mulu, wenjian
39

40
def traverse_webdav(base_url, username, password, max_depth=10, current_depth=0):
41
  """
42
  递归遍历WebDAV目录，获取所有目录和文件
43

44
  Args:
45
    base_url: WebDAV基础URL
46
    username: 用户名
47
    password: 密码
48
    max_depth: 最大递归深度，防止无限递归
49
    current_depth: 当前递归深度
50

51
  Returns:
52
    tuple: (所有目录列表, 所有文件URL列表)
53
  """
54
  if current_depth >= max_depth:
55
    print(f"警告: 达到最大递归深度 {max_depth}，停止在 {base_url} 的递归")
56
    return [], []
57

58
  print(f"扫描目录: {base_url}")
59

60
  # 获取当前目录下的文件和子目录
61
  directories, files = list_files(base_url, username, password)
62

63
  all_directories = [base_url]
64
  all_files = [base_url + f for f in files]
65

66
  # 递归处理子目录
67
  for directory in directories:
68
    dir_url = base_url + directory
69
    sub_dirs, sub_files = traverse_webdav(dir_url, username, password, max_depth, current_depth + 1)
70
    all_directories.extend(sub_dirs)
71
    all_files.extend(sub_files)
72

73
  return all_directories, all_files
74

75
def process_media_file(file_url, save_path, webdav_url):
76
  """处理媒体文件，创建.strm文件"""
77
  if not os.path.exists(save_path + file_url.replace(webdav_url, '')[:-3] + 'strm'):
78
    print('正在处理：' + file_url.replace(webdav_url, ''))
79
    try:
80
      os.makedirs(os.path.dirname(save_path + file_url.replace(webdav_url, '')[:-3] + 'strm'), exist_ok=True)
81
      with open(save_path + file_url.replace(webdav_url, '')[:-3] + 'strm', "w", encoding='utf-8') as f:
82
        f.write(file_url.replace('/dav', '/d'))
83
    except:
84
      try:
85
        # 处理包含冒号等特殊字符的文件名
86
        os.makedirs(os.path.dirname(save_path + file_url.replace(webdav_url, '').replace('：', '.')[:-3] + 'strm'), exist_ok=True)
87
        with open(save_path + file_url.replace(webdav_url, '').replace('：', '.')[:-3] + 'strm', "w", encoding='utf-8') as f:
88
          f.write(file_url.replace('/dav', '/d'))
89
      except Exception as e:
90
        print(f"处理失败: {file_url.replace(webdav_url, '')} - 错误: {str(e)}")
91
        print(f"{file_url.replace(webdav_url, '')}处理失败，文件名包含特殊符号，建议重命名！")
92

93
def download_subtitle_file(file_url, save_path, webdav_url):
94
  """下载字幕文件"""
95
  if not os.path.exists(save_path + file_url.replace(webdav_url, '')):
96
    p = 1
97
    while p < 10:
98
      try:
99
        print('正在下载：' + save_path + file_url.replace(webdav_url, ''))
100
        r = requests.get(file_url.replace('/dav', '/d'))
101
        os.makedirs(os.path.dirname(save_path + file_url.replace(webdav_url, '')), exist_ok=True)
102
        with open(save_path + file_url.replace(webdav_url, ''), 'wb') as f:
103
          f.write(r.content)
104
      except Exception as e:
105
        p += 1
106
        print(f'下载失败，1秒后重试... 错误: {str(e)}')
107
        time.sleep(1)
108
      else:
109
        if p > 1:
110
          print('重新下载成功！')
111
        break
112
    if p >= 10:
113
      print(f"下载失败: {file_url.replace(webdav_url, '')} - 已达到最大重试次数")
114

115
# 主程序
116
if __name__ == "__main__":
117
  # 输入WebDAV地址、用户名和密码
118
  webdav_url = 'https://example.com/dav/folder/'  # alist webdav 地址 folder 为你的文件夹名称
119
  save_mulu = './'  # 输出路径
120
  username = ''  # 用户名 alist 用户名
121
  password = ''  # 密码 alist 密码
122

123
  # 控制是否下载字幕文件
124
  download_subtitle = False  # 设置为 False 将不会下载字幕文件
125

126
  # 设置最大递归深度
127
  max_depth = 10  # 可以根据实际需求调整
128

129
  print(f"开始遍历 WebDAV 目录: {webdav_url}")
130
  print(f"字幕下载功能: {'已启用' if download_subtitle else '已禁用'}")
131
  start_time = time.time()
132

133
  # 递归遍历所有目录和文件
134
  all_directories, all_files = traverse_webdav(webdav_url, username, password, max_depth)
135

136
  print(f"遍历完成，共发现 {len(all_directories)} 个目录，{len(all_files)} 个文件")
137
  print(f"遍历耗时: {time.time() - start_time:.2f} 秒")
138

139
  # 处理所有文件
140
  processed_count = 0
141
  media_count = 0
142
  subtitle_count = 0
143

144
  for file_url in all_files:
145
    extension = file_url.split('.')[-1].upper() if '.' in file_url else ''
146

147
    if extension in ['MP4', 'MKV', 'FLV', 'AVI']:
148
      process_media_file(file_url, save_mulu, webdav_url)
149
      processed_count += 1
150
      media_count += 1
151
    elif extension in ['ASS', 'SRT', 'SSA'] and download_subtitle:
152
      # 只有当 download_subtitle 为 True 时才下载字幕文件
153
      download_subtitle_file(file_url, save_mulu, webdav_url)
154
      processed_count += 1
155
      subtitle_count += 1
156

157
  print(f'处理完毕！共处理 {processed_count} 个文件（媒体文件: {media_count}, 字幕文件: {subtitle_count}）')
158
  print(f'字幕下载功能: {"已启用" if download_subtitle else "已禁用"}')
159
  input('按任意键退出...')