update

src/ark_helper_backend/backend.py

@@ -26,9 +26,9 @@ class ArkHelperBackend:
         self.user_config_dict = self.load_users()
         self.user_data_dict = self.initiate_user_data()
 
-        # # 更新用户token
-        # self.refresh_user_token()
-
+        # 更新用户token
+        self.refresh_user_token()
+        self.is_retrieving_data = True
         self.logger.info('ArkHelperBackend initiated')
 
     def load_users(self):
@@ -184,10 +184,21 @@ class ArkHelperBackend:
             pay_dict = {item['orderId']: item for item in pay_data}
             payment_data.update(pay_dict)
             self.user_data_dict[phone]['payment'] = payment_data
+            self.logger.info(f'{phone} payment data updated')
         self.save_data(token_status=False, gacha_status=False)
 
+    def data_retrieve(self):
+        self.refresh_user_token()
+        self.update_payment_data()
+        self.update_gacha_data()
+
+    def show_data(self):
+        gacha_data = {user: content['gacha'] for user, content in self.user_data_dict.items()}
+        payment_data = {user: content['payment'] for user, content in self.user_data_dict.items()}
+        return gacha_data, payment_data
+
 
 if __name__ == '__main__':
     ak_back = ArkHelperBackend()
-    ak_back.update_payment_data()
-
+    # ak_back.update_payment_data()
+    ak_back.skland_sign()

src/util/mail.py

@@ -0,0 +1,75 @@
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from email.mime.base import MIMEBase
+from email import encoders
+
+
+def send_batch_email(receiver_email_list, subject, content, file_path=None):
+    """
+    Send Email to the receiver list. Can attach files if needed.
+    :param receiver_email_list:
+    :param subject: Title
+    :param content: Word Content
+    :param file_path: file list
+    :return:
+    """
+    # sender config
+    sender_email = "ArmorServer@outlook.com"
+    password = "ServerOnly2024!"
+
+    # mail content
+    message = MIMEMultipart()
+    message["Subject"] = subject
+    message["From"] = sender_email
+
+    # main body
+    part1 = MIMEText(content, "plain")
+    message.attach(part1)
+
+    if file_path and isinstance(file_path, list):
+        for filename in file_path:
+            # 指定附件的路径和文件名
+            attachment = open(filename, 'rb')
+
+            # 创建MIMEBase实例
+            part = MIMEBase('application', 'octet-stream')
+            part.set_payload(attachment.read())
+            encoders.encode_base64(part)
+
+            # 添加邮件头
+            part.add_header(
+                'Content-Disposition',
+                f'attachment; filename= {filename}',
+            )
+            message.attach(part)
+            attachment.close()
+
+    # 登录SMTP服务器并发送邮件
+    # 使用Hotmail的SMTP服务器
+    server = smtplib.SMTP('smtp.office365.com', 587)
+    try:
+        server.starttls()  # 启用安全传输模式
+        server.login(sender_email, password)
+
+        for receiver_email in receiver_email_list:
+            message["To"] = receiver_email
+            server.sendmail(sender_email, receiver_email, message.as_string())
+            print(f"Email to {receiver_email} sent successfully!")
+    except Exception as e:
+        print(f"Error: {e}")
+    finally:
+        server.quit()
+
+
+if __name__ == '__main__':
+    target_email_list = ['2654988228@qq.com']
+    email_subject = 'Test Message'
+    email_body = """
+    This is a test email 
+    """
+
+    email_file_list = []
+
+    send_batch_email(target_email_list, email_subject, email_body, email_file_list)
+