2 files changed, 36 insertions, 4 deletions
diff --git a/ATRI/plugins/rich/__init__.py b/ATRI/plugins/rich/__init__.py
index d20a234..7f9369e 100644
--- a/ATRI/plugins/rich/__init__.py
+++ b/ATRI/plugins/rich/__init__.py
@@ -1,3 +1,19 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+'''
+File: __init__.py
+Created Date: 2021-02-07 09:34:23
+Author: Kyomotoi
+Email: [email protected]
+License: GPLv3
+Project: https://github.com/Kyomotoi/ATRI
+--------
+Last Modified: Sunday, 7th March 2021 3:14:28 pm
+Modified By: Kyomotoi ([email protected])
+--------
+Copyright (c) 2021 Kyomotoi
+'''
+
 import re
 import json
 from aiohttp.client import ClientSession
@@ -25,7 +41,7 @@ bilibili_rich = on_message(
 @bilibili_rich.handle()
 async def _bilibili_rich(bot: Bot, event: MessageEvent) -> None:
     global waiting_list
-    msg = str(event.raw_message)
+    msg = str(event.raw_message).replace("\\", "")
     user = event.user_id
     bv = False
     
@@ -45,9 +61,9 @@ async def _bilibili_rich(bot: Bot, event: MessageEvent) -> None:
             except:
                 return
     else:
-        bv_url = re.findall(r"(..........b23...\S+\=)", msg)
-        bv_url = bv_url[0].replace("\\", "")
-        print(bv_url)
+        patt = r"(?:(?:https?|ftp):\/\/)?[\w/\-?=%.]+\.[\w/\-&?=%.]+"
+        bv_url = re.findall(patt, msg)
+        bv_url = bv_url[3]
         async with ClientSession() as session:
             async with session.get(
                 url=bv_url) as r:
diff --git a/ATRI/plugins/rich/data_source.py b/ATRI/plugins/rich/data_source.py
index 32ac219..d3e7d6e 100644
--- a/ATRI/plugins/rich/data_source.py
+++ b/ATRI/plugins/rich/data_source.py
@@ -1,3 +1,19 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+'''
+File: data_source.py
+Created Date: 2021-02-07 09:34:31
+Author: Kyomotoi
+Email: [email protected]
+License: GPLv3
+Project: https://github.com/Kyomotoi/ATRI
+--------
+Last Modified: Sunday, 7th March 2021 3:14:31 pm
+Modified By: Kyomotoi ([email protected])
+--------
+Copyright (c) 2021 Kyomotoi
+'''
+
 table = 'fZodR9XQDSUm21yCkr6zBqiveYah8bt4xsWpHnJE7jL5VG3guMTKNPAwcF'
 tr = {}
 for i in range(58):