Added messaging functions

models.py

@@ -3,8 +3,7 @@ from flask_login import UserMixin
 from werkzeug.security import generate_password_hash, check_password_hash
 from datetime import datetime
 from sqlalchemy.orm import relationship
-
-db = SQLAlchemy()
+from extensions import db
 
 # Association table for room members
 room_members = db.Table('room_members',
@@ -12,6 +11,12 @@ room_members = db.Table('room_members',
     db.Column('user_id', db.Integer, db.ForeignKey('user.id'), primary_key=True)
 )
 
+# Association table for conversation members
+conversation_members = db.Table('conversation_members',
+    db.Column('conversation_id', db.Integer, db.ForeignKey('conversation.id'), primary_key=True),
+    db.Column('user_id', db.Integer, db.ForeignKey('user.id'), primary_key=True)
+)
+
 class User(UserMixin, db.Model):
     id = db.Column(db.Integer, primary_key=True)
     username = db.Column(db.String(150), unique=True, nullable=False)
@@ -151,4 +156,49 @@ class SiteSettings(db.Model):
             settings = cls()
             db.session.add(settings)
             db.session.commit()
-        return settings 
+        return settings 
+
+class Conversation(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(100), nullable=False)
+    description = db.Column(db.Text)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    created_by = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
+    
+    # Relationships
+    creator = db.relationship('User', backref='created_conversations', foreign_keys=[created_by])
+    members = db.relationship('User', secondary=conversation_members, backref=db.backref('conversations', lazy='dynamic'))
+    messages = db.relationship('Message', back_populates='conversation', cascade='all, delete-orphan')
+
+    def __repr__(self):
+        return f'<Conversation {self.name}>'
+
+class Message(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    content = db.Column(db.Text, nullable=False)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    conversation_id = db.Column(db.Integer, db.ForeignKey('conversation.id'), nullable=False)
+    user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
+    
+    # Relationships
+    conversation = db.relationship('Conversation', back_populates='messages')
+    user = db.relationship('User', backref='messages')
+    attachments = db.relationship('MessageAttachment', back_populates='message', cascade='all, delete-orphan')
+
+    def __repr__(self):
+        return f'<Message {self.id}>'
+
+class MessageAttachment(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    message_id = db.Column(db.Integer, db.ForeignKey('message.id'), nullable=False)
+    name = db.Column(db.String(255), nullable=False)
+    path = db.Column(db.String(512), nullable=False)
+    type = db.Column(db.String(100))
+    size = db.Column(db.Integer)  # Size in bytes
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    
+    # Relationships
+    message = db.relationship('Message', back_populates='attachments')
+
+    def __repr__(self):
+        return f'<MessageAttachment {self.name}>'