From 927bd73fcb0b74bd3d7b144466e431901ef70449 Mon Sep 17 00:00:00 2001 From: Drew Short Date: Wed, 30 Jan 2019 20:25:18 -0600 Subject: [PATCH] Fixing creation_time errors --- db.py | 2 +- models.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/db.py b/db.py index d2d7e0e..c0dc682 100644 --- a/db.py +++ b/db.py @@ -59,7 +59,7 @@ def get_registration_code(code: str) -> Optional[RegistrationCode]: def add_registration_code( - expiration_time: datetime = datetime.now(), + expiration_time: datetime = None, max_usages: int = 1) -> RegistrationCode: code = RegistrationCode( uuid.uuid4().__str__(), expiration_time=expiration_time, max_usages=max_usages) diff --git a/models.py b/models.py index a20fc43..1018846 100644 --- a/models.py +++ b/models.py @@ -5,12 +5,12 @@ from typing import Tuple class RegistrationCode: def __init__(self, code: str, - creation_time: datetime = datetime.now(), + creation_time: datetime = None, expiration_time: datetime = None, usages: int = 0, max_usages: int = 1): self.code = code - self.creation_time = creation_time + self.creation_time = creation_time if creation_time is not None else datetime.now() self.expiration_time = expiration_time self.usages = usages self.max_usages = max_usages @@ -37,10 +37,10 @@ class RegisteredUser: def __init__(self, registration_code: str, username: str, - registered_time: datetime = datetime.now()): + registered_time: datetime = None): self.registration_code = registration_code self.username = username - self.registered_time = registered_time + self.registered_time = registered_time if registered_time is not None else datetime.now() @staticmethod def from_db(db_registered_user: Tuple) -> 'RegisteredUser':