Difficult-Rocket/.github/workflows/dsm.py

#  -------------------------------
#  Difficult Rocket
#  Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
#  All rights reserved
#  -------------------------------

import os
import time
from pprint import pprint
from synology_api import filestation


class DSM:
    def __init__(self, docs_path: str, dsm_path: str):
        self.docs_path = docs_path
        self.dsm_path = dsm_path
        self.token = os.environ['DSM_TOKEN']
        self.fl = filestation.FileStation(ip_address='hws.shenjack.top',
                                          port=5000,
                                          username='github',
                                          password=self.token,
                                          secure=False,
                                          cert_verify=False,
                                          dsm_version=7,
                                          debug=True,
                                          interactive_output=False)

    def list_files(self):
        # 输出 文档构建目录 的内容
        print(f'==========输出 {self.docs_path} 的内容==========')
        for root, dirs, files in os.walk(self.docs_path):
            print(root, dirs)
        print('==========就这些==========')

    def clear_dsm(self):
        # 清空 DSM 的 /web/dr 目录
        delete_task = self.fl.start_delete_task(self.dsm_path, recursive=True)
        delete_task_id = delete_task['taskid']
        time.sleep(1)  # 等待 1 秒 保证任务已经完成
        pprint(self.fl.get_delete_status(delete_task_id)['data']['finished'])

    def check_md5(self, local_md5: str) -> bool:
        """
        检查本地构建的文档和 DSM 上的文档的 md5 是否一致
        :param local_md5:
        :return: True: 一致 False: 不一致
        """
        # 打开提供的md5文件
        try:
            with open(local_md5, 'r', encoding='utf-8') as f:
                md5 = f.read()
        except FileNotFoundError:
            print(f'文件 {local_md5} 不存在')
            return False
        # 检测是否存在 dsm 上的 md5.txt
        dsm_files = self.fl.get_file_list(folder_path='/web/dr')
        if dsm_files.get('error'):
            print(dsm_files['error'])
            return False
        dsm_files = dsm_files['data']['files']
        if '/web/dr/md5.txt' not in [file['path'] for file in dsm_files]:
            print('dsm md5.txt 不存在')
            return False
        # 下载 dsm 上的 md5.txt
        try:
            self.fl.get_file(path='/web/dr/md5.txt', mode='download', dest_path='./docs/book')
            with open('./docs/book/md5.txt', 'r', encoding='utf-8') as f:
                md5_last = f.read()
            if md5 == md5_last:
                return True
        except Exception as e:
            print(e)
            return False

    def upload_docs(self, local_md5: str):
        # 上传本地构建的文档到 DSM
        print(f'==========上传 {self.docs_path} 到 {self.dsm_path}==========')
        # 使用 os.walk 递归遍历文件夹 依次按照对应路径上传
        # 上传的时候 目标路径为本地路径的相对路径
        for root, dirs, files in os.walk(self.docs_path):
            for file in files:
                file_path = os.path.join(root, file)
                dest_path = f'{self.dsm_path}{root[len(self.docs_path):]}'
                dest_path = dest_path.replace('\\', '/')
                # 输出 文件路径 和 目标路径
                print(f'{file_path} -> {dest_path}', end=' ')
                pprint(self.fl.upload_file(dest_path=dest_path,
                                           file_path=file_path,
                                           overwrite=True))
                # self.fl.upload_file(dest_path=dest_path,
                #                     file_path=file_path,
                #                     overwrite=True)
        # 上传本地的 md5 文件
        print(f'{local_md5} -> {self.dsm_path}', end=' ')
        pprint(self.fl.upload_file(dest_path=self.dsm_path,
                                   file_path=local_md5,
                                   overwrite=True))
        print('==========上传完成==========')


def main():
    docs_path = 'docs/book'
    dsm_path = '/web/dr'
    dsm = DSM(docs_path, dsm_path)
    dsm.list_files()
    if dsm.check_md5('docs/md5.txt'):
        print('md5 一致，不需要上传')
        return 0
    dsm.clear_dsm()
    dsm.upload_docs('docs/md5.txt')

    dsm.fl.logout()


if __name__ == '__main__':
    main()