Update app.py

app.py

@@ -1,6 +1,4 @@
 import os
-import re
-import json
 import requests
 from flask import Flask, request
 from simpleeval import simple_eval
@@ -44,22 +42,37 @@ def api():
 
                 matrix = [[0 for _ in range(len(users))] for _ in range(len(users))]
 
+                should_checkout = True
+
                 for msg_id, data in message_list.items():
                     quoted_msg_id = data.get("quoted_msg_id")
-                    quoted_number = len([key for key, value in message_list.items() if value.get("quoted_msg_id") == msg_id])
+                    quoted_msg_list = {key: value for key, value in message_list.items() if value.get("quoted_msg_id") == msg_id}
+
+                    if not quoted_msg_id and len(quoted_msg_list) != 0:  # 檢查是否付清
+                        amount: float = data.get("amount")
+                        paid: float = 0.0
+                        for _, value in quoted_msg_list.items():
+                            paid += value.get("amount")
+                        if amount-paid > 1:
+                            send_text(payload.get("token"), f"$ {amount-paid} 未付清", data.get("quote_token"))
+                            should_checkout = False
+                            break
+
+                    if not quoted_msg_id and len(quoted_msg_list) == 0:  # 要平分的情況
+                        to = users.index(data.get("user_id"))
+                        for row in matrix:
+                            row[to] += (data.get("amount") / users_number)
 
                     if quoted_msg_id:
                         fr = users.index(data.get("user_id"))
                         to = users.index(message_list.get(quoted_msg_id).get("user_id"))
                         matrix[fr][to] += data.get("amount")
 
-                    if not quoted_msg_id and quoted_number == 0:  # 要平分的情況
-                        to = users.index(data.get("user_id"))
-                        for row in matrix:
-                            row[to] += (data.get("amount") / users_number)
-
                 print(matrix)
 
+                if not should_checkout:
+                    return
+
                 matrix_copy = [[0 for _ in range(len(matrix))] for _ in range(len(matrix))]
 
                 for i in range(len(matrix_copy)):
@@ -70,19 +83,25 @@ def api():
                 for r in matrix_copy:
                     print(r)
 
-                l = []
+                result = []
 
                 for i in range(len(matrix_copy)):
                     for j in range(len(matrix_copy)):
                         amount = matrix_copy[i][j]
                         if amount > 0:
-                            l.append({"from": get_username(users[i]), "to": get_username(users[j]), "amount": amount})
+                            result.append({"from": get_username(users[i]), "to": get_username(users[j]), "amount": amount})
                         if amount < 0:
-                            l.append({"from": get_username(users[j]), "to": get_username(users[i]), "amount": -amount})
+                            result.append({"from": get_username(users[j]), "to": get_username(users[i]), "amount": -amount})
+
+                sorted_result = sorted(result, key=lambda x: x["from"])
 
-                for r in l:
+                for r in sorted_result:
                     print(r)
 
+                bubble = get_checkout_bubble(sorted_result)
+                print(bubble)
+                send_flex_text(payload.get("token"), bubble)
+
             raise ValueError("Action Success")
 
         if "$" not in payload.get("msg_text"):
@@ -123,20 +142,6 @@ def api():
                     send_text(payload.get("token"), f"$ {amount-paid} 未付清", data.get("quote_token"))
                     break
 
-            # if quoted_msg_id:
-            #     fr = users.index(data.get("user_id"))
-            #     to = users.index(message_list.get(quoted_msg_id).get("user_id"))
-            #     matrix[fr][to] += data.get("amount")
-
-            # if not quoted_msg_id and quoted_number == 0:  # 要平分的情況
-            #     to = users.index(data.get("user_id"))
-            #     for row in matrix:
-            #         row[to] += (data.get("amount") / users_number)
-
-        # username = get_username(payload)
-        # text = payload.get("msg_text")
-        # print(username)
-        # send_text(payload.get("token"), f"{username}:{text}")
     except Exception as e:
         print(f"An error occurred: {e}")
 
@@ -184,12 +189,14 @@ def send_flex_text(token, bubble):
 
 
 def get_username(user_id: str):
+    if user_id == "others":
+        return "其他人"
     url = f"https://api.line.me/v2/bot/group/{GROUP_ID}/member/{user_id}"
     try:
         res_json = requests.get(url, headers={"Authorization": "Bearer " + LINE_CHANNEL_ACCESS_TOKEN}).json()
         return res_json.get("displayName")
     except:
