|
|
|
|
|
from Util.OssUtil import *
|
|
|
|
|
|
from Util.CommonUtil import *
|
|
|
|
|
|
# 备份思路：
|
|
|
|
|
|
# 使用Navicat备份，然后上传到OSS
|
|
|
|
|
|
# 参考文档：https://blog.csdn.net/weixin_56631459/article/details/133904669
|
|
|
|
|
|
# 备份的目录：
|
|
|
|
|
|
# C:\Users\Administrator\Documents\Navicat\PostgreSQL\Servers\10.10.14.71\huiya_db\public
|
|
|
|
|
|
|
|
|
|
|
|
# 遍历目录下所有nb3文件，删除30天前的文件
|
|
|
|
|
|
import os
|
|
|
|
|
|
from datetime import datetime
|
|
|
|
|
|
|
|
|
|
|
|
# PostgreSQL 配置
|
|
|
|
|
|
backup_dir = r'C:\Users\Administrator\Documents\Navicat\PostgreSQL\Servers\10.10.14.71\huiya_db\public'
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def calculate_days_difference(timestamp_str):
|
|
|
|
|
|
# 将文件名中的时间戳字符串转换为datetime对象
|
|
|
|
|
|
file_datetime = datetime.strptime(timestamp_str, "%Y%m%d%H%M%S")
|
|
|
|
|
|
|
|
|
|
|
|
# 获取当前的datetime
|
|
|
|
|
|
current_datetime = datetime.now()
|
|
|
|
|
|
|
|
|
|
|
|
# 计算两个datetime之间的差异，并转换为天数
|
|
|
|
|
|
time_difference = current_datetime - file_datetime
|
|
|
|
|
|
days_difference = time_difference.days
|
|
|
|
|
|
|
|
|
|
|
|
return days_difference
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
|
auth = oss2.Auth(accessKeyId, accessKeySecret)
|
|
|
|
|
|
bucket = oss2.Bucket(auth, endpoint, bucketName)
|
|
|
|
|
|
|
|
|
|
|
|
paths = os.walk(backup_dir)
|
|
|
|
|
|
for path, dir_lst, file_lst in paths:
|
|
|
|
|
|
for file_name in file_lst:
|
|
|
|
|
|
if file_name.endswith(".nb3"):
|
|
|
|
|
|
timestamp_str = file_name.replace(".nb3", "")
|
|
|
|
|
|
days_diff = calculate_days_difference(timestamp_str)
|
|
|
|
|
|
# print(f"时间戳与当前时间相差 {days_diff} 天")
|
|
|
|
|
|
if days_diff > 30:
|
|
|
|
|
|
os.remove(os.path.join(path, file_name))
|
|
|
|
|
|
printf("发现30天前的备份文件，已删除掉：" + file_name)
|
|
|
|
|
|
else:
|
|
|
|
|
|
# 上传所有nb3文件到oss上去，如果上面已经存在，则跳过
|
|
|
|
|
|
prefix = 'DataBaseBackup/'
|
|
|
|
|
|
key = prefix + file_name
|
|
|
|
|
|
if not bucket.object_exists(key):
|
|
|
|
|
|
uploadOss(key, os.path.join(path, file_name))
|
|
|
|
|
|
printf(f"上传文件 {file_name} 到oss成功。")
|
|
|
|
|
|
else:
|
|
|
|
|
|
# 获取文件元信息
|
|
|
|
|
|
meta = bucket.head_object(key)
|
|
|
|
|
|
if meta.content_length != os.path.getsize(os.path.join(path, file_name)):
|
|
|
|
|
|
bucket.delete_object(key)
|
|
|
|
|
|
uploadOss(key, os.path.join(path, file_name))
|
|
|
|
|
|
printf(f"文件已存在，但文件大小不一致，已删除并重新上传 {file_name} 到oss成功。")
|
|
|
|
|
|
else:
|
|
|
|
|
|
printf(f"文件 {file_name} 已经存在oss上。")
|
|
|
|
|
|
printf("恭喜，所有备份操作成功完成！")
|
