first

models.py

@@ -0,0 +1,127 @@
+from flask_sqlalchemy import SQLAlchemy
+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()
+
+# Association table for room members
+room_members = db.Table('room_members',
+    db.Column('room_id', db.Integer, db.ForeignKey('room.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)
+    last_name = db.Column(db.String(150), nullable=False, default='(You)')
+    email = db.Column(db.String(150), unique=True, nullable=False)
+    password_hash = db.Column(db.String(256))
+    is_admin = db.Column(db.Boolean, default=False)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    phone = db.Column(db.String(20))
+    company = db.Column(db.String(100))
+    position = db.Column(db.String(100))
+    notes = db.Column(db.Text)
+    is_active = db.Column(db.Boolean, default=True)
+    profile_picture = db.Column(db.String(255))
+    preferred_view = db.Column(db.String(10), default='grid', nullable=False)  # 'grid' or 'list'
+    room_permissions = relationship('RoomMemberPermission', back_populates='user')
+
+    def set_password(self, password):
+        self.password_hash = generate_password_hash(password)
+
+    def check_password(self, password):
+        return check_password_hash(self.password_hash, password)
+
+    def __repr__(self):
+        return f'<User {self.username}>'
+
+class Room(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_rooms', foreign_keys=[created_by])
+    members = db.relationship('User', secondary=room_members, backref=db.backref('rooms', lazy='dynamic'))
+    member_permissions = relationship('RoomMemberPermission', back_populates='room', cascade='all, delete-orphan')
+
+    def __repr__(self):
+        return f'<Room {self.name}>'
+
+# Association table for room members with permissions
+class RoomMemberPermission(db.Model):
+    __tablename__ = 'room_member_permissions'
+    room_id = db.Column(db.Integer, db.ForeignKey('room.id'), primary_key=True)
+    user_id = db.Column(db.Integer, db.ForeignKey('user.id'), primary_key=True)
+    can_view = db.Column(db.Boolean, default=True, nullable=False)
+    can_download = db.Column(db.Boolean, default=False, nullable=False)
+    can_upload = db.Column(db.Boolean, default=False, nullable=False)
+    can_delete = db.Column(db.Boolean, default=False, nullable=False)
+    can_rename = db.Column(db.Boolean, default=False, nullable=False)
+    can_move = db.Column(db.Boolean, default=False, nullable=False)
+    can_share = db.Column(db.Boolean, default=False, nullable=False)
+    # Relationships
+    user = relationship('User', back_populates='room_permissions')
+    room = relationship('Room', back_populates='member_permissions')
+
+class RoomFile(db.Model):
+    __tablename__ = 'room_file'
+    id = db.Column(db.Integer, primary_key=True)
+    room_id = db.Column(db.Integer, db.ForeignKey('room.id'), nullable=False)
+    name = db.Column(db.String(255), nullable=False)
+    path = db.Column(db.String(255), nullable=False, default='')
+    type = db.Column(db.String(10), nullable=False)  # 'file' or 'folder'
+    size = db.Column(db.Integer)  # in bytes, null for folders
+    modified = db.Column(db.Float)  # timestamp
+    uploaded_by = db.Column(db.Integer, db.ForeignKey('user.id'))
+    uploaded_at = db.Column(db.DateTime, default=datetime.utcnow)
+    deleted = db.Column(db.Boolean, default=False)  # New field for deleted status
+    deleted_by = db.Column(db.Integer, db.ForeignKey('user.id'))  # New field for tracking who deleted the file
+    deleted_at = db.Column(db.DateTime)  # New field for tracking when the file was deleted
+    uploader = db.relationship('User', backref='uploaded_files', foreign_keys=[uploaded_by])
+    deleter = db.relationship('User', backref='deleted_room_files', foreign_keys=[deleted_by])
+    room = db.relationship('Room', backref='files')
+    starred_by = db.relationship('User', secondary='user_starred_file', backref='starred_files')
+
+    def __repr__(self):
+        return f'<RoomFile {self.name} ({self.type}) in {self.path}>'
+
+class UserStarredFile(db.Model):
+    __tablename__ = 'user_starred_file'
+    id = db.Column(db.Integer, primary_key=True)
+    user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
+    file_id = db.Column(db.Integer, db.ForeignKey('room_file.id'), nullable=False)
+    starred_at = db.Column(db.DateTime, default=datetime.utcnow)
+    
+    # Add unique constraint to prevent duplicate stars
+    __table_args__ = (
+        db.UniqueConstraint('user_id', 'file_id', name='unique_user_file_star'),
+    )
+
+    def __repr__(self):
+        return f'<UserStarredFile user_id={self.user_id} file_id={self.file_id}>'
+
+class TrashedFile(db.Model):
+    __tablename__ = 'trashed_file'
+    id = db.Column(db.Integer, primary_key=True)
+    room_id = db.Column(db.Integer, db.ForeignKey('room.id'), nullable=False)
+    name = db.Column(db.String(255), nullable=False)
+    original_path = db.Column(db.String(255), nullable=False, default='')
+    type = db.Column(db.String(10), nullable=False)  # 'file' or 'folder'
+    size = db.Column(db.Integer)  # in bytes, null for folders
+    modified = db.Column(db.Float)  # timestamp
+    uploaded_by = db.Column(db.Integer, db.ForeignKey('user.id'))
+    uploaded_at = db.Column(db.DateTime, default=datetime.utcnow)
+    deleted_by = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
+    deleted_at = db.Column(db.DateTime, default=datetime.utcnow)
+    room = db.relationship('Room', backref='trashed_files')
+    uploader = db.relationship('User', foreign_keys=[uploaded_by], backref='uploaded_trashed_files')
+    deleter = db.relationship('User', foreign_keys=[deleted_by], backref='deleted_trashed_files')  # Changed from deleted_files to deleted_trashed_files
+
+    def __repr__(self):
+        return f'<TrashedFile {self.name} ({self.type}) from {self.original_path}>'