파이썬 기초(24) — 종합 실습 2, CSV 저장과 명령행 실행까지 완성하기

"""Python 3.11+ 표준 라이브러리로 만든 단일 사용자 지출 기록기."""
import argparse
import csv
from datetime import date
from pathlib import Path
import re
import sys
import tempfile

FIELDS = ["date", "item", "amount"]
DEFAULT_FILE = Path(__file__).resolve().parent / "data" / "expenses.csv"

def validate_date(text):
    if not isinstance(text, str) or not re.fullmatch(r"[0-9]{4}-[0-9]{2}-[0-9]{2}", text):
        raise ValueError("날짜는 YYYY-MM-DD 형식이어야 합니다.")
    try:
        return date.fromisoformat(text).isoformat()
    except ValueError:
        raise ValueError("실제 달력에 있는 날짜를 입력해 주세요.") from None

def validate_month(text):
    if not isinstance(text, str) or not re.fullmatch(r"[0-9]{4}-[0-9]{2}", text):
        raise ValueError("월은 YYYY-MM 형식이어야 합니다.")
    validate_date(text + "-01")
    return text

def validate_item(text):
    if not isinstance(text, str):
        raise ValueError("항목은 문자열이어야 합니다.")
    text = text.strip()
    if not text or any(ord(char) < 32 or ord(char) == 127 for char in text):
        raise ValueError("항목은 비어 있지 않은 한 줄이어야 합니다.")
    if text.startswith(("=", "+", "-", "@")):
        raise ValueError("항목 첫 글자에 =, +, -, @를 사용할 수 없습니다.")
    return text

def validate_amount(amount):
    if type(amount) is not int or amount <= 0:
        raise ValueError("금액은 0보다 큰 정수여야 합니다.")
    return amount

def validate_row(row):
    return {
        "date": validate_date(row["date"]),
        "item": validate_item(row["item"]),
        "amount": validate_amount(row["amount"]),
    }

def load_expenses(path):
    path = Path(path)
    rows = []
    try:
        file = path.open("r", encoding="utf-8-sig", newline="")
    except FileNotFoundError:
        return rows
    with file:
        reader = csv.DictReader(file, strict=True)
        if reader.fieldnames != FIELDS:
            raise ValueError("CSV 헤더는 date,item,amount 순서여야 합니다.")
        for row in reader:
            try:
                if set(row) != set(FIELDS) or any(value is None for value in row.values()):
                    raise ValueError("열 개수가 맞지 않습니다.")
                row["amount"] = int(row["amount"])
                rows.append(validate_row(row))
            except (ValueError, TypeError, KeyError) as error:
                raise ValueError(f"CSV {reader.line_num}행: {error}") from error
    return rows

def save_expenses(path, rows):
    path = Path(path)
    checked = [validate_row(row) for row in rows]
    path.parent.mkdir(parents=True, exist_ok=True)
    temporary = None
    try:
        with tempfile.NamedTemporaryFile(
            mode="w", encoding="utf-8-sig", newline="",
            dir=path.parent, prefix=".expenses-", suffix=".tmp", delete=False,
        ) as file:
            temporary = Path(file.name)
            writer = csv.DictWriter(file, fieldnames=FIELDS)
            writer.writeheader()
            writer.writerows(checked)
        temporary.replace(path)
    finally:
        if temporary is not None:
            temporary.unlink(missing_ok=True)

def summarize(rows, month=None):
    if month is not None:
        month = validate_month(month)
    selected = [row for row in rows if month is None or row["date"].startswith(month + "-")]
    return len(selected), sum(row["amount"] for row in selected)

def build_parser():
    parser = argparse.ArgumentParser(description="CSV 지출 기록기")
    parser.add_argument("--file", type=Path, default=DEFAULT_FILE, help="지출 CSV 경로")
    commands = parser.add_subparsers(dest="command", required=True)
    add = commands.add_parser("add", help="지출 한 건 추가")
    add.add_argument("--date", required=True, help="YYYY-MM-DD")
    add.add_argument("--item", required=True, help="지출 항목")
    add.add_argument("--amount", required=True, type=int, help="원 단위 양의 정수")
    commands.add_parser("list", help="모든 지출 조회")
    summary = commands.add_parser("summary", help="건수와 합계")
    summary.add_argument("--month", help="YYYY-MM, 생략하면 전체")
    return parser

def main(argv=None):
    args = build_parser().parse_args(argv)
    try:
        rows = load_expenses(args.file)
        if args.command == "add":
            row = validate_row({"date": args.date, "item": args.item, "amount": args.amount})
            rows.append(row)
            save_expenses(args.file, rows)
            print(f"저장했습니다: {row['date']} / {row['item']} / {row['amount']:,}원")
        elif args.command == "list":
            if not rows:
                print("아직 기록이 없습니다.")
            for number, row in enumerate(rows, start=1):
                print(f"{number}. {row['date']} / {row['item']} / {row['amount']:,}원")
        else:
            count, total = summarize(rows, args.month)
            print(f"{args.month or '전체'} / {count}건 / 합계 {total:,}원")
    except (OSError, UnicodeError, csv.Error, ValueError) as error:
        print(f"오류: {error}", file=sys.stderr)
        return 1
    return 0

if __name__ == "__main__":
    raise SystemExit(main())