Merge pull request #9 from ElieenAndBella/main

添加邮箱支持

Author: gwtak

.github/workflows/main.yml

@@ -22,4 +22,8 @@ jobs:
       - name: 'Start Sign'
         env:
           BDUSS: ${{ secrets.BDUSS }}
+          HOST: ${{ secrets.HOST }}
+          FROM: ${{ secrets.FROM }}
+          TO: ${{ secrets.TO }}
+          AUTH: ${{ secrets.AUTH }}
         run: python main.py

README.md

@@ -8,24 +8,15 @@
 
 1. Fork 本仓库，然后点击你的仓库右上角的 Settings，找到 Secrets 这一项，添加一个库秘密变量。其中 `BDUSS` 存放你的 BDUSS。支持同时添加多个帐户，BDUSS 之间用 `#` 隔开即可。
 
-![添加库秘密变量](/img/new_repository_secret.png)
-![添加BDUSS](/img/add_BDUSS.png)
-
 2. 设置好环境变量后点击你的仓库上方的 `Actions` 选项，第一次打开需要点击 `I understand...` 按钮，确认在 Fork 的仓库上启用 GitHub Actions 。
 
 3. 任意发起一次commit，可以参考下图流程修改readme文件。
 
 - 打开`README.md`，点击修改按钮
-
-![修改README](/img/edit.png)
-
 - 修改任意内容，这里在末尾插入了空格。移动到最下面，点击提交。
 
-![添加空行](/img/update.png)
-
 4. 至此自动签到就搭建完毕了，可以再次点击`Actions`查看工作记录，如果有`Baidu Tieba Auto Sign`则说明workflow创建成功了。点击右侧记录可以查看详细签到情况。
 
-![查看Action](/img/check.png)
 
 
 

main.py

@@ -7,6 +7,10 @@
 import logging
 import random
 
+import smtplib
+from email.mime.text import MIMEText
+
+
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
@@ -15,6 +19,12 @@
 TBS_URL = "http://tieba.baidu.com/dc/common/tbs"
 SIGN_URL = "http://c.tieba.baidu.com/c/c/forum/sign"
 
+# EMAIL
+HOST = os.environ['HOST']
+FROM = os.environ['FROM']
+TO = os.environ['TO'].split('#')
+AUTH = os.environ['AUTH']
+
 HEADERS = {
     'Host': 'tieba.baidu.com',
     'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.71 Safari/537.36',
@@ -167,6 +177,39 @@ def client_sign(bduss, tbs, fid, kw):
     res = s.post(url=SIGN_URL, data=data, timeout=5).json()
     return res
 
+def send_email(sign_list):
+    if HOST is None or FROM is None or TO is None or AUTH is None:
+        logger.info("未配置邮箱")
+        return
+    length = len(sign_list)
+    subject = f"{time.strftime('%Y-%m-%d', time.localtime())} 签到{length}个贴吧"
+    body = """
+    <style>
+    .child {
+      background-color: rgba(173, 216, 230, 0.19);
+      padding: 10px;
+    }
+
+    .child * {
+      margin: 5px;
+    }
+    </style>
+    """
+    for i in sign_list:
+        body += f"""
+        <div class="child">
+            <div class="name"> 贴吧名称: { i['name'] }</div>
+            <div class="slogan"> 贴吧简介: { i['slogan'] }</div>
+        </div>
+        <hr>
+        """
+    msg = MIMEText(body, 'html', 'utf-8')
+    msg['subject'] = subject
+    smtp = smtplib.SMTP()
+    smtp.connect(HOST)
+    smtp.login(FROM, AUTH)
+    smtp.sendmail(FROM, TO, msg.as_string())
+    smtp.quit()
 
 def main():
     b = os.environ['BDUSS'].split('#')
@@ -181,6 +224,7 @@ def main():
             time.sleep(random.randint(1,5))
             client_sign(i, tbs, j["id"], j["name"])
         logger.info("完成第" + str(n) + "个用户签到")
+    send_email(favorites)
     logger.info("所有用户签到结束")