AutoInfo/utils/gotify.py

# -*- coding: utf-8 -*-
import httpx
import json
import os


class GotifyNotifier:
    def __init__(self, title, message, token_name=''):
        self.gotify_url = 'https://gotify.erhe.top'
        self.app_token = self.match_token_name(token_name)
        self.title = title
        self.message = message

    def match_token_name(self, name):
        token_name_dict = {}
        # 读取项目根目录下的 gotify_config.json 文件
        gotify_config_path = os.path.join(PROJECT_PATH, 'gotify_config.json')
        with open(gotify_config_path, 'r') as f:
            token_name_dict = json.load(f)
        token = token_name_dict.get(name)
        if token:
            return token
        else:
            return token_name_dict['base']

    def send_message(self):
        # 构建POST请求的headers
        headers = {
            'Content-Type': 'application/json'
        }

        # 构建POST请求的body
        body = {
            'title': self.title,
            'message': self.message
        }

        # 发送POST请求
        with httpx.Client() as client:
            response = client.post(
                url=f"{self.gotify_url}/message?token={self.app_token}",
                headers=headers,
                json=body
            )

        # 或者可以使用 curl
        # curl -k "https://gotify.erhe.top/message?token=A0Xg6ZE5946iBYg" -F "title=测试发送信息" -F "message=假装有信息,测试发送" -F "priority=5"

        # 检查响应状态码
        if response.status_code == 200:
            print('Gotify Message sent successfully!')
        else:
            print('Failed to send message:', response.text)