diff --git a/models/tree.py b/models/tree.py new file mode 100644 index 0000000..05ee62f --- /dev/null +++ b/models/tree.py @@ -0,0 +1,20 @@ +from sqlalchemy import Column, String, Integer, ForeignKey +from sqlalchemy.orm import relationship +from sqlalchemy.sql import func +from app.api.db.base_class import Base +import uuid + +class Tree(Base): + __tablename__ = "trees" + + id = Column(Integer, primary_key=True, index=True) + name = Column(String, nullable=False) + species = Column(String, nullable=False) + height = Column(Integer, nullable=False) + diameter = Column(Integer, nullable=False) + location_id = Column(Integer, ForeignKey("locations.id"), nullable=False) + + location = relationship("Location", back_populates="trees") + + created_at = Column(DateTime, default=func.now()) + updated_at = Column(DateTime, default=func.now(), onupdate=func.now()) \ No newline at end of file