from sqlalchemy.orm import Session
from server.models.mail import mail
from server.models.contact import Contact
from datetime import datetime

def get_all_mails(db: Session):
    return db.query(mail).all()

def get_top_n_mails(db: Session, topk: int):
    return db.query(mail).limit(topk).all()

def get_mails_by_contact_id(db: Session, contact_id: int):
    return db.query(mail).filter(mail.contactId == contact_id).all()

def get_contact_id_by_email(db: Session, email: str):
    contact = db.query(Contact).filter(Contact.email == email).first()
    return contact.id if contact else None

def get_mails_by_date_range(db: Session, start_date: datetime, end_date: datetime):                         ### format (2022, 4, 25)
    return db.query(mail).filter(mail.date.between(start_date, end_date)).all()