117 lines
4.5 KiB
Python
117 lines
4.5 KiB
Python
# -------------------------------
|
|
# 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()
|