models.py

from app import db
from flask_login import UserMixin
from datetime import datetime
from werkzeug.security import generate_password_hash, check_password_hash

class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    first_name = db.Column(db.String(64), nullable=False)
    middle_name = db.Column(db.String(64))
    surname = db.Column(db.String(64), nullable=False)
    username = db.Column(db.String(64), unique=True, nullable=False)
    email = db.Column(db.String(120), unique=True, nullable=False)
    password_hash = db.Column(db.String(256))
    date_of_birth = db.Column(db.Date, nullable=False)
    gender = db.Column(db.String(10), nullable=False)
    profile_image = db.Column(db.String(255))  # Added profile image field
    campaigns = db.relationship('Campaign', backref='creator', lazy=True)

    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)

class Package(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), nullable=False)
    price = db.Column(db.Integer, nullable=False)
    description = db.Column(db.Text)
    features = db.Column(db.JSON)
    duration_months = db.Column(db.Float, nullable=False)
    type = db.Column(db.String(20), nullable=False, default='association')  # 'association', 'business', 'company'
    recommended_for = db.Column(db.JSON, nullable=True)  # Store recommendation criteria as JSON

class Campaign(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(120), nullable=False)
    description = db.Column(db.Text)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
    package_id = db.Column(db.Integer, db.ForeignKey('package.id'), nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    target_amount = db.Column(db.Float, nullable=False)
    current_amount = db.Column(db.Float, default=0)
    invite_link = db.Column(db.String(120), unique=True)
    contributions = db.relationship('Contribution', backref='campaign', lazy=True)
    package = db.relationship('Package')
    # Add field to track last notification milestone
    last_milestone_notified = db.Column(db.Integer, default=0)  # Store the last milestone percentage notified

    @property
    def progress_percentage(self):
        """Calculate the current funding progress as a percentage"""
        if self.target_amount == 0:
            return 0
        return (self.current_amount / self.target_amount) * 100

    def check_and_notify_milestones(self):
        """Check if any new milestones have been reached"""
        from utils.email_service import check_and_notify_milestones
        current_percentage = self.progress_percentage
        check_and_notify_milestones(self, self.last_milestone_notified, current_percentage)
        self.last_milestone_notified = int(current_percentage)
        db.session.commit()

class Contribution(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    amount = db.Column(db.Float, nullable=False)
    campaign_id = db.Column(db.Integer, db.ForeignKey('campaign.id'), nullable=False)
    contributor_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    payment_id = db.Column(db.String(120))
    contributor = db.relationship('User')

class RecommendationCriteria(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    business_size = db.Column(db.String(20), nullable=False)  # 'small', 'medium', 'large'
    online_presence = db.Column(db.Boolean, default=False)
    processing_speed = db.Column(db.String(20))  # 'standard', 'fast', 'express'
    budget_range = db.Column(db.String(20))  # 'low', 'medium', 'high'
    ecommerce_needs = db.Column(db.Boolean, default=False)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=True)