ximu

python3 --version

pip --version

from webdav3.client import Client
import argparse, os, requests, time

'''
Function to traverse the WebDAV server
If depth is None, it will recursively traverse the entire WebDAV server
If depth is a positive integer, it will recursively traverse to the specified depth
If depth is 0, it will only traverse the files and folders in the current folder, without continuing to recursively traverse the next level of folders.
'''
def list_files(webdav_url, username, password, show_path, depth=None, path='', count=0, proxies=None):
    options = {
        'webdav_hostname': webdav_url,
        'webdav_login': username,
        'webdav_password': password,
        'proxies': proxies
    }

    client = Client(options)
    directory = []
    files = []
    q = 1
    while q < 15:
        try:
            items = client.list()
        except:
            print(f'Connection failed for the {q} time, retrying in {q+1} seconds...')
            q += 1
            time.sleep(q)
        else:
            if q > 1:
                print('Reconnection successful...')
            break

    if q == 15:
        print('Connection failed, please check your network settings!')
        exit()

    for item in items[1:]:
        if item[-1] == '/':
            if depth is None or depth > 0:
                subdirectory, subfiles, count = list_files(webdav_url + item, username, password, show_path, depth=None if depth is None else depth - 1, path=path+item, count=count)
                directory += [item + subitem for subitem in subdirectory]
                files += [item + subitem for subitem in subfiles]
            else:
                directory.append(item)
        else:
            files.append(item)
            count += 1
    if show_path and path:
        print(f'Current folder path: {path}')
    return directory, files, count

'''
Download function
Used for downloading 'ASS', 'SRT', 'SSA', 'NFO', 'JPG', 'PNG' files
'''   
def download_file(url, local_path, filename, total_count):
    p = 1
    while p < 10:
        try:
            print('Downloading: ' + filename)
            r = requests.get(url.replace('/dav', '/d'), proxies=proxies)
            os.makedirs(os.path.dirname(local_path), exist_ok=True)
            with open(local_path, 'wb') as f:
                f.write(r.content)
                f.close()
        except:
            print(f'Download failed for the {p} time, retrying in {p + 1} seconds...')
            p += 1
            time.sleep(p)
        else:
            if p > 1:
                print('Re-download successful!')
            print(filename + ' downloaded successfully!')
            break
        progress = int((p / 10) * 100)
        print(f'Completed {progress}%, total {total_count} files')

parser = argparse.ArgumentParser(description='Autofilm script')
parser.add_argument('--webdav_url', type=str, help='WebDAV server address', required=True)
parser.add_argument('--username', type=str, help='WebDAV account', required=True)
parser.add_argument('--password', type=str, help='WebDAV password', required=True)
parser.add_argument('--output_path', type=str, help='Output file directory', default='./Media/')
parser.add_argument('--subtitle', type=str, help='Whether to download subtitle files', choices=['true', 'false'], default='true')
parser.add_argument('--nfo', type=str, help='Whether to download NFO files', choices=['true', 'false'], default='false')
parser.add_argument('--img', type=str, help='Whether to download JPG and PNG files', choices=['true', 'false'], default='false')
parser.add_argument('--show_path', type=str, help='Whether to show folder path during traversal', choices=['true', 'false'], default='false')
parser.add_argument('--proxy', type=str, help='HTTP proxy server in the format IP:port')
args = parser.parse_args()

print('Startup parameters:')
print(f'WebDAV server address: {args.webdav_url}')
print(f'WebDAV login username: {args.username}')
print(f'WebDAV login password: {args.password}')
print(f'File output path: {args.output_path}')
print(f'Whether to download subtitles: {args.subtitle}')
print(f'Whether to download movie information: {args.nfo}')
print(f'Whether to download images: {args.img}')
print(f'Whether to show folder path during traversal: {args.show_path}')

proxies = None
if args.proxy:
    proxies = {
        'http': f'http://{args.proxy}',
        'https': f'http://{args.proxy}'
    }

directory, files, count = list_files(args.webdav_url, args.username, args.password, args.show_path, depth=None, path='', count=0, proxies=proxies)

urls = [args.webdav_url + item for item in directory + files]

download_count = 0

for url in urls:
    if url[-1] == '/':
        continue
    filename = os.path.basename(url)
    local_path = os.path.join(args.output_path, url.replace(args.webdav_url, '').lstrip('/'))
    file_ext = filename[-3:].upper()

    if file_ext in ['MP4', 'MKV', 'FLV', 'AVI', 'WMV']:
        if not os.path.exists(os.path.join(args.output_path, filename[:-3] + 'strm')):
            print('Processing: ' + filename)
            try:
                os.makedirs(os.path.dirname(local_path), exist_ok=True)
                with open(os.path.join(local_path[:-3] + 'strm'), "w", encoding='utf-8') as f:
                    f.write(url.replace('/dav', '/d'))
            except:
                print(filename + ' processing failed, the filename contains special characters, it is recommended to rename it!')
    elif args.subtitle == 'true' and file_ext in ['ASS', 'SRT', 'SSA', 'SUB']:
        if not os.path.exists(local_path):
            download_file(url, local_path, filename, count)
            download_count += 1
    elif args.nfo == 'true' and file_ext == 'NFO':
        if not os.path.exists(local_path):
            download_file(url, local_path, filename, count)
            download_count += 1
    elif args.img == 'true' and file_ext in ['JPG', 'PNG']:
        if not os.path.exists(local_path):
            download_file(url, local_path, filename, count)
            download_count += 1

    progress = int((download_count / count) * 100)
    print(f'Completed {progress}%, total {count} files')

print('Processing complete!')