42 lines
1.8 KiB
Python

from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Numeric, Text
from sqlalchemy.sql import func
from sqlalchemy.orm import relationship
from app.db.base import Base
class Invoice(Base):
__tablename__ = "invoices"
id = Column(Integer, primary_key=True, index=True)
invoice_number = Column(String, unique=True, nullable=False, index=True)
issue_date = Column(DateTime(timezone=True), nullable=False)
due_date = Column(DateTime(timezone=True), nullable=False)
status = Column(String, default="draft") # draft, sent, paid, overdue
subtotal = Column(Numeric(10, 2), default=0)
tax_rate = Column(Numeric(5, 2), default=0)
tax_amount = Column(Numeric(10, 2), default=0)
total_amount = Column(Numeric(10, 2), default=0)
notes = Column(Text, nullable=True)
user_id = Column(Integer, ForeignKey("users.id"))
customer_id = Column(Integer, ForeignKey("customers.id"))
created_at = Column(DateTime(timezone=True), server_default=func.now())
updated_at = Column(DateTime(timezone=True), onupdate=func.now())
user = relationship("User", back_populates="invoices")
customer = relationship("Customer", back_populates="invoices")
items = relationship("InvoiceItem", back_populates="invoice", cascade="all, delete-orphan")
class InvoiceItem(Base):
__tablename__ = "invoice_items"
id = Column(Integer, primary_key=True, index=True)
description = Column(String, nullable=False)
quantity = Column(Numeric(10, 2), nullable=False)
unit_price = Column(Numeric(10, 2), nullable=False)
total_price = Column(Numeric(10, 2), nullable=False)
invoice_id = Column(Integer, ForeignKey("invoices.id"))
created_at = Column(DateTime(timezone=True), server_default=func.now())
invoice = relationship("Invoice", back_populates="items")