67 lines
No EOL
2.8 KiB
Python
67 lines
No EOL
2.8 KiB
Python
from . import db
|
|
from flask_login import UserMixin
|
|
from datetime import datetime
|
|
|
|
# User roles: admin, technician, customer
|
|
class User(db.Model, UserMixin):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
username = db.Column(db.String(80), unique=True, nullable=False)
|
|
password_hash = db.Column(db.String(128), nullable=False)
|
|
role = db.Column(db.String(20), default='customer')
|
|
customer = db.relationship('Customer', uselist=False, backref='user')
|
|
|
|
class Customer(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(120))
|
|
email = db.Column(db.String(120), unique=True)
|
|
phone = db.Column(db.String(20))
|
|
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
|
|
repairs = db.relationship('RepairJob', backref='customer')
|
|
|
|
class Technician(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(120))
|
|
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
|
|
repairs = db.relationship('RepairJob', backref='technician')
|
|
|
|
class RepairJob(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
tracking_number = db.Column(db.String(50), unique=True)
|
|
device = db.Column(db.String(120))
|
|
description = db.Column(db.Text)
|
|
status = db.Column(db.String(50), default='New')
|
|
customer_id = db.Column(db.Integer, db.ForeignKey('customer.id'))
|
|
technician_id = db.Column(db.Integer, db.ForeignKey('technician.id'))
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
cost_estimate = db.Column(db.Float)
|
|
final_cost = db.Column(db.Float)
|
|
parts_used = db.relationship('RepairPart', backref='repair')
|
|
barcode_path = db.Column(db.String(256))
|
|
notes = db.relationship('RepairNote', backref='repair')
|
|
|
|
class RepairNote(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
repair_id = db.Column(db.Integer, db.ForeignKey('repair_job.id'))
|
|
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
|
|
note = db.Column(db.Text)
|
|
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
class RepairPart(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
repair_id = db.Column(db.Integer, db.ForeignKey('repair_job.id'))
|
|
part_id = db.Column(db.Integer, db.ForeignKey('inventory_item.id'))
|
|
quantity = db.Column(db.Integer)
|
|
|
|
class InventoryItem(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
part_name = db.Column(db.String(120))
|
|
quantity = db.Column(db.Integer)
|
|
low_stock_threshold = db.Column(db.Integer, default=1)
|
|
|
|
class AuditLog(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
|
|
action = db.Column(db.String(256))
|
|
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
|
|
details = db.Column(db.Text) |