-        return "Unknow"
+        return "未知"
 
 
 def get_users_number() -> int:
@@ -216,12 +223,12 @@ def get_amount(text: str) -> float | None:
 
 def get_summary_bubble(title: str, payer: str, borrowers: list):
     total = 0
-    borrower_list = []
+    borrower_contents = []
 
     for borrower in borrowers:
         amount = float(borrower.get("amount"))
         total += amount
-        borrower_list.append({
+        borrower_contents.append({
             "type": "box",
             "layout": "horizontal",
             "contents": [
@@ -272,14 +279,14 @@ def get_summary_bubble(title: str, payer: str, borrowers: list):
                 },
                 {
                     "type": "separator",
-                    "margin": "xl"
+                    "margin": "lg"
                 },
                 {
                     "type": "box",
                     "layout": "vertical",
-                    "margin": "xl",
+                    "margin": "lg",
                     "spacing": "sm",
-                    "contents": borrower_list
+                    "contents": borrower_contents
                 },
                 {
                     "type": "separator",
@@ -297,7 +304,7 @@ def get_summary_bubble(title: str, payer: str, borrowers: list):
                         },
                         {
                             "type": "text",
-                            "text": f"${total}",
+                            "text": f"${round(total,2)}",
                             "size": "sm",
                             "color": "#111111",
                             "align": "end",
@@ -316,5 +323,118 @@ def get_summary_bubble(title: str, payer: str, borrowers: list):
     }
 
 
+def get_checkout_bubble(checkout_list: list):
+
+    current_user = checkout_list[0].get("from")
+    current_total = 0
+    current_pay_to_contents = []
+    carousel_contents = []
+
+    for item in checkout_list:
+
+        if current_user != item.get("from"):
+            carousel_contents.append({
+                "type": "bubble",
+                "body": {
+                    "type": "box",
+                    "layout": "vertical",
+                    "contents": [
+                        {
+                            "type": "text",
+                            "text": "結算明細",
+                            "weight": "bold",
+                            "color": "#1DB446",
+                            "size": "sm"
+                        },
+                        {
+                            "type": "text",
+                            "text": current_user,
+                            "weight": "bold",
+                            "size": "xxl",
+                            "margin": "md"
+                        },
+                        {
+                            "type": "text",
+                            "text": "需付款給：",
+                            "size": "sm",
+                            "color": "#aaaaaa",
+                            "wrap": True,
+                            "margin": "lg"
+                        },
+                        {
+                            "type": "box",
+                            "layout": "vertical",
+                            "margin": "md",
+                            "spacing": "sm",
+                            "contents": current_pay_to_contents},
+                        {
+                            "type": "separator",
+                            "margin": "lg"
+                        },
+                        {
+                            "type": "box",
+                            "layout": "horizontal",
+                            "contents": [
+                                {
+                                    "type": "text",
+                                    "text": "合計",
+                                    "size": "sm",
+                                    "color": "#555555"
+                                },
+                                {
+                                    "type": "text",
+                                    "text": f"${round(current_total,2)}",
+                                    "size": "sm",
+                                    "color": "#111111",
+                                    "align": "end",
+                                    "weight": "bold"
+                                }
+                            ],
+                            "margin": "lg"
+                        }
+                    ]
+                },
+                "styles": {
+                    "footer": {
+                        "separator": True
+                    }
+                }
+            }
+            )
+
+            current_user = item.get("from")
+            current_total = 0
+            current_pay_to_contents = []
+
+        amount = float(item.get("amount"))
+        current_total += amount
+
+        current_pay_to_contents.append({
+            "type": "box",
+            "layout": "horizontal",
+            "contents": [
+                {
+                    "type": "text",
+                    "text": item.get("to"),
+                    "size": "sm",
+                    "color": "#555555",
+                    "flex": 0
+                },
+                {
+                    "type": "text",
+                    "text": f"${round(amount,2)}",
+                    "size": "sm",
+                    "color": "#111111",
+                    "align": "end"
+                }
+            ]
+        },)
+
+    return {
+        "type": "carousel",
+        "contents": carousel_contents
+    }
+
+
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)