.env.bak

@@ -1,6 +0,0 @@
- BOT_ACCOUNT="Your-bot-account"
- ROOT_ACCOUNT="Your-root-account"
- WS_URI="ws://Your-uri:port"
- WS_TOKEN="Your-token"
- WEBUI_URI="http://Your-uri:port"
- WEBUI_TOKEN="Your-token

.gitignore

@@ -1,4 +1 @@
-tests
+tests
-.env
-src/data/pics
-src/logs

requirements.txt

@@ -1,4 +0,0 @@
-requests~=2.32.4
-ncatbot~=3.8.8.post7
-dotenv~=0.9.9
-python-dotenv~=1.1.1

src/main.py

@@ -0,0 +1,92 @@
+from ncatbot.core import BotClient, GroupMessage, PrivateMessage
+# from ncatbot.core import (MessageChain, Text, Image)
+from ncatbot.utils import get_log
+import requests
+
+bot = BotClient()
+_log = get_log()
+
+@bot.private_event()
+async def on_receive_images_private_event(msg: PrivateMessage):
+    _log.info(msg)
+
+    if "url=https://" not in msg.raw_message:
+        return
+
+    else:
+        try:
+            file_name = msg.raw_message.split(",")[1].split("=")[1]
+            image_url = await bot.api.get_image(file_name)
+            real_url = image_url['data']['url']
+        except IndexError as e:
+            _log.debug(e)
+            real_url = "None"
+            file_name = "None"
+        except TypeError as e:
+            _log.debug(e)
+            real_url = "None"
+            file_name = "None"
+
+        if real_url == "None":
+            return
+        else:
+            response = requests.get(real_url)
+            img = response.content
+            with open(f"data/pics/{file_name}", 'wb') as f:
+                f.write(img)
+
+    # file_name = msg.raw_message.split(",")[1].split("=")[1]
+    # image_url = await bot.api.get_image(file_name)
+    # real_url = image_url['data']['url']
+    #
+    # _log.debug(f"image's real url: {real_url}")
+    #
+    # try:
+    #     response = requests.get(real_url)
+    #     img = response.content
+    #     with open(f"data/pics/{file_name}", 'wb') as f:
+    #         f.write(img)
+    # except Exception as e:
+    #     _log.error(e)
+    #     pass
+
+@bot.group_event()
+async def on_receive_images_group_event(msg: GroupMessage):
+    _log.info(msg)
+
+    if "url=https://" not in msg.raw_message:
+        return
+
+    else:
+        try:
+            file_name = msg.raw_message.split(",")[1].split("=")[1]
+            image_url = await bot.api.get_image(file_name)
+            real_url = image_url['data']['url']
+        except IndexError as e:
+            _log.debug(e)
+            real_url = "None"
+            file_name = "None"
+        except TypeError as e:
+            _log.debug(e)
+            real_url = "None"
+            file_name = "None"
+
+
+        if real_url == "None":
+            return
+        else:
+            response = requests.get(real_url)
+            img = response.content
+            with open(f"data/pics/{file_name}", 'wb') as f:
+                f.write(img)
+
+if __name__ == "__main__":
+
+    api = bot.run(
+        bt_uin="203758258",
+        root="1670671958",
+        ws_uri="ws://192.168.10.78:3001",
+        ws_token="",
+        webui_uri="http://192.168.10.78:6099",
+        webui_token="@Jtbx2mtblj"
+    )

src/utils/get_image.py

@@ -1,77 +0,0 @@
-from ncatbot.core import BotClient, GroupMessage, PrivateMessage
-# from ncatbot.core import (MessageChain, Text, Image)
-from ncatbot.utils import get_log
-from dotenv import load_dotenv
-import requests
-import os
-
-bot = BotClient()
-_log = get_log()
-
-
-@bot.private_event()
-async def on_receive_images_private_event(msg: PrivateMessage):
-    meta_message = str(msg)
-    _log.info(f"The message's meta data is: {meta_message}")
-    _log.info(f"The message's raw message is:  {msg.raw_message}")
-
-    if "[CQ:image," in msg.raw_message:
-        divided_message = msg.raw_message.split(',')
-
-        for item in divided_message:
-            if "file=" in item:
-                file_name = item.split("=")[1]
-                _log.info(f"Name of the image is: {file_name}")
-                image_url = await bot.api.get_image(file_name)
-                _log.info(f"Image's information collected by \"get_image\" is: {image_url}")
-                real_url = image_url['data']['url']
-                _log.info(f"Image's real url is: {real_url}")
-
-                response = requests.get(real_url)
-                img = response.content
-                with open(f"D:/Projects/memebot/src/data/pics/{file_name}", 'wb') as f:
-                    f.write(img)
-
-
-@bot.group_event()
-async def on_receive_images_group_event(msg: GroupMessage):
-    meta_message = str(msg)
-    _log.info(f"The message's meta data is: {meta_message}")
-    _log.info(f"The message's raw message is:  {msg.raw_message}")
-
-    # if "[CQ:image," in msg.raw_message:
-    if msg.raw_message.startswith("[CQ:image,"):
-        divided_message = msg.raw_message.split(',')
-
-        for item in divided_message:
-            # if "file=" in item:
-            if item.startswith("file="):
-                file_name = item.split("=")[1]
-                _log.info(f"Name of the image is: {file_name}")
-                image_url = await bot.api.get_image(file_name)
-                _log.info(f"Image's information collected by \"get_image\" is: {image_url}")
-                real_url = image_url['data']['url']
-                _log.info(f"Image's real url is: {real_url}")
-
-                response = requests.get(real_url)
-                img = response.content
-                with open(f"../../data/pics/{file_name}", 'wb') as f:
-                    f.write(img)
-
-
-"""
-当单独运行这个程序时，会连接napcat并保存图片到本地。
-
-"""
-
-if __name__ == "__main__":
-    load_dotenv(dotenv_path=r"../../.env")
-
-    api = bot.run(
-        bt_uin=os.getenv("BOT_ACCOUNT"),
-        root=os.getenv("ROOT_ACCOUNT"),
-        ws_uri=os.getenv("WS_URI"),
-        ws_token=os.getenv("WS_TOKEN"),
-        webui_uri=os.getenv("WEBUI_URI"),
-        webui_token=os.getenv("WEBUI_TOKEN")
-    )