126 lines
4.9 KiB
Python
126 lines
4.9 KiB
Python
# plugins/plant/models.py
|
|
|
|
from datetime import datetime
|
|
import uuid as uuid_lib
|
|
from app import db
|
|
# from plugins.auth.models import User
|
|
|
|
|
|
# Association table for Plant ↔ Tag (unchanged)
|
|
plant_tags = db.Table(
|
|
'plant_tags',
|
|
db.metadata,
|
|
db.Column('plant_id', db.Integer, db.ForeignKey('plant.id'), primary_key=True),
|
|
db.Column('tag_id', db.Integer, db.ForeignKey('tag.id'), primary_key=True),
|
|
extend_existing=True
|
|
)
|
|
|
|
class Tag(db.Model):
|
|
__tablename__ = 'tag'
|
|
__table_args__ = {'extend_existing': True}
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(128), unique=True, nullable=False)
|
|
# … any other columns you had …
|
|
|
|
class PlantCommonName(db.Model):
|
|
__tablename__ = 'plant_common_name'
|
|
__table_args__ = {'extend_existing': True}
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(128), unique=True, nullable=False)
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
scientific_names = db.relationship(
|
|
'plugins.plant.models.PlantScientificName',
|
|
backref=db.backref('common', lazy='joined'),
|
|
lazy=True,
|
|
cascade='all, delete-orphan'
|
|
)
|
|
|
|
class PlantScientificName(db.Model):
|
|
__tablename__ = 'plant_scientific_name'
|
|
__table_args__ = {'extend_existing': True}
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(256), unique=True, nullable=False)
|
|
common_id = db.Column(db.Integer, db.ForeignKey('plant_common_name.id'), nullable=False)
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
# We removed the “plants” relationship from here to avoid backref conflicts.
|
|
# If you need it, you can still do Plant.query.filter_by(scientific_id=<this id>).
|
|
|
|
class PlantOwnershipLog(db.Model):
|
|
__tablename__ = 'plant_ownership_log'
|
|
__table_args__ = {'extend_existing': True}
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
plant_id = db.Column(db.Integer, db.ForeignKey('plant.id'), nullable=False)
|
|
user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
|
|
date_acquired = db.Column(db.DateTime, default=datetime.utcnow)
|
|
transferred = db.Column(db.Boolean, default=False, nullable=False)
|
|
is_verified = db.Column(db.Boolean, default=False, nullable=False)
|
|
|
|
# Optional: if you ever want to store a pointer to the Neo4j node, you can re-add:
|
|
# graph_node_id = db.Column(db.String(255), nullable=True)
|
|
|
|
user = db.relationship(
|
|
'plugins.auth.models.User',
|
|
backref='ownership_logs',
|
|
lazy=True
|
|
)
|
|
|
|
class Plant(db.Model):
|
|
__tablename__ = 'plant'
|
|
__table_args__ = {'extend_existing': True}
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
uuid = db.Column(db.String(36), default=lambda: str(uuid_lib.uuid4()), unique=True, nullable=False)
|
|
custom_slug = db.Column(db.String(255), unique=True, nullable=True)
|
|
|
|
owner_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
|
|
common_id = db.Column(db.Integer, db.ForeignKey('plant_common_name.id'), nullable=False)
|
|
scientific_id = db.Column(db.Integer, db.ForeignKey('plant_scientific_name.id'), nullable=False)
|
|
|
|
plant_type = db.Column(db.String(50), nullable=False)
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
updated_at = db.Column(db.DateTime, onupdate=datetime.utcnow)
|
|
|
|
# ─── NEW: Flag that indicates whether the common/scientific name pair was human-verified ─────────────────
|
|
data_verified = db.Column(db.Boolean, default=False, nullable=False)
|
|
|
|
# Relationships
|
|
updates = db.relationship(
|
|
'plugins.growlog.models.PlantUpdate',
|
|
backref='plant',
|
|
lazy=True,
|
|
cascade='all, delete-orphan'
|
|
)
|
|
tags = db.relationship(
|
|
'plugins.plant.models.Tag',
|
|
secondary=plant_tags,
|
|
backref='plants',
|
|
lazy='dynamic'
|
|
)
|
|
|
|
common_name = db.relationship(
|
|
'plugins.plant.models.PlantCommonName',
|
|
backref=db.backref('plants', lazy='dynamic'),
|
|
lazy=True
|
|
)
|
|
scientific_name = db.relationship(
|
|
'plugins.plant.models.PlantScientificName',
|
|
backref=db.backref('plants', lazy='dynamic'),
|
|
lazy=True
|
|
)
|
|
|
|
ownership_logs = db.relationship(
|
|
'plugins.plant.models.PlantOwnershipLog',
|
|
backref='plant',
|
|
lazy=True,
|
|
cascade='all, delete-orphan'
|
|
)
|
|
|
|
def __repr__(self):
|
|
return f"<Plant {self.uuid} ({self.plant_type})>"
|