Optimized code

main.py

@@ -10,6 +10,7 @@ from pushplus import send_message
 url = os.environ.get("URL")
 username = os.environ.get("USERNAME")
 password = os.environ.get("PASSWORD")
+token = os.environ.get("TOKEN")
 
 
 # 定义一个md5加密的封装函数

pushplus.py

@@ -1,10 +1,8 @@
 import requests
 import json
-import os
 
 
-def send_message(title, content):
+def send_message(token, title, content):
-    token = os.environ.get("TOKEN")
     url = "http://www.pushplus.plus/send"
     data = {"token": token, "title": title, "content": content}
     body = json.dumps(data).encode(encoding="utf-8")