110 lines
2.7 KiB
Python
110 lines
2.7 KiB
Python
|
from dataclasses import dataclass
|
||
|
|
|
||
|
|
from sqlalchemy import or_
|
||
|
|
from sqlalchemy.orm import joinedload
|
||
|
|
|
||
|
|
from app.config import PAGE_LIMIT
|
||
|
|
from app.extensions import db
|
||
|
|
|
||
|
|
from app.models import Alias, Mailbox, Contact, EmailLog
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class AliasInfo:
|
||
|
|
alias: Alias
|
||
|
|
|
||
|
|
nb_forward: int
|
||
|
|
nb_blocked: int
|
||
|
|
nb_reply: int
|
||
|
|
|
||
|
|
|
||
|
|
def serialize_alias_info(alias_info: AliasInfo) -> dict:
|
||
|
|
return {
|
||
|
|
# Alias field
|
||
|
|
"id": alias_info.alias.id,
|
||
|
|
"email": alias_info.alias.email,
|
||
|
|
"creation_date": alias_info.alias.created_at.format(),
|
||
|
|
"creation_timestamp": alias_info.alias.created_at.timestamp,
|
||
|
|
"enabled": alias_info.alias.enabled,
|
||
|
|
"note": alias_info.alias.note,
|
||
|
|
# activity
|
||
|
|
"nb_forward": alias_info.nb_forward,
|
||
|
|
"nb_block": alias_info.nb_blocked,
|
||
|
|
"nb_reply": alias_info.nb_reply,
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def serialize_contact(contact: Contact) -> dict:
|
||
|
|
res = {
|
||
|
|
"id": contact.id,
|
||
|
|
"creation_date": contact.created_at.format(),
|
||
|
|
"creation_timestamp": contact.created_at.timestamp,
|
||
|
|
"last_email_sent_date": None,
|
||
|
|
"last_email_sent_timestamp": None,
|
||
|
|
"contact": contact.website_email,
|
||
|
|
"reverse_alias": contact.website_send_to(),
|
||
|
|
}
|
||
|
|
|
||
|
|
email_log: EmailLog = contact.last_reply()
|
||
|
|
if email_log:
|
||
|
|
res["last_email_sent_date"] = email_log.created_at.format()
|
||
|
|
res["last_email_sent_timestamp"] = email_log.created_at.timestamp
|
||
|
|
|
||
|
|
return res
|
||
|
|
|
||
|
|
|
||
|
|
def get_alias_infos_with_pagination(user, page_id=0, query=None) -> [AliasInfo]:
|
||
|
|
ret = []
|
||
|
|
q = (
|
||
|
|
db.session.query(Alias)
|
||
|
|
.filter(Alias.user_id == user.id)
|
||
|
|
.order_by(Alias.created_at.desc())
|
||
|
|
)
|
||
|
|
|
||
|
|
if query:
|
||
|
|
q = q.filter(
|
||
|
|
or_(Alias.email.ilike(f"%{query}%"), Alias.note.ilike(f"%{query}%"))
|
||
|
|
)
|
||
|
|
|
||
|
|
q = q.limit(PAGE_LIMIT).offset(page_id * PAGE_LIMIT)
|
||
|
|
|
||
|
|
for alias in q:
|
||
|
|
ret.append(get_alias_info(alias))
|
||
|
|
|
||
|
|
return ret
|
||
|
|
|
||
|
|
|
||
|
|
def get_alias_info(alias: Alias) -> AliasInfo:
|
||
|
|
q = (
|
||
|
|
db.session.query(Contact, EmailLog)
|
||
|
|
.filter(Contact.alias_id == alias.id)
|
||
|
|
.filter(EmailLog.contact_id == Contact.id)
|
||
|
|
)
|
||
|
|
|
||
|
|
alias_info = AliasInfo(alias=alias, nb_blocked=0, nb_forward=0, nb_reply=0,)
|
||
|
|
|
||
|
|
for _, el in q:
|
||
|
|
if el.is_reply:
|
||
|
|
alias_info.nb_reply += 1
|
||
|
|
elif el.blocked:
|
||
|
|
alias_info.nb_blocked += 1
|
||
|
|
else:
|
||
|
|
alias_info.nb_forward += 1
|
||
|
|
|
||
|
|
return alias_info
|
||
|
|
|
||
|
|
|
||
|
|
def get_alias_contacts(alias, page_id: int) -> [dict]:
|
||
|
|
q = (
|
||
|
|
Contact.query.filter_by(alias_id=alias.id)
|
||
|
|
.order_by(Contact.id.desc())
|
||
|
|
.limit(PAGE_LIMIT)
|
||
|
|
.offset(page_id * PAGE_LIMIT)
|
||
|
|
)
|
||
|
|
|
||
|
|
res = []
|
||
|
|
for fe in q.all():
|
||
|
|
res.append(serialize_contact(fe))
|
||
|
|
|
||
|
|
return res
|