nonebot-plugin-apod 0.0.1__tar.gz

nonebot_plugin_apod-0.0.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 lyqgzbl
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

nonebot_plugin_apod-0.0.1/PKG-INFO

@@ -0,0 +1,28 @@
+Metadata-Version: 2.1
+Name: nonebot-plugin-apod
+Version: 0.0.1
+Summary: 定时发送 NASA 每日提供的天文图片
+Home-page: https://github.com/lyqgzbl/nonebot-plugin-apod
+License: MIT
+Keywords: nonebot2,NASA
+Author: lyqgzbl
+Author-email: admin@lyqgzbl.com
+Requires-Python: >=3.8,<4.0
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: nonebot-plugin-alconna (>=0.54.0,<0.55.0)
+Requires-Dist: nonebot-plugin-apscheduler (>=0.5.0,<0.6.0)
+Requires-Dist: nonebot-plugin-localstore (>=0.7.0,<0.8.0)
+Requires-Dist: nonebot-plugin-send-anything-anywhere (>=0.7.0,<0.8.0)
+Requires-Dist: nonebot2 (>=2.2.1,<3.0.0)
+Project-URL: Documentation, https://github.com/lyqgzbl/nonebot-plugin-apod#readme
+Project-URL: Repository, https://github.com/lyqgzbl/nonebot-plugin-apod
+Description-Content-Type: text/markdown
+
+# nonebot-plugin-apod

nonebot_plugin_apod-0.0.1/README.md

@@ -0,0 +1 @@
+# nonebot-plugin-apod

nonebot_plugin_apod-0.0.1/nonebot_plugin_apod/__init__.py

@@ -0,0 +1,133 @@
+import re
+import json
+
+from nonebot import require, get_plugin_config
+from nonebot.rule import Rule
+from nonebot.log import logger
+from nonebot.permission import SUPERUSER
+from nonebot.plugin import PluginMetadata, inherit_supported_adapters
+
+require("nonebot_plugin_saa")
+require("nonebot_plugin_alconna")
+require("nonebot_plugin_localstore")
+require("nonebot_plugin_apscheduler")
+import nonebot_plugin_localstore as store
+from nonebot_plugin_apscheduler import scheduler
+from nonebot_plugin_alconna import Args, Match, Option, Alconna, CommandMeta, on_alconna
+from nonebot_plugin_saa import SaaTarget, enable_auto_select_bot, PlatformTarget, get_target
+
+from .config import Config
+from .apod import send_apod, remove_apod_task, schedule_apod_task
+
+
+__plugin_meta__ = PluginMetadata(
+    name="每日天文一图",
+    description="定时发送 NASA 每日提供的天文图片",
+    usage="/apod 状态; /apod 关闭; /apod 开启 13:30",
+    type="application",
+    homepage="https://github.com/lyqgzbl/nonebot-plugin-apod",
+    config=Config,
+    supported_adapters=inherit_supported_adapters(
+		"nonebot_plugin_alconna", "nonebot_plugin_saa"
+    ),
+)
+
+
+enable_auto_select_bot()
+plugin_config = get_plugin_config(Config)
+if not plugin_config.nasa_api_key:
+    logger.opt(colors=True).warning("<yellow>缺失必要配置项 'nasa_api_key'，已禁用该插件</yellow>")
+def is_enable() -> Rule:
+    def _rule() -> bool:
+        return bool(plugin_config.nasa_api_key)
+    return Rule(_rule)
+
+
+apod = on_alconna(
+    Alconna(
+        "apod",
+        Option("状态|status"),
+        Option("关闭|stop"),
+        Option("开启|start", Args["send_time?#每日一图发送时间", str]),
+        meta=CommandMeta(
+            compact=True,
+            description="NASA 每日天文图片设置",
+            usage=__plugin_meta__.usage,
+            example=(
+                "/apod 状态\n"
+                "/apod 关闭\n"
+                "/apod 开启 13:30"
+            ),
+        ),
+    ),
+    rule=is_enable(),
+    aliases={"APOD"},
+    permission=SUPERUSER,
+    use_cmd_start=True,
+)
+
+
+def is_valid_time_format(time_str: str) -> bool:
+    if not re.match(r"^\d{1,2}:\d{2}$", time_str):
+        return False
+    try:
+        hour, minute = map(int, time_str.split(":"))
+        return 0 <= hour <= 23 and 0 <= minute <= 59
+    except ValueError:
+        return False
+
+
+@apod.assign("status")
+async def apod_status(event):
+    task_config_file = store.get_plugin_data_file("apod_task_config.json")
+    if not task_config_file.exists():
+        await apod.finish("NASA 每日天文一图定时任务未开启")
+    try:
+        with task_config_file.open("r", encoding="utf-8") as f:
+            config = json.load(f)
+        tasks = config.get("tasks", [])
+    except Exception as e:
+        await apod.finish(f"加载任务配置时发生错误：{e}")
+    if not tasks:
+        await apod.finish("NASA 每日天文一图定时任务未开启")
+    current_target = get_target(event)
+    for task in tasks:
+        target_data = task["target"]
+        target = PlatformTarget.deserialize(target_data)
+        if target == current_target:
+            send_time = task["send_time"]
+            job_id = f"send_apod_task_{target.dict()}"
+            job = scheduler.get_job(job_id)
+            if job:
+                next_run = (
+                    job.next_run_time.strftime("%Y-%m-%d %H:%M:%S")
+                    if job.next_run_time else "未知"
+                )
+                await apod.finish(f"NASA 每日天文一图定时任务已开启 | 下次发送时间: {next_run}")
+            else:
+                await apod.finish("NASA 每日天文一图定时任务未开启")
+    await apod.finish("NASA 每日天文一图定时任务未开启")
+
+
+@apod.assign("stop")
+async def apod_stop(target: SaaTarget):
+    remove_apod_task(target)
+    await apod.finish("已关闭 NASA 每日天文一图定时任务")
+
+
+@apod.assign("start")
+async def apod_start(send_time: Match[str], target: SaaTarget):
+    if send_time.available:
+        time = send_time.result
+        if not is_valid_time_format(time):
+            await apod.send("时间格式不正确,请使用 HH:MM 格式")
+        try:
+            schedule_apod_task(time, target)
+            await apod.send(f"已开启 NASA 每日天文一图定时任务,发送时间为 {time}")
+        except Exception as e:
+            logger.error(f"设置 NASA 每日天文一图定时任务时发生错误:{e}")
+            await apod.finish("设置 NASA 每日天文一图定时任务时发生错误")
+    else:
+        default_time = plugin_config.default_apod_send_time
+        schedule_apod_task(default_time, target)
+        await apod.finish(f"已开启 NASA 每日天文一图定时任务,默认发送时间为 {default_time}")

