feat: feishu notify

Author: cc3630

README.md

@@ -144,14 +144,17 @@ S3 支持使用虚拟域名作为 endpoint，即可以将 region 或者 bucket 
 
 - S3_ENABLE: 选填，是否启用 S3 存储备份，true 表示启用
 - S3_EP: 选填，S3 url，例如 https://minio-api.36node.com
-- S3_EP_VIRTUAL: 选填，是否启用虚拟 host url，true 表示启用，使用阿里云 oss 时，需要设置为true，minio 无需处理
+- S3_EP_VIRTUAL: 选填，是否启用虚拟 host url，true 表示启用，使用阿里云 oss 时，需要设置为 true，minio 无需处理
 - S3_ACCESS_KEY: 选填，S3 access key
 - S3_ACCESS_SECRET: 选填，S3 access secret
 - S3_REGION: 选填，地区名
 - S3_BUCKET: 选填，要存储的桶名
 - S3_PREFIX: 选填，要存储的前缀
 - S3_SIGNATURE_VERSION: 选填，签名版本，使用阿里云 oss 时，需要设置为"s3"，minio 无需处理
 
+- FEISHU_NOTIFY_TOKEN: 选填，飞书推送 token，为空时不推送
+- FEISHU_NOTIFY_TITLE: 选填，飞书推送标题，FEISHU_NOTIFY_TOKEN 为空时无效，例如 haiviv cn，此时飞书通知的标题为 `haivivi cn mongodb 备份出错`
+
 ### restore
 
 - S3_ENABLE: 选填，是否从 S3 中进行恢复，true 表示启用

docker/Dockerfile

@@ -5,7 +5,7 @@ RUN apt-get update && \
   apt-get install -y zip psmisc python3 python3-pip
 
 RUN pip3 install --upgrade pip
-RUN pip3 install boto3
+RUN pip3 install boto3 requests
 
 WORKDIR /app
 ADD . /app

docker/backup.py

@@ -6,6 +6,7 @@
 from urllib.parse import urlparse
 import boto3
 from botocore.client import Config
+import requests
 
 # 数据库清理备份脚本
 # 1. 按要求备份数据，并保存到指定路径
@@ -35,6 +36,7 @@
     "S3_REGION",
     "S3_BUCKET",
     "S3_PREFIX",
+    "FEISHU_TOKEN",
 ]
 
 for key in must_inputs:
@@ -93,6 +95,13 @@ def check_bool(key):
     os.environ["S3_SIGNATURE_VERSION"] if check_var("S3_SIGNATURE_VERSION") else None
 )
 
+feishu_notify_token = (
+    os.environ["FEISHU_NOTIFY_TOKEN"] if check_var("FEISHU_NOTIFY_TOKEN") else None
+)
+feishu_notify_title = (
+    os.environ["FEISHU_NOTIFY_TITLE"] if check_var("FEISHU_NOTIFY_TITLE") else None
+)
+
 # 计算当前日期，按照 年月日时分 格式
 date = (datetime.now() + timedelta(hours=8)).strftime("%Y%m%d%H%M%S")
 
@@ -236,6 +245,34 @@ def upload_s3(prefix):
             print(f"Deleted s3 old backup files: {keys_to_remove}")
 
 
+def send_feishu_notify(msg):
+    data = {
+        "msg_type": "post",
+        "content": {
+            "post": {
+                "zh_cn": {
+                    "title": f"{feishu_notify_title} mongodb 备份出错",
+                    "content": [
+                        [
+                            {
+                                "tag": "text",
+                                "text": msg,
+                            }
+                        ]
+                    ],
+                }
+            }
+        },
+    }
+    try:
+        requests.post(
+            f"https://open.feishu.cn/open-apis/bot/v2/hook/{feishu_notify_token}",
+            json=data,
+        )
+    except Exception as e:
+        print("Feishu notify error:", e)
+
+
 try:
     if not os.path.exists(backup_path):
         os.makedirs(backup_path)
@@ -263,4 +300,8 @@ def upload_s3(prefix):
 
 except Exception as e:
     print("backup error:", e)
+
+    if feishu_notify_token:
+        send_feishu_notify(f"错误信息: {e}")
+
     sys.exit()