OpenSTR/app/models.py

48 lines
No EOL
2 KiB
Python

from flask_sqlalchemy import SQLAlchemy
from app import db
class Customer(db.Model):
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(120), unique=True, nullable=False)
username = db.Column(db.String(80), unique=True, nullable=False)
appointments = db.relationship('Appointment', backref='customer', lazy=True)
repairs = db.relationship('Repair', backref='customer', lazy=True)
class Technician(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), unique=True, nullable=False)
repairs = db.relationship('Repair', backref='technician', lazy=True)
class Repair(db.Model):
id = db.Column(db.Integer, primary_key=True)
device_info = db.Column(db.String(200), nullable=False)
status = db.Column(db.String(50), nullable=False)
technician_id = db.Column(db.Integer, db.ForeignKey('technician.id'), nullable=True)
customer_id = db.Column(db.Integer, db.ForeignKey('customer.id'), nullable=False)
notes = db.relationship('RepairNote', backref='repair', lazy=True)
class Appointment(db.Model):
id = db.Column(db.Integer, primary_key=True)
date_time = db.Column(db.DateTime, nullable=False)
customer_id = db.Column(db.Integer, db.ForeignKey('customer.id'), nullable=False)
class RepairNote(db.Model):
id = db.Column(db.Integer, primary_key=True)
content = db.Column(db.Text, nullable=False)
repair_id = db.Column(db.Integer, db.ForeignKey('repair.id'), nullable=False)
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(64), unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
def __repr__(self):
return f'<User {self.username}>'
class RepairBay(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(100), unique=True, nullable=False)
location = db.Column(db.String(100), nullable=False)
def __repr__(self):
return f'<RepairBay {self.name}>'