nonebot_plugin_apod-0.0.1/nonebot_plugin_apod/apod.py

@@ -0,0 +1,121 @@
+import datetime
+import httpx
+import json
+import nonebot_plugin_localstore as store
+
+from nonebot import get_plugin_config, get_bot
+from nonebot.log import logger
+from nonebot_plugin_apscheduler import scheduler
+from nonebot_plugin_saa import Text, Image, PlatformTarget
+from .config import Config
+
+plugin_config = get_plugin_config(Config)
+NASA_API_URL = "https://api.nasa.gov/planetary/apod"
+NASA_API_KEY = plugin_config.nasa_api_key
+task_config_file = store.get_plugin_data_file("apod_task_config.json")
+
+
+def save_task_configs(tasks: list):
+    try:
+        serialized_tasks = [
+            {"send_time": task["send_time"], "target": task["target"].dict()} for task in tasks
+        ]
+        with task_config_file.open("w", encoding="utf-8") as f:
+            json.dump({"tasks": serialized_tasks}, f, ensure_ascii=False, indent=4)
+        logger.info("NASA 每日天文一图定时任务配置已保存")
+    except Exception as e:
+        logger.error(f"保存 NASA 每日天文一图定时任务配置时发生错误：{e}")
+
+
+def load_task_configs():
+    if not task_config_file.exists():
+        return []
+    try:
+        with task_config_file.open("r", encoding="utf-8") as f:
+            config = json.load(f)
+        tasks = [
+            {"send_time": task["send_time"], "target": PlatformTarget.deserialize(task["target"])}
+            for task in config.get("tasks", [])
+        ]
+        return tasks
+    except Exception as e:
+        logger.error(f"加载 NASA 每日天文一图定时任务配置时发生错误：{e}")
+        return []
+
+
+async def fetch_apod_data():
+    try:
+        async with httpx.AsyncClient() as client:
+            response = await client.get(NASA_API_URL, params={"api_key": NASA_API_KEY})
+            response.raise_for_status()
+            return response.json()
+    except httpx.RequestError as e:
+        logger.error(f"获取 NASA 每日天文一图数据时发生错误: {e}")
+        return None
+
+
+async def send_apod(target: PlatformTarget):
+    apod_data = await fetch_apod_data()
+    if apod_data:
+        title = apod_data.get("title", "NASA APOD")
+        url = apod_data.get("url")
+        try:
+            await Image(url).send_to(target, bot=get_bot())
+            await Text(f"链接：{url}").send_to(target, bot=get_bot())
+        except Exception as e:
+            logger.error(f"发送 NASA 每日天文一图时发生错误：{e}")
+            await Text("发送 NASA 每日天文一图时发生错误").send_to(target, bot=get_bot())
+    else:
+        logger.error("无法获取今天的天文图片")
+        await Text("无法获取今天的天文图片。").send_to(target, bot=get_bot())
+
+
+def schedule_apod_task(send_time: str, target: PlatformTarget):
+    try:
+        hour, minute = map(int, send_time.split(":"))
+        job_id = f"send_apod_task_{target.dict()}"
+        scheduler.add_job(
+            func=send_apod,
+            trigger="cron",
+            args=[target],
+            hour=hour,
+            minute=minute,
+            id=job_id,
+            max_instances=1,
+            replace_existing=True,
+        )
+        logger.info(f"已成功设置 NASA 每日天文一图定时任务，发送时间为 {send_time} (目标: {target})")
+        tasks = load_task_configs()
+        tasks = [task for task in tasks if task["target"] != target]
+        tasks.append({"send_time": send_time, "target": target})
+        save_task_configs(tasks)
+    except ValueError:
+        logger.error(f"时间格式错误：{send_time}，请使用 HH:MM 格式")
+        raise ValueError(f"时间格式错误：{send_time}")
+    except Exception as e:
+        logger.error(f"设置 NASA 每日天文一图定时任务时发生错误：{e}")
+
+
+def remove_apod_task(target: PlatformTarget):
+    job_id = f"send_apod_task_{target.dict()}"
+    job = scheduler.get_job(job_id)
+    if job:
+        job.remove()
+        logger.info(f"已移除 NASA 每日天文一图定时任务 (目标: {target})")
+        tasks = load_task_configs()
+        tasks = [task for task in tasks if task["target"] != target]
+        save_task_configs(tasks)
+    else:
+        logger.info(f"未找到 NASA 每日天文一图定时任务 (目标: {target})")
+
+
+try:
+    tasks = load_task_configs()
+    for task in tasks:
+        send_time = task["send_time"]
+        target = task["target"]
+        if send_time and target:
+            schedule_apod_task(send_time, target)
+    logger.debug("已恢复所有 NASA 每日天文一图定时任务")
+except Exception as e:
+    logger.error(f"恢复 NASA 每日天文一图定时任务时发生错误：{e}")

