From dfa97c9d24b8124ca38ec5eb605cde932cd6ea78 Mon Sep 17 00:00:00 2001
From: 简律纯 <hsiangnianian@outlook.com>
Date: Tue, 2 May 2023 23:30:19 +0800
Subject:

---
 tests/plugins/test.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 tests/plugins/test.py

(limited to 'tests/plugins/test.py')

diff --git a/tests/plugins/test.py b/tests/plugins/test.py
new file mode 100644
index 0000000..6f3bc3b
--- /dev/null
+++ b/tests/plugins/test.py
@@ -0,0 +1,35 @@
+from iamai import Plugin
+from iamai.exceptions import GetEventTimeout
+
+
+class Weather(Plugin):
+    async def handle(self) -> None:
+        args = self.event.get_plain_text().split(" ")
+        if len(args) >= 2:
+            await self.event.reply(await self.get_weather(args[1]))
+        else:
+            await self.event.reply("请输入想要查询天气的城市：")
+            try:
+                city_event = await self.event.adapter.get(
+                    lambda x: x.type == "message", timeout=10
+                )
+            except GetEventTimeout:
+                return
+            else:
+                await self.event.reply(
+                    await self.get_weather(city_event.get_plain_text())
+                )
+
+    async def rule(self) -> bool:
+        if self.event.adapter.name != "cqhttp":
+            return False
+        if self.event.type != "message":
+            return False
+        return self.event.message.startswith("天气")
+
+    @staticmethod
+    async def get_weather(city):
+        if city not in ["北京", "上海"]:
+            return "你想查询的城市暂不支持！"
+        return f"{city}的天气是..."
+
-- 
cgit v1.2.3-70-g09d2