From 1162495b36a91c85c453c12bf8426aab2e911cb9 Mon Sep 17 00:00:00 2001 From: doanguyen Date: Tue, 31 Dec 2019 20:20:17 +0100 Subject: [PATCH] remove dataclasses completely, remove the slots magic --- app/dashboard/views/alias_log.py | 17 ++++++++--------- app/dashboard/views/index.py | 5 ++++- 2 files changed, 12 insertions(+), 10 deletions(-) diff --git a/app/dashboard/views/alias_log.py b/app/dashboard/views/alias_log.py index f2ed75d3..d7ec6388 100644 --- a/app/dashboard/views/alias_log.py +++ b/app/dashboard/views/alias_log.py @@ -1,4 +1,5 @@ -from flask import render_template, flash, redirect, url_for, abort +import arrow +from flask import render_template, flash, redirect, url_for from flask_login import login_required, current_user from app.dashboard.base import dashboard_bp @@ -9,14 +10,12 @@ _LIMIT = 15 class AliasLog: - __slots__ = [ - "website_email", - "website_from", - "alias", - "when", - "is_reply", - "blocked", - ] # memory efficiency + website_email: str + website_from: str + alias: str + when: arrow.Arrow + is_reply: bool + blocked: bool def __init__(self, **kwargs): for k, v in kwargs.items(): diff --git a/app/dashboard/views/index.py b/app/dashboard/views/index.py index d9899ab3..614e4e9d 100644 --- a/app/dashboard/views/index.py +++ b/app/dashboard/views/index.py @@ -19,7 +19,6 @@ from app.models import ( ) -@dataclass class AliasInfo: gen_email: GenEmail nb_forward: int @@ -29,6 +28,10 @@ class AliasInfo: show_intro_test_send_email: bool = False highlight: bool = False + def __init__(self, **kwargs): + for k, v in kwargs.items(): + setattr(self, k, v) + @dashboard_bp.route("/", methods=["GET", "POST"]) @login_required