nonebot_plugin_apod-0.0.1/nonebot_plugin_apod/config.py

@@ -0,0 +1,7 @@
+from typing import Optional
+from pydantic import BaseModel
+
+
+class Config(BaseModel):
+    nasa_api_key: Optional[str] = None
+    default_apod_send_time: str = "13:00"

nonebot_plugin_apod-0.0.1/pyproject.toml

@@ -0,0 +1,23 @@
+[tool.poetry]
+name = "nonebot-plugin-apod"
+version = "0.0.1"
+description = "定时发送 NASA 每日提供的天文图片"
+readme = "README.md"
+authors = ["lyqgzbl <admin@lyqgzbl.com>"]
+license = "MIT"
+homepage = "https://github.com/lyqgzbl/nonebot-plugin-apod"
+repository = "https://github.com/lyqgzbl/nonebot-plugin-apod"
+documentation = "https://github.com/lyqgzbl/nonebot-plugin-apod#readme"
+keywords = ["nonebot2", "NASA"]
+
+[tool.poetry.dependencies]
+python = "^3.8"
+nonebot2 = "^2.2.1"
+nonebot-plugin-alconna = "^0.54.0"
+nonebot-plugin-localstore = "^0.7.0"
+nonebot-plugin-apscheduler = "^0.5.0"
+nonebot-plugin-send-anything-anywhere = "^0.7.0"
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"