from sqlalchemy.orm import Session
from sqlalchemy import text
from db import models, schemas
from datetime import datetime

def create_card_expense(db: Session, card_expense: schemas.CardExpenseCreate):
    db_card_expense = models.CardExpense(jang=card_expense.jang, yeo=card_expense.yeo, kim=card_expense.kim, choi=card_expense.choi, amount=card_expense.amount, purpose=card_expense.purpose, created_at=datetime.now())
    db.add(db_card_expense)
    db.commit()
    db.refresh(db_card_expense)
    return db_card_expense


def get_card_expenses_by_date(db: Session, year: int, month: int):
    query = text("""
        SELECT * FROM card_expenses 
        WHERE EXTRACT(YEAR FROM created_at) = :year
        AND EXTRACT(MONTH FROM created_at) = :month
    """)
    return db.execute(query, {'year': year, 'month': month}).fetchall()