tv/getwebdavlist.py

#!/usr/bin/env python
# -*- coding: utf- -*-

import argparse
from webdav3.client import Client
import os
import time
import re

count=0

def walk(client, current_path="/", output_file=None, replaceroot=None):
    items=client.list(current_path)
    filetype_re=re.compile(r'\.(png|jpg|jpeg|bmp|gif|doc|nfo|flac|mp3|wma|ape|cue|wav|dst|dff|dts|ac3|eac3|txt)$')
    # print(items)
    for item in items[1:]:
        full_path = f"{current_path}/{item}".replace("//", "/")
        try:
            # is_dir = client.is_dir(full_path)
            is_dir = full_path.endswith("/")
        except:
            continue
        if is_dir:  # 判断是否为目录
            #global count
            #count=count+1
            #if count%4==0:
            #    time.sleep(1)
            try:
                walk(client, full_path, output_file, replaceroot)
            except:
                pass
        else:
            if filetype_re.search(full_path) != None or "BDMV" in full_path:
                continue
            size = client.info(full_path).get('size', 0)
            if replaceroot!=None:
                if replaceroot=="":
                    full_path=os.path.join("/",*full_path.split("/")[2:])
                else:
                    full_path=os.path.join("/",replaceroot,*full_path.split("/")[2:])
            print(f"{full_path}\t{size}")
            if output_file != None:
                output_file.write(f"{full_path}\t{size}\n")


from urllib.parse import urlparse

def extract_url_components(url):
    """分解 URL 为 schema（协议）、hostname（主机地址）、path（路径）"""
    parsed = urlparse(url)

    # 提取核心组件
    schema = parsed.scheme or "http"  # 默认协议处理
    hostname = f"{parsed.hostname}:{parsed.port}"        # 自动过滤端口和认证信息
    path = parsed.path.rstrip('/') or '/'  # 路径标准化

    return schema, hostname, path


def main():
    parser = argparse.ArgumentParser(description='快速遍历WebDAV目录')
    parser.add_argument('--url', type=str, required=True, help='WebDAV URL')
    parser.add_argument('--username', type=str, required=True, help='WebDAV username')
    parser.add_argument('--password', type=str, required=True, help='WebDAV password')
    parser.add_argument('--output', type=str, required=True, help='outputfile')
    parser.add_argument('--replaceroot', type=str, default=None, required=False, help='替换根目录名称')
    args = parser.parse_args()

    schema, hostname, path = extract_url_components(args.url)
    print(schema,hostname,path)

    options = {
        'webdav_hostname': schema+"://"+hostname,
        'webdav_root': '/',
        'webdav_login': args.username,
        'webdav_password': args.password,
        'disable_check': True,  # 跳过 SSL 证书验证:ml-citation{ref="6" data="citationList"}
        'disable_head': True,  # 跳过 SSL 证书验证:ml-citation{ref="6" data="citationList"}
    }
    client = Client(options)

    print("WebDAV URL:", args.url)
    output_file = open(args.output, mode="a", encoding="utf-8")
    walk(client, path, output_file,args.replaceroot)
    output_file.close()

if __name__ == '__main__':
    main()