login

app/models.py

@@ -1,19 +1,36 @@
 from sqlalchemy.dialects.postgresql import UUID
 from sqlalchemy.orm import declarative_base
 from flask_sqlalchemy import SQLAlchemy
+from werkzeug.security import generate_password_hash, check_password_hash
+from datetime import datetime
+from flask_login import UserMixin
 
 import uuid
 
 Base = declarative_base()
 db = SQLAlchemy(model_class=Base)
 
-class User(Base):
+class User(Base, UserMixin):
     __tablename__ = 'users'
     id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    first_name = db.Column(db.String(255), nullable=False)
+    last_name = db.Column(db.String(255), nullable=False)
     email = db.Column(db.String(255), unique=True, nullable=False)
-    # Placeholders for Milestone 1
-    password_hash = db.Column(db.LargeBinary) 
+    password_hash = db.Column(db.String(2048), nullable=False)
     imap_config = db.Column(db.JSON) # Using db.JSON instead of db.JSONB for compatibility
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+    def set_password(self, password):
+        """Hash and set the password."""
+        self.password_hash = generate_password_hash(password)
+    
+    def check_password(self, password):
+        """Check if the provided password matches the stored hash."""
+        return check_password_hash(self.password_hash, password)
+    
+    def __repr__(self):
+        return f'<User {self.first_name} {self.last_name} ({self.email})>'
 
 class Folder(Base):
     __tablename__ = 'folders'