from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

class Violation(db.Model):

id = db.Column(db.Integer, primary_key=True)

student_id = db.Column(db.String(20))

violation_type = db.Column(db.String(100))

description = db.Column(db.Text)

status = db.Column(db.String(20), default='pending')

class Notification(db.Model):

id = db.Column(db.Integer, primary_key=True)

violation_id = db.Column(db.Integer, db.ForeignKey('violation.id'))

message = db.Column(db.Text)

sent = db.Column(db.Boolean, default=False)