mirror of
https://github.com/henry4682/linebot_finance.git
synced 2026-05-16 04:41:52 +00:00
feat: linbot
restruct the project
This commit is contained in:
59
app/DB/Models.py
Normal file
59
app/DB/Models.py
Normal file
@@ -0,0 +1,59 @@
|
||||
from sqlalchemy import Column, String, Numeric, Date, DateTime, BigInteger
|
||||
from sqlalchemy.orm import declarative_base
|
||||
from datetime import datetime
|
||||
|
||||
Base = declarative_base()
|
||||
|
||||
|
||||
class User(Base):
|
||||
__tablename__ = "users"
|
||||
id = Column(BigInteger, primary_key=True)
|
||||
name = Column(String)
|
||||
email = Column(String, unique=True)
|
||||
created_at = Column(DateTime, default=datetime.now)
|
||||
updated_at = Column(DateTime, default=datetime.now)
|
||||
|
||||
|
||||
class LineUser(Base):
|
||||
__tablename__ = "line_users"
|
||||
id = Column(BigInteger, primary_key=True)
|
||||
line_user_id = Column(String, unique=True)
|
||||
user_id = Column(BigInteger)
|
||||
created_at = Column(DateTime, default=datetime.now)
|
||||
|
||||
|
||||
class Category(Base):
|
||||
__tablename__ = "categories"
|
||||
id = Column(BigInteger, primary_key=True)
|
||||
user_id = Column(BigInteger)
|
||||
parent_id = Column(BigInteger, nullable=True)
|
||||
name = Column(String)
|
||||
color = Column(String, default='#3B82F6')
|
||||
icon = Column(String, nullable=True)
|
||||
created_at = Column(DateTime, default=datetime.now)
|
||||
updated_at = Column(DateTime, default=datetime.now)
|
||||
|
||||
|
||||
class CategoryRule(Base):
|
||||
__tablename__ = "category_rules"
|
||||
id = Column(BigInteger, primary_key=True)
|
||||
user_id = Column(BigInteger)
|
||||
category_id = Column(BigInteger)
|
||||
keyword = Column(String)
|
||||
created_at = Column(DateTime, default=datetime.now)
|
||||
updated_at = Column(DateTime, default=datetime.now)
|
||||
|
||||
|
||||
class Expense(Base):
|
||||
__tablename__ = "expenses"
|
||||
id = Column(BigInteger, primary_key=True)
|
||||
user_id = Column(BigInteger)
|
||||
category_id = Column(BigInteger, nullable=True)
|
||||
amount = Column(Numeric(10, 2))
|
||||
note = Column(String, nullable=True)
|
||||
invoice_number = Column(String, nullable=True)
|
||||
seller_name = Column(String, nullable=True)
|
||||
item_name = Column(String, nullable=True)
|
||||
date = Column(Date)
|
||||
created_at = Column(DateTime, default=datetime.now)
|
||||
updated_at = Column(DateTime, default=datetime.now)
|
||||
13
app/DB/Session.py
Normal file
13
app/DB/Session.py
Normal file
@@ -0,0 +1,13 @@
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
from app.config import DATABASE_URL
|
||||
|
||||
engine = create_engine(DATABASE_URL)
|
||||
SessionLocal = sessionmaker(bind=engine)
|
||||
|
||||
def get_db():
|
||||
db = SessionLocal()
|
||||
try:
|
||||
yield db
|
||||
finally:
|
||||
db.close()
|
||||
22
app/Invoice/Router.py
Normal file
22
app/Invoice/Router.py
Normal file
@@ -0,0 +1,22 @@
|
||||
import asyncio
|
||||
import threading
|
||||
from fastapi import APIRouter
|
||||
|
||||
router = APIRouter(prefix="/Invoice")
|
||||
|
||||
|
||||
def _run_fetch():
|
||||
from invoice_fetcher import main as fetch_main
|
||||
loop = asyncio.new_event_loop()
|
||||
asyncio.set_event_loop(loop)
|
||||
try:
|
||||
loop.run_until_complete(fetch_main())
|
||||
finally:
|
||||
loop.close()
|
||||
|
||||
|
||||
@router.get("/fetch")
|
||||
async def fetch_invoices():
|
||||
print("🚀 開始抓取發票...")
|
||||
threading.Thread(target=_run_fetch).start()
|
||||
return {"status": "started"}
|
||||
4
app/Line/Captcha_state.py
Normal file
4
app/Line/Captcha_state.py
Normal file
@@ -0,0 +1,4 @@
|
||||
import threading
|
||||
|
||||
captcha_answer: str | None = None
|
||||
captcha_event: threading.Event | None = None
|
||||
259
app/Line/Expense.py
Normal file
259
app/Line/Expense.py
Normal file
@@ -0,0 +1,259 @@
|
||||
import re
|
||||
from datetime import datetime, date
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy import text
|
||||
from app.db.models import User, LineUser, Category, CategoryRule, Expense
|
||||
from app.db.session import SessionLocal
|
||||
|
||||
EXPENSE_TEMPLATE = (
|
||||
"請填寫以下記帳資料後傳回：\n\n"
|
||||
"品項: \n"
|
||||
"類別: \n"
|
||||
"金額: \n"
|
||||
"店家: \n"
|
||||
"備註: "
|
||||
)
|
||||
|
||||
DEFAULT_CATEGORIES = [
|
||||
("餐飲", "#F97316", "🍽"),
|
||||
("交通", "#3B82F6", "🚗"),
|
||||
("購物", "#8B5CF6", "🛍"),
|
||||
("娛樂", "#EC4899", "🎮"),
|
||||
("醫療", "#10B981", "🏥"),
|
||||
("其他", "#6B7280", "📦"),
|
||||
]
|
||||
|
||||
DEFAULT_RULES = {
|
||||
"餐飲": ["早餐", "午餐", "晚餐", "飲料", "便當", "餐飲"],
|
||||
"交通": ["捷運", "公車", "計程車", "油錢", "停車", "交通"],
|
||||
"購物": ["衣服", "3C", "日用品", "購物"],
|
||||
"娛樂": ["電影", "遊戲", "旅遊", "娛樂"],
|
||||
"醫療": ["藥局", "診所", "醫院", "醫療"],
|
||||
}
|
||||
|
||||
|
||||
def get_or_create_user(db: Session, line_user_id: str) -> int:
|
||||
line_user = db.query(LineUser).filter(
|
||||
LineUser.line_user_id == line_user_id
|
||||
).first()
|
||||
if line_user:
|
||||
return line_user.user_id
|
||||
|
||||
new_user = User(name="LINE用戶", email=f"{line_user_id}@line.user")
|
||||
db.add(new_user)
|
||||
db.flush()
|
||||
|
||||
parent_ids = {}
|
||||
for name, color, icon in DEFAULT_CATEGORIES:
|
||||
cat = Category(user_id=new_user.id, name=name, color=color, icon=icon)
|
||||
db.add(cat)
|
||||
db.flush()
|
||||
parent_ids[name] = cat.id
|
||||
|
||||
for cat_name, keywords in DEFAULT_RULES.items():
|
||||
for kw in keywords:
|
||||
db.add(CategoryRule(
|
||||
user_id=new_user.id,
|
||||
category_id=parent_ids[cat_name],
|
||||
keyword=kw
|
||||
))
|
||||
|
||||
db.add(LineUser(line_user_id=line_user_id, user_id=new_user.id))
|
||||
db.commit()
|
||||
return new_user.id
|
||||
|
||||
|
||||
def resolve_category(db: Session, user_id: int, category_input: str) -> tuple[int, int | None]:
|
||||
"""回傳 (category_id, subcategory_id)"""
|
||||
# 查 rule
|
||||
rule = db.query(CategoryRule).filter(
|
||||
CategoryRule.user_id == user_id,
|
||||
CategoryRule.keyword == category_input
|
||||
).first()
|
||||
if rule:
|
||||
return rule.category_id, None
|
||||
|
||||
# 查是否是大類
|
||||
parent = db.query(Category).filter(
|
||||
Category.user_id == user_id,
|
||||
Category.name == category_input,
|
||||
Category.parent_id == None
|
||||
).first()
|
||||
if parent:
|
||||
return parent.id, None
|
||||
|
||||
# 查是否是已存在小類
|
||||
sub = db.query(Category).filter(
|
||||
Category.user_id == user_id,
|
||||
Category.name == category_input,
|
||||
Category.parent_id != None
|
||||
).first()
|
||||
if sub:
|
||||
return sub.parent_id, sub.id
|
||||
|
||||
# 找不到 → 新增 subcategory 到「其他」
|
||||
other = db.query(Category).filter(
|
||||
Category.user_id == user_id,
|
||||
Category.name == "其他",
|
||||
Category.parent_id == None
|
||||
).first()
|
||||
parent_id = other.id if other else None
|
||||
|
||||
new_sub = Category(user_id=user_id, parent_id=parent_id, name=category_input, color="#6B7280")
|
||||
db.add(new_sub)
|
||||
db.flush()
|
||||
|
||||
db.add(CategoryRule(user_id=user_id, category_id=new_sub.id, keyword=category_input))
|
||||
return parent_id, new_sub.id
|
||||
|
||||
|
||||
def parse_multiline_expense(text: str) -> dict | None:
|
||||
fields = {}
|
||||
field_map = {
|
||||
"品項": "item_name",
|
||||
"類別": "category",
|
||||
"金額": "amount",
|
||||
"店家": "seller_name",
|
||||
"備註": "note",
|
||||
}
|
||||
for line in text.strip().splitlines():
|
||||
for key, field in field_map.items():
|
||||
if line.startswith(f"{key}:") or line.startswith(f"{key}："):
|
||||
value = re.split(r"[:：]", line, 1)[-1].strip()
|
||||
if value:
|
||||
fields[field] = value
|
||||
break
|
||||
|
||||
if not all(k in fields for k in ["item_name", "category", "amount"]):
|
||||
return None
|
||||
|
||||
try:
|
||||
fields["amount"] = float(re.sub(r"[^\d.]", "", fields["amount"]))
|
||||
except ValueError:
|
||||
return None
|
||||
|
||||
return fields
|
||||
|
||||
|
||||
def save_expense(line_user_id: str, fields: dict) -> str:
|
||||
db = SessionLocal()
|
||||
try:
|
||||
user_id = get_or_create_user(db, line_user_id)
|
||||
category_id, subcategory_id = resolve_category(db, user_id, fields["category"])
|
||||
|
||||
db.add(Expense(
|
||||
user_id=user_id,
|
||||
category_id=subcategory_id or category_id,
|
||||
amount=fields["amount"],
|
||||
note=fields.get("note"),
|
||||
seller_name=fields.get("seller_name"),
|
||||
item_name=fields["item_name"],
|
||||
date=date.today(),
|
||||
))
|
||||
db.commit()
|
||||
|
||||
reply = (
|
||||
f"✅ 已記錄！\n"
|
||||
f"品項：{fields['item_name']}\n"
|
||||
f"類別：{fields['category']}\n"
|
||||
f"金額：${fields['amount']:.0f}\n"
|
||||
)
|
||||
if fields.get("seller_name"):
|
||||
reply += f"店家：{fields['seller_name']}\n"
|
||||
if fields.get("note"):
|
||||
reply += f"備註：{fields['note']}\n"
|
||||
return reply.strip()
|
||||
except Exception as e:
|
||||
db.rollback()
|
||||
print("❌ 儲存失敗:", e)
|
||||
return "儲存失敗，請稍後再試"
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
|
||||
def delete_expense(line_user_id: str, target: str) -> str:
|
||||
db = SessionLocal()
|
||||
try:
|
||||
user_id = get_or_create_user(db, line_user_id)
|
||||
rows = db.query(Expense).filter(
|
||||
Expense.user_id == user_id,
|
||||
text("DATE(date) = :today")
|
||||
).params(today=date.today()).all()
|
||||
|
||||
if not rows:
|
||||
return "今天還沒有記錄 📭"
|
||||
|
||||
if target.isdigit():
|
||||
idx = int(target) - 1
|
||||
if idx < 0 or idx >= len(rows):
|
||||
return f"沒有第 {target} 筆記錄，請先輸入「查今天」確認編號"
|
||||
row = rows[idx]
|
||||
db.delete(row)
|
||||
db.commit()
|
||||
return f"✅ 已刪除：{row.item_name} ${float(row.amount):.0f}"
|
||||
|
||||
matched = [r for r in rows if r.item_name == target]
|
||||
if not matched:
|
||||
return f"今天沒有「{target}」的記錄"
|
||||
row = matched[-1]
|
||||
db.delete(row)
|
||||
db.commit()
|
||||
return f"✅ 已刪除：{row.item_name} ${float(row.amount):.0f}"
|
||||
except Exception as e:
|
||||
db.rollback()
|
||||
print("❌ 刪除失敗:", e)
|
||||
return "刪除失敗，請稍後再試"
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
|
||||
def query_today(line_user_id: str) -> str:
|
||||
db = SessionLocal()
|
||||
try:
|
||||
user_id = get_or_create_user(db, line_user_id)
|
||||
rows = db.query(Expense).filter(
|
||||
Expense.user_id == user_id,
|
||||
text("DATE(date) = :today")
|
||||
).params(today=date.today()).all()
|
||||
|
||||
if not rows:
|
||||
return "今天還沒有記錄 📭"
|
||||
|
||||
total = sum(float(r.amount) for r in rows)
|
||||
lines = [
|
||||
f"{i+1}. {r.item_name} ${float(r.amount):.0f}" + (f"（{r.note}）" if r.note else "")
|
||||
for i, r in enumerate(rows)
|
||||
]
|
||||
return "📋 今日記錄：\n" + "\n".join(lines) + f"\n\n💰 合計：${total:.0f}\n\n🗑 刪除請輸入：刪除 編號\n例如：刪除 1"
|
||||
except Exception as e:
|
||||
print("❌ 查詢失敗:", e)
|
||||
return "查詢失敗，請稍後再試"
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
|
||||
def query_month(line_user_id: str) -> str:
|
||||
db = SessionLocal()
|
||||
try:
|
||||
user_id = get_or_create_user(db, line_user_id)
|
||||
now = datetime.now()
|
||||
rows = db.query(Expense).filter(
|
||||
Expense.user_id == user_id,
|
||||
text("EXTRACT(YEAR FROM date) = :year AND EXTRACT(MONTH FROM date) = :month")
|
||||
).params(year=now.year, month=now.month).all()
|
||||
|
||||
if not rows:
|
||||
return "本月還沒有記錄 📭"
|
||||
|
||||
total = sum(float(r.amount) for r in rows)
|
||||
summary = {}
|
||||
for r in rows:
|
||||
key = r.item_name or "其他"
|
||||
summary[key] = summary.get(key, 0) + float(r.amount)
|
||||
lines = [f"{cat}：${amt:.0f}" for cat, amt in sorted(summary.items(), key=lambda x: -x[1])]
|
||||
return f"📊 本月統計（{now.month}月）：\n" + "\n".join(lines) + f"\n\n💰 總計：${total:.0f}"
|
||||
except Exception as e:
|
||||
print("❌ 查詢失敗:", e)
|
||||
return "查詢失敗，請稍後再試"
|
||||
finally:
|
||||
db.close()
|
||||
50
app/Line/Handlers.py
Normal file
50
app/Line/Handlers.py
Normal file
@@ -0,0 +1,50 @@
|
||||
from app.line import captcha_state
|
||||
from app.line.expense import (
|
||||
EXPENSE_TEMPLATE,
|
||||
save_expense,
|
||||
delete_expense,
|
||||
query_today,
|
||||
query_month,
|
||||
parse_multiline_expense,
|
||||
)
|
||||
|
||||
|
||||
def handle_text(line_user_id: str, text: str) -> str:
|
||||
if text == "查今天":
|
||||
return query_today(line_user_id)
|
||||
if text == "查本月":
|
||||
return query_month(line_user_id)
|
||||
if text in ("記帳", "記帳說明"):
|
||||
return EXPENSE_TEMPLATE
|
||||
if text.startswith("刪除 "):
|
||||
return delete_expense(line_user_id, text[3:].strip())
|
||||
|
||||
# 多行記帳
|
||||
if "\n" in text:
|
||||
fields = parse_multiline_expense(text)
|
||||
if fields:
|
||||
return save_expense(line_user_id, fields)
|
||||
return (
|
||||
"格式不正確 😅\n\n"
|
||||
"必填欄位：品項、類別、金額\n"
|
||||
"點下方「記帳」取得模板！"
|
||||
)
|
||||
|
||||
return (
|
||||
"點下方「記帳」取得記帳模板\n"
|
||||
"或輸入「查今天」/「查本月」查詢"
|
||||
)
|
||||
|
||||
|
||||
def handle_captcha(text: str) -> bool:
|
||||
"""回傳 True 表示是驗證碼輸入"""
|
||||
if (
|
||||
text.isdigit()
|
||||
and len(text) == 5
|
||||
and captcha_state.captcha_event
|
||||
and not captcha_state.captcha_event.is_set()
|
||||
):
|
||||
captcha_state.captcha_answer = text
|
||||
captcha_state.captcha_event.set()
|
||||
return True
|
||||
return False
|
||||
59
app/Line/Router.py
Normal file
59
app/Line/Router.py
Normal file
@@ -0,0 +1,59 @@
|
||||
from fastapi import APIRouter, Request, HTTPException
|
||||
from linebot.v3 import WebhookHandler
|
||||
from linebot.v3.messaging import (
|
||||
Configuration,
|
||||
ApiClient,
|
||||
MessagingApi,
|
||||
ReplyMessageRequest,
|
||||
TextMessage,
|
||||
)
|
||||
from linebot.v3.webhooks import MessageEvent, TextMessageContent, FollowEvent
|
||||
from linebot.v3.exceptions import InvalidSignatureError
|
||||
from app.config import LINE_CHANNEL_ACCESS_TOKEN, LINE_CHANNEL_SECRET
|
||||
from app.line.handlers import handle_text, handle_captcha
|
||||
|
||||
router = APIRouter()
|
||||
configuration = Configuration(access_token=LINE_CHANNEL_ACCESS_TOKEN)
|
||||
handler = WebhookHandler(LINE_CHANNEL_SECRET)
|
||||
|
||||
|
||||
@router.post("/webhook")
|
||||
async def webhook(request: Request):
|
||||
signature = request.headers.get("X-Line-Signature", "")
|
||||
body = await request.body()
|
||||
try:
|
||||
handler.handle(body.decode(), signature)
|
||||
except InvalidSignatureError:
|
||||
raise HTTPException(status_code=400, detail="Invalid signature")
|
||||
return "OK"
|
||||
|
||||
|
||||
def _reply(reply_token: str, text: str):
|
||||
with ApiClient(configuration) as api_client:
|
||||
MessagingApi(api_client).reply_message(
|
||||
ReplyMessageRequest(
|
||||
reply_token=reply_token,
|
||||
messages=[TextMessage(text=text)]
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
@handler.add(MessageEvent, message=TextMessageContent)
|
||||
def on_message(event):
|
||||
line_user_id = event.source.user_id
|
||||
msg = event.message.text.strip()
|
||||
|
||||
if handle_captcha(msg):
|
||||
_reply(event.reply_token, "✅ 驗證碼已送出！")
|
||||
else:
|
||||
_reply(event.reply_token, handle_text(line_user_id, msg))
|
||||
|
||||
|
||||
@handler.add(FollowEvent)
|
||||
def on_follow(event):
|
||||
_reply(event.reply_token, (
|
||||
"👋 歡迎使用 Myfinance 記帳 Bot！\n\n"
|
||||
"📝 點下方「記帳」開始記帳\n"
|
||||
"📊 點「查今天」或「查本月」查詢\n\n"
|
||||
"開始記帳吧！💪"
|
||||
))
|
||||
9
app/Trading/Router.py
Normal file
9
app/Trading/Router.py
Normal file
@@ -0,0 +1,9 @@
|
||||
from fastapi import APIRouter
|
||||
|
||||
router = APIRouter(prefix="/Trading")
|
||||
|
||||
|
||||
@router.get("/check-signals")
|
||||
async def check_signals():
|
||||
# TODO: 移植策略邏輯
|
||||
return {"status": "not_implemented"}
|
||||
248
app/main.py
248
app/main.py
@@ -1,247 +1,13 @@
|
||||
import os
|
||||
import re
|
||||
import asyncio
|
||||
import threading
|
||||
import nest_asyncio
|
||||
nest_asyncio.apply()
|
||||
import captcha_state
|
||||
from invoice_fetcher import main
|
||||
from dotenv import load_dotenv
|
||||
from fastapi import FastAPI, Request, HTTPException
|
||||
from fastapi.staticfiles import StaticFiles
|
||||
from linebot.v3 import WebhookHandler
|
||||
from linebot.v3.messaging import (
|
||||
Configuration,
|
||||
ApiClient,
|
||||
MessagingApi,
|
||||
ReplyMessageRequest,
|
||||
TextMessage,
|
||||
)
|
||||
from linebot.v3.webhooks import MessageEvent, TextMessageContent, FollowEvent
|
||||
from linebot.v3.exceptions import InvalidSignatureError
|
||||
from sqlalchemy import create_engine, Column, Integer, String, Float, DateTime, text
|
||||
from sqlalchemy.orm import declarative_base, sessionmaker
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
load_dotenv()
|
||||
from fastapi import FastAPI
|
||||
from app.line.router import router as line_router
|
||||
from app.invoice.router import router as invoice_router
|
||||
from app.trading.router import router as trading_router
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
# LINE 設定
|
||||
configuration = Configuration(access_token=os.getenv("LINE_CHANNEL_ACCESS_TOKEN"))
|
||||
handler = WebhookHandler(os.getenv("LINE_CHANNEL_SECRET"))
|
||||
|
||||
# DB 設定
|
||||
# DATABASE_URL = os.getenv("DATABASE_URL")
|
||||
# engine = create_engine(DATABASE_URL)
|
||||
# SessionLocal = sessionmaker(bind=engine)
|
||||
# Base = declarative_base()
|
||||
|
||||
# 資料表
|
||||
# class Transaction(Base):
|
||||
# __tablename__ = "transactions"
|
||||
# id = Column(Integer, primary_key=True, index=True)
|
||||
# user_id = Column(String)
|
||||
# category = Column(String)
|
||||
# amount = Column(Float)
|
||||
# note = Column(String, nullable=True)
|
||||
# created_at = Column(DateTime, default=datetime.now)
|
||||
|
||||
# Base.metadata.create_all(bind=engine)
|
||||
|
||||
# Webhook endpoint
|
||||
@app.post("/webhook")
|
||||
async def webhook(request: Request):
|
||||
signature = request.headers.get("X-Line-Signature", "")
|
||||
body = await request.body()
|
||||
try:
|
||||
handler.handle(body.decode(), signature)
|
||||
except InvalidSignatureError:
|
||||
raise HTTPException(status_code=400, detail="Invalid signature")
|
||||
return "OK"
|
||||
|
||||
def run_fetch_in_thread():
|
||||
# 開一個全新的 event loop 跑 Playwright
|
||||
loop = asyncio.new_event_loop()
|
||||
asyncio.set_event_loop(loop)
|
||||
try:
|
||||
loop.run_until_complete(main())
|
||||
finally:
|
||||
loop.close()
|
||||
|
||||
@app.get("/fetch")
|
||||
async def fetch_invoices():
|
||||
print("🚀 開始抓取發票...")
|
||||
thread = threading.Thread(target=run_fetch_in_thread)
|
||||
thread.start()
|
||||
return {"status": "started"} # 立刻回傳，不等爬蟲
|
||||
|
||||
@handler.add(MessageEvent, message=TextMessageContent)
|
||||
def handle_message(event):
|
||||
user_id = event.source.user_id
|
||||
text = event.message.text.strip()
|
||||
if text.isdigit() and captcha_state.captcha_event and not captcha_state.captcha_event.is_set():
|
||||
captcha_state.captcha_answer = text
|
||||
captcha_state.captcha_event.set() # 通知爬蟲
|
||||
reply = "✅ 驗證碼已送出！"
|
||||
else:
|
||||
reply = parse_and_save(user_id, text)
|
||||
|
||||
with ApiClient(configuration) as api_client:
|
||||
line_bot_api = MessagingApi(api_client)
|
||||
line_bot_api.reply_message(
|
||||
ReplyMessageRequest(
|
||||
reply_token=event.reply_token,
|
||||
messages=[TextMessage(text=reply)]
|
||||
)
|
||||
)
|
||||
|
||||
@handler.add(FollowEvent)
|
||||
def handle_follow(event):
|
||||
with ApiClient(configuration) as api_client:
|
||||
line_bot_api = MessagingApi(api_client)
|
||||
line_bot_api.reply_message(
|
||||
ReplyMessageRequest(
|
||||
reply_token=event.reply_token,
|
||||
messages=[TextMessage(text=(
|
||||
"👋 歡迎使用 Myfinance 記帳 Bot！\n\n"
|
||||
"📝 記帳格式：\n"
|
||||
"類別 金額\n"
|
||||
"類別 金額 備註\n\n"
|
||||
"範例：\n"
|
||||
"早餐 80\n"
|
||||
"午餐 120 便當\n"
|
||||
"交通 50 捷運\n\n"
|
||||
"📊 查詢指令：\n"
|
||||
"查今天 → 今日明細\n"
|
||||
"查本月 → 本月統計\n\n"
|
||||
"開始記帳吧！💪"
|
||||
))]
|
||||
)
|
||||
)
|
||||
|
||||
def parse_and_save(user_id: str, text: str) -> str:
|
||||
# 查詢指令
|
||||
if text == "查今天":
|
||||
return query_today(user_id)
|
||||
if text == "查本月":
|
||||
return query_month(user_id)
|
||||
if text == "記帳說明":
|
||||
return (
|
||||
"📝 記帳格式：\n"
|
||||
"類別 金額\n"
|
||||
"類別 金額 備註\n\n"
|
||||
"範例：\n"
|
||||
"早餐 80\n"
|
||||
"午餐 120 便當\n"
|
||||
"交通 50 捷運\n\n"
|
||||
"輸入後 Bot 會自動記錄 ✅"
|
||||
)
|
||||
|
||||
# 刪除指令：刪除 1 或 刪除 早餐
|
||||
if text.startswith("刪除 "):
|
||||
target = text[3:].strip()
|
||||
return delete_transaction(user_id, target)
|
||||
|
||||
|
||||
if re.match(r"^\d{5}$", text):
|
||||
return f"接收到驗證碼{text}"
|
||||
|
||||
return "格式錯誤 😅\n記帳請輸入：類別 金額\n例如：早餐 80\n\n查詢請輸入：查今天 / 查本月"
|
||||
|
||||
def save_transaction(user_id, category, amount, note):
|
||||
print(f"記錄交易（模擬）：{user_id} {category} {amount} {note}")
|
||||
# db = SessionLocal()
|
||||
# try:
|
||||
# db.add(Transaction(user_id=user_id, category=category, amount=amount, note=note))
|
||||
# db.commit()
|
||||
# finally:
|
||||
# db.close()
|
||||
|
||||
def delete_transaction(user_id: str, target: str) -> str:
|
||||
# db = SessionLocal()
|
||||
try:
|
||||
today = datetime.now().date()
|
||||
# rows = db.query(Transaction).filter(
|
||||
# Transaction.user_id == user_id,
|
||||
# text("DATE(created_at) = :today")
|
||||
# ).params(today=today).all()
|
||||
|
||||
# if not rows:
|
||||
# return "今天還沒有記錄 📭"
|
||||
|
||||
# 用編號刪除
|
||||
# if target.isdigit():
|
||||
# idx = int(target) - 1
|
||||
# if idx < 0 or idx >= len(rows):
|
||||
# return f"沒有第 {target} 筆記錄，請先輸入「查今天」確認編號"
|
||||
# row = rows[idx]
|
||||
# db.delete(row)
|
||||
# db.commit()
|
||||
# return f"✅ 已刪除：{row.category} ${row.amount:.0f}"
|
||||
|
||||
# 用類別刪除（刪最後一筆）
|
||||
# matched = [r for r in rows if r.category == target]
|
||||
# if not matched:
|
||||
# return f"今天沒有「{target}」的記錄"
|
||||
# row = matched[-1]
|
||||
# db.delete(row)
|
||||
# db.commit()
|
||||
# return f"✅ 已刪除：{row.category} ${row.amount:.0f}"
|
||||
return f"已刪除（模擬）"
|
||||
|
||||
# finally:
|
||||
# db.close()
|
||||
except Exception as e:
|
||||
print("❌ 刪除失敗:", e)
|
||||
return "刪除失敗，請稍後再試"
|
||||
|
||||
def query_today(user_id):
|
||||
# db = SessionLocal()
|
||||
try:
|
||||
today = datetime.now().date()
|
||||
# rows = db.query(Transaction).filter(
|
||||
# Transaction.user_id == user_id,
|
||||
# text("DATE(created_at) = :today")
|
||||
# ).params(today=today).all()
|
||||
# if not rows:
|
||||
# return "今天還沒有記錄 📭"
|
||||
# total = sum(r.amount for r in rows)
|
||||
# lines = [
|
||||
# f"{i+1}. {r.category} ${r.amount:.0f}" + (f"（{r.note}）" if r.note else "")
|
||||
# for i, r in enumerate(rows)
|
||||
# ]
|
||||
# return "📋 今日記錄：\n" + "\n".join(lines) + f"\n\n💰 合計：${total:.0f}\n\n🗑 刪除請輸入：刪除 編號\n例如：刪除 1"
|
||||
print(f"查詢今日記錄（模擬）")
|
||||
return "📋 今日記錄：\n1. 早餐 $80\n2. 午餐 $120 便當\n3. 交通 $50 捷運\n\n💰 合計：$250\n\n🗑 刪除請輸入：刪除 編號\n例如：刪除 1"
|
||||
# finally:
|
||||
# db.close()
|
||||
except Exception as e:
|
||||
print("❌ 查詢失敗:", e)
|
||||
return "查詢失敗，請稍後再試"
|
||||
|
||||
def query_month(user_id):
|
||||
# db = SessionLocal()
|
||||
try:
|
||||
now = datetime.now()
|
||||
# rows = db.query(Transaction).filter(
|
||||
# Transaction.user_id == user_id,
|
||||
# text("EXTRACT(YEAR FROM created_at) = :year AND EXTRACT(MONTH FROM created_at) = :month")
|
||||
# ).params(year=now.year, month=now.month).all()
|
||||
# if not rows:
|
||||
# return "本月還沒有記錄 📭"
|
||||
# total = sum(r.amount for r in rows)
|
||||
# 依類別統計
|
||||
# summary = {}
|
||||
# for r in rows:
|
||||
# summary[r.category] = summary.get(r.category, 0) + r.amount
|
||||
# lines = [f"{cat}：${amt:.0f}" for cat, amt in sorted(summary.items(), key=lambda x: -x[1])]
|
||||
# return f"📊 本月統計（{now.month}月）：\n" + "\n".join(lines) + f"\n\n💰 總計：${total:.0f}"
|
||||
print(f"查詢本月記錄（模擬）")
|
||||
return f"📊 本月統計（{now.month}月）：\n早餐：$800\n午餐：$1200\n交通：$500\n\n💰 總計：$2500"
|
||||
# finally:
|
||||
# db.close()
|
||||
except Exception as e:
|
||||
print("❌ 查詢失敗:", e)
|
||||
return "查詢失敗，請稍後再試"
|
||||
app.include_router(line_router)
|
||||
app.include_router(invoice_router)
|
||||
app.include_router(trading_router)
|
||||
247
app/old_main.py
Normal file
247
app/old_main.py
Normal file
@@ -0,0 +1,247 @@
|
||||
import os
|
||||
import re
|
||||
import asyncio
|
||||
import threading
|
||||
import nest_asyncio
|
||||
nest_asyncio.apply()
|
||||
import captcha_state
|
||||
from invoice_fetcher import main
|
||||
from dotenv import load_dotenv
|
||||
from fastapi import FastAPI, Request, HTTPException
|
||||
from fastapi.staticfiles import StaticFiles
|
||||
from linebot.v3 import WebhookHandler
|
||||
from linebot.v3.messaging import (
|
||||
Configuration,
|
||||
ApiClient,
|
||||
MessagingApi,
|
||||
ReplyMessageRequest,
|
||||
TextMessage,
|
||||
)
|
||||
from linebot.v3.webhooks import MessageEvent, TextMessageContent, FollowEvent
|
||||
from linebot.v3.exceptions import InvalidSignatureError
|
||||
from sqlalchemy import create_engine, Column, Integer, String, Float, DateTime, text
|
||||
from sqlalchemy.orm import declarative_base, sessionmaker
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
load_dotenv()
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
# LINE 設定
|
||||
configuration = Configuration(access_token=os.getenv("LINE_CHANNEL_ACCESS_TOKEN"))
|
||||
handler = WebhookHandler(os.getenv("LINE_CHANNEL_SECRET"))
|
||||
|
||||
# DB 設定
|
||||
# DATABASE_URL = os.getenv("DATABASE_URL")
|
||||
# engine = create_engine(DATABASE_URL)
|
||||
# SessionLocal = sessionmaker(bind=engine)
|
||||
# Base = declarative_base()
|
||||
|
||||
# 資料表
|
||||
# class Transaction(Base):
|
||||
# __tablename__ = "transactions"
|
||||
# id = Column(Integer, primary_key=True, index=True)
|
||||
# user_id = Column(String)
|
||||
# category = Column(String)
|
||||
# amount = Column(Float)
|
||||
# note = Column(String, nullable=True)
|
||||
# created_at = Column(DateTime, default=datetime.now)
|
||||
|
||||
# Base.metadata.create_all(bind=engine)
|
||||
|
||||
# Webhook endpoint
|
||||
@app.post("/webhook")
|
||||
async def webhook(request: Request):
|
||||
signature = request.headers.get("X-Line-Signature", "")
|
||||
body = await request.body()
|
||||
try:
|
||||
handler.handle(body.decode(), signature)
|
||||
except InvalidSignatureError:
|
||||
raise HTTPException(status_code=400, detail="Invalid signature")
|
||||
return "OK"
|
||||
|
||||
def run_fetch_in_thread():
|
||||
# 開一個全新的 event loop 跑 Playwright
|
||||
loop = asyncio.new_event_loop()
|
||||
asyncio.set_event_loop(loop)
|
||||
try:
|
||||
loop.run_until_complete(main())
|
||||
finally:
|
||||
loop.close()
|
||||
|
||||
@app.get("/fetch")
|
||||
async def fetch_invoices():
|
||||
print("🚀 開始抓取發票...")
|
||||
thread = threading.Thread(target=run_fetch_in_thread)
|
||||
thread.start()
|
||||
return {"status": "started"} # 立刻回傳，不等爬蟲
|
||||
|
||||
@handler.add(MessageEvent, message=TextMessageContent)
|
||||
def handle_message(event):
|
||||
user_id = event.source.user_id
|
||||
text = event.message.text.strip()
|
||||
if text.isdigit() and captcha_state.captcha_event and not captcha_state.captcha_event.is_set():
|
||||
captcha_state.captcha_answer = text
|
||||
captcha_state.captcha_event.set() # 通知爬蟲
|
||||
reply = "✅ 驗證碼已送出！"
|
||||
else:
|
||||
reply = parse_and_save(user_id, text)
|
||||
|
||||
with ApiClient(configuration) as api_client:
|
||||
line_bot_api = MessagingApi(api_client)
|
||||
line_bot_api.reply_message(
|
||||
ReplyMessageRequest(
|
||||
reply_token=event.reply_token,
|
||||
messages=[TextMessage(text=reply)]
|
||||
)
|
||||
)
|
||||
|
||||
@handler.add(FollowEvent)
|
||||
def handle_follow(event):
|
||||
with ApiClient(configuration) as api_client:
|
||||
line_bot_api = MessagingApi(api_client)
|
||||
line_bot_api.reply_message(
|
||||
ReplyMessageRequest(
|
||||
reply_token=event.reply_token,
|
||||
messages=[TextMessage(text=(
|
||||
"👋 歡迎使用 Myfinance 記帳 Bot！\n\n"
|
||||
"📝 記帳格式：\n"
|
||||
"類別 金額\n"
|
||||
"類別 金額 備註\n\n"
|
||||
"範例：\n"
|
||||
"早餐 80\n"
|
||||
"午餐 120 便當\n"
|
||||
"交通 50 捷運\n\n"
|
||||
"📊 查詢指令：\n"
|
||||
"查今天 → 今日明細\n"
|
||||
"查本月 → 本月統計\n\n"
|
||||
"開始記帳吧！💪"
|
||||
))]
|
||||
)
|
||||
)
|
||||
|
||||
def parse_and_save(user_id: str, text: str) -> str:
|
||||
# 查詢指令
|
||||
if text == "查今天":
|
||||
return query_today(user_id)
|
||||
if text == "查本月":
|
||||
return query_month(user_id)
|
||||
if text == "記帳說明":
|
||||
return (
|
||||
"📝 記帳格式：\n"
|
||||
"類別 金額\n"
|
||||
"類別 金額 備註\n\n"
|
||||
"範例：\n"
|
||||
"早餐 80\n"
|
||||
"午餐 120 便當\n"
|
||||
"交通 50 捷運\n\n"
|
||||
"輸入後 Bot 會自動記錄 ✅"
|
||||
)
|
||||
|
||||
# 刪除指令：刪除 1 或 刪除 早餐
|
||||
if text.startswith("刪除 "):
|
||||
target = text[3:].strip()
|
||||
return delete_transaction(user_id, target)
|
||||
|
||||
|
||||
if re.match(r"^\d{5}$", text):
|
||||
return f"接收到驗證碼{text}"
|
||||
|
||||
return "格式錯誤 😅\n記帳請輸入：類別 金額\n例如：早餐 80\n\n查詢請輸入：查今天 / 查本月"
|
||||
|
||||
def save_transaction(user_id, category, amount, note):
|
||||
print(f"記錄交易（模擬）：{user_id} {category} {amount} {note}")
|
||||
# db = SessionLocal()
|
||||
# try:
|
||||
# db.add(Transaction(user_id=user_id, category=category, amount=amount, note=note))
|
||||
# db.commit()
|
||||
# finally:
|
||||
# db.close()
|
||||
|
||||
def delete_transaction(user_id: str, target: str) -> str:
|
||||
# db = SessionLocal()
|
||||
try:
|
||||
today = datetime.now().date()
|
||||
# rows = db.query(Transaction).filter(
|
||||
# Transaction.user_id == user_id,
|
||||
# text("DATE(created_at) = :today")
|
||||
# ).params(today=today).all()
|
||||
|
||||
# if not rows:
|
||||
# return "今天還沒有記錄 📭"
|
||||
|
||||
# 用編號刪除
|
||||
# if target.isdigit():
|
||||
# idx = int(target) - 1
|
||||
# if idx < 0 or idx >= len(rows):
|
||||
# return f"沒有第 {target} 筆記錄，請先輸入「查今天」確認編號"
|
||||
# row = rows[idx]
|
||||
# db.delete(row)
|
||||
# db.commit()
|
||||
# return f"✅ 已刪除：{row.category} ${row.amount:.0f}"
|
||||
|
||||
# 用類別刪除（刪最後一筆）
|
||||
# matched = [r for r in rows if r.category == target]
|
||||
# if not matched:
|
||||
# return f"今天沒有「{target}」的記錄"
|
||||
# row = matched[-1]
|
||||
# db.delete(row)
|
||||
# db.commit()
|
||||
# return f"✅ 已刪除：{row.category} ${row.amount:.0f}"
|
||||
return f"已刪除（模擬）"
|
||||
|
||||
# finally:
|
||||
# db.close()
|
||||
except Exception as e:
|
||||
print("❌ 刪除失敗:", e)
|
||||
return "刪除失敗，請稍後再試"
|
||||
|
||||
def query_today(user_id):
|
||||
# db = SessionLocal()
|
||||
try:
|
||||
today = datetime.now().date()
|
||||
# rows = db.query(Transaction).filter(
|
||||
# Transaction.user_id == user_id,
|
||||
# text("DATE(created_at) = :today")
|
||||
# ).params(today=today).all()
|
||||
# if not rows:
|
||||
# return "今天還沒有記錄 📭"
|
||||
# total = sum(r.amount for r in rows)
|
||||
# lines = [
|
||||
# f"{i+1}. {r.category} ${r.amount:.0f}" + (f"（{r.note}）" if r.note else "")
|
||||
# for i, r in enumerate(rows)
|
||||
# ]
|
||||
# return "📋 今日記錄：\n" + "\n".join(lines) + f"\n\n💰 合計：${total:.0f}\n\n🗑 刪除請輸入：刪除 編號\n例如：刪除 1"
|
||||
print(f"查詢今日記錄（模擬）")
|
||||
return "📋 今日記錄：\n1. 早餐 $80\n2. 午餐 $120 便當\n3. 交通 $50 捷運\n\n💰 合計：$250\n\n🗑 刪除請輸入：刪除 編號\n例如：刪除 1"
|
||||
# finally:
|
||||
# db.close()
|
||||
except Exception as e:
|
||||
print("❌ 查詢失敗:", e)
|
||||
return "查詢失敗，請稍後再試"
|
||||
|
||||
def query_month(user_id):
|
||||
# db = SessionLocal()
|
||||
try:
|
||||
now = datetime.now()
|
||||
# rows = db.query(Transaction).filter(
|
||||
# Transaction.user_id == user_id,
|
||||
# text("EXTRACT(YEAR FROM created_at) = :year AND EXTRACT(MONTH FROM created_at) = :month")
|
||||
# ).params(year=now.year, month=now.month).all()
|
||||
# if not rows:
|
||||
# return "本月還沒有記錄 📭"
|
||||
# total = sum(r.amount for r in rows)
|
||||
# 依類別統計
|
||||
# summary = {}
|
||||
# for r in rows:
|
||||
# summary[r.category] = summary.get(r.category, 0) + r.amount
|
||||
# lines = [f"{cat}：${amt:.0f}" for cat, amt in sorted(summary.items(), key=lambda x: -x[1])]
|
||||
# return f"📊 本月統計（{now.month}月）：\n" + "\n".join(lines) + f"\n\n💰 總計：${total:.0f}"
|
||||
print(f"查詢本月記錄（模擬）")
|
||||
return f"📊 本月統計（{now.month}月）：\n早餐：$800\n午餐：$1200\n交通：$500\n\n💰 總計：$2500"
|
||||
# finally:
|
||||
# db.close()
|
||||
except Exception as e:
|
||||
print("❌ 查詢失敗:", e)
|
||||
return "查詢失敗，請稍後再試"
|
||||
19
structure.txt
Normal file
19
structure.txt
Normal file
@@ -0,0 +1,19 @@
|
||||
app/
|
||||
├── main.py # include 三個 router
|
||||
├── config.py # 所有環境變數統一管理
|
||||
├── db/
|
||||
│ ├── session.py # engine + SessionLocal
|
||||
│ └── models.py # 所有 SQLAlchemy model
|
||||
├── line/
|
||||
│ ├── router.py # /webhook
|
||||
│ ├── handlers.py # 訊息進來的入口
|
||||
│ ├── expense.py # 記帳所有邏輯（原本的 main.py）
|
||||
│ └── captcha_state.py
|
||||
├── invoice/
|
||||
│ └── router.py # /invoice/fetch
|
||||
└── trading/
|
||||
├── router.py # /trading/check-signals（骨架）
|
||||
├── data.py # TODO: 移植行情抓取
|
||||
├── indicators.py # TODO: 移植指標計算
|
||||
├── strategy.py # TODO: 移植策略邏輯
|
||||
└── broker.py # TODO: 永豐 Shioaji
|
||||
Reference in New Issue
Block a user