39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
from app import db
|
|
from app.utils.time_utils import now_eastern
|
|
|
|
class Facility(db.Model):
|
|
__tablename__ = 'facilities'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(255), nullable=False)
|
|
address = db.Column(db.Text)
|
|
contact_person = db.Column(db.String(100))
|
|
contact_phone = db.Column(db.String(20))
|
|
active = db.Column(db.Boolean, default=True)
|
|
created_at = db.Column(db.DateTime, default=now_eastern, nullable=True)
|
|
|
|
# Phase 1: link facility to a project (nullable for backward compatibility)
|
|
project_id = db.Column(db.Integer, db.ForeignKey('projects.id', ondelete='SET NULL'),
|
|
nullable=True, index=True)
|
|
|
|
# Relationships
|
|
areas = db.relationship('Area', backref='facility', lazy='dynamic')
|
|
inspections = db.relationship('Inspection', backref='facility', lazy='dynamic')
|
|
|
|
def __repr__(self):
|
|
return f'<Facility {self.name}>'
|
|
|
|
class Area(db.Model):
|
|
__tablename__ = 'areas'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
facility_id = db.Column(db.Integer, db.ForeignKey('facilities.id'), nullable=False)
|
|
name = db.Column(db.String(255), nullable=False)
|
|
area_type = db.Column(db.String(50))
|
|
|
|
# Relationships
|
|
inspections = db.relationship('Inspection', backref='area', lazy='dynamic')
|
|
issues = db.relationship('Issue', backref='area', lazy='dynamic')
|
|
|
|
def __repr__(self):
|
|
return f'<Area {self.name}>' |