#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import argparse
from p115 import P115Client, P115FileSystem
import time
import traceback

count=0

def walk_dir(fs,f):
    dirlist=list()
    file_list = fs.listdir_attr()
    for file_obj in file_list:
        if not file_obj.is_directory:
            print(file_obj.path+"\t"+str(file_obj.size))
            f.write(file_obj.path+"\t"+str(file_obj.size)+"\n")
            f.flush()
        else:
            dirlist.append(file_obj.path)
    for dirItem in dirlist:
        fs.chdir(dirItem)
        global count
        count=count+1
        if count%4==0:
            time.sleep(1)
        walk_dir(fs,f)
    return

def main():
    parser = argparse.ArgumentParser(description='快速遍历115分享目录')
    parser.add_argument('--cookie', type=str, required=True, help='115Cookie')
    parser.add_argument('--url', type=str, required=True, help='115ShareUrl')
    parser.add_argument('--output', type=str, required=True, help='outputfile')
    args = parser.parse_args()
    cookie=args.cookie
    shareUrl=args.url
    client=P115Client(cookie)

    if not shareUrl.startswith("http"):
        shareUrl = "https://115.com/s/" + shareUrl

    print("cookie:"+args.cookie+", shareurl:"+shareUrl)
    fs = client.get_share_fs(shareUrl)

    count=0
    f=open(args.output, mode="a", encoding="utf-8")
    walk_dir(fs,f)

if __name__ == '__main__':
    main()