feat: 增加 ff 登录逻辑 TUN-41

mas2ff2.py

@@ -7,6 +7,7 @@ from datetime import datetime
 from loguru import logger
 import os
 import sentry_sdk
+from bs4 import BeautifulSoup
 
 sentry_sdk.init(
     dsn="https://c42de79f364a12228f184bc54fec4dfd@o4506942768021504.ingest.us.sentry.io/4506982441615360",
@@ -59,9 +60,10 @@ def post_fanfou(cookie, token, fanfou_status, images=None):
                   response.status_code))
       if not '你在做什么？' in response.text:
         logger.error('Post %s maybe failed', fanfou_status)
+        raise requests.exceptions.RequestException
     except requests.exceptions.RequestException:
         logger.error('Post %s failed', fanfou_status)
-
+        raise requests.exceptions.RequestException
   else:
     # post with images
     image_urls = [media['url'] for media in images if media['type'] == 'image']
@@ -124,6 +126,7 @@ def post_fanfou(cookie, token, fanfou_status, images=None):
                                   response.status_code))
                   except requests.exceptions.RequestException:
                       logger.error('Post %s failed', status)
+                      sentry_sdk.capture_exception(requests.exceptions.RequestException)
       except:
         continue
 
@@ -178,10 +181,65 @@ def run(instance, access_token, fanfou_cookie, fanfou_token):
             f.write(str(min_id) + '\n')
         min_id = None
 
+def login_to_fanfou(email, password):
+  url = 'https://m.fanfou.com/home'
+  resp = requests.get(url)
+  soup = BeautifulSoup(resp.text, 'html.parser')
+  try:
+    token = soup.find('input', {'name': 'token'}).get('value')
+  except:
+    token = None
+  if not token:
+    logger.error('Get token failed')
+    return False
+  data = {
+      "loginname": email,
+      "loginpass": password,
+      "auto_login": "on",
+      "action": "login",
+      "token": token,
+  }
+  headers = {
+      "Content-Type": "application/x-www-form-urlencoded",
+      "Origin": "https://m.fanfou.com",
+      "Referer": "https://m.fanfou.com/home",
+      "Sec-Fetch-Dest": "document",
+      "Sec-Fetch-Mode": "navigate",
+      "Sec-Fetch-Site": "same-origin",
+      "Sec-Fetch-User": "?1",
+      "Upgrade-Insecure-Requests": "1",
+      "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/118.0.0.0 Safari/537.36",
+  }
+  resp = requests.post(url, data=data, headers=headers)
+  soup = BeautifulSoup(resp.text, 'html.parser')
+  if not '你在做什么？' in soup.text:
+    return False
+  cookie = resp.headers['Set-Cookie']
+  token = None
+  if soup.find('input', {'name': 'token'}):
+    token = soup.find('input', {'name': 'token'}).get('value')
+
+  with open(file_path + '/cookie.txt', 'w') as f:
+    f.write(cookie)
+  with open(file_path + '/token.txt', 'w') as f:
+    f.write(token)
+  return cookie, token
+
 if __name__ == '__main__':
   cookie = 'u=stokesia; uuid=24edfe42401cbb03cf1f.1676446846.21; PHPSESSID=d4l9m3tp09gasff6r3u8mv9p12; m=yvettezhong%40gmail.com; tgw_l7_route=f174d6f255742a6ee2e9a07cf9cca0fa'
   token = '742ac481'
+  with open(file_path + '/cookie.txt', 'r') as f:
+    cookie = f.read()
+  with open(file_path + '/token.txt', 'r') as f:
+    token = f.read()
   m_ins = 'https://nofan.xyz'
   m_token = 'C5zKq2Mf54E7gaSDe47Bds080ySpsdvaykZBAtt5jvo'
-  run(m_ins, m_token, cookie, token)
+  email = ''
+  password = ''
+  try:
+    run(m_ins, m_token, cookie, token)
+  except Exception as e:
+    logger.error(e)
+    sentry_sdk.capture_exception(e)
+    login_to_fanfou(email, password)