toolbox/baidu_netdisk/upload/upload.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
import argparse
import json

from project_settings import project_path
from toolbox.baidu_netdisk.baidu_netdisk_client import BaiduNetdiskClient


class UploadClient(BaiduNetdiskClient):
    def __init__(self, configdir: str = None):
        super(UploadClient, self).__init__(configdir=configdir)

    def upload_by_filename(self, src_file: str, tgt_file: str):
        # self.bypy_client.upload(
        #     r"C:\Users\tianx\Desktop\滴滴POC-2\lQLPJwCwSltNquvNBLTNDHKwOlnFBSTMn88JJBTGVuNCAA_3186_1204.png",
        #     "xianyu/customers/佳小雨/20251224/lQLPJwCwSltNquvNBLTNDHKwOlnFBSTMn88JJBTGVuNCAA_3186_1204.png"
        # )
        self.bypy_client.upload(
            src_file,
            tgt_file
        )
        return tgt_file


def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--key_of_credentials",
        default="baidu_netdisk_honeytian_credentials",
        type=str,
    )
    parser.add_argument(
        "--config_dir",
        default=(project_path / "dotenv/baidu_netdisk/config_dir").as_posix(),
        type=str,
    )
    parser.add_argument(
        "--src_file",
        default=(project_path / "README.md").as_posix(),
        type=str,
    )
    parser.add_argument(
        "--tgt_file",
        default="video_platform/upload/README.md",
        type=str,
    )
    args = parser.parse_args()
    return args


def main():
    args = get_args()

    import log
    from project_settings import environment, project_path, log_directory, time_zone_info

    log.setup_size_rotating(log_directory=log_directory, tz_info=time_zone_info)

    client = UploadClient(
        # configdir=args.config_dir
    )

    flag = client.check_login()
    print(f"flag: {flag}")
    credentials_info = environment.get(key=args.key_of_credentials, dtype=json.loads)
    client.login_with_credentials_info(credentials_info=credentials_info)
    flag = client.check_login()
    print(f"flag: {flag}")

    client.upload_by_filename(
        src_file=args.src_file,
        tgt_file=args.tgt_file,
    )
    return


if __name__ == "__main__":
    main()