Browse Source

Fixing creation_time errors

master
Drew Short 5 years ago
parent
commit
927bd73fcb
  1. 2
      db.py
  2. 8
      models.py

2
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)

8
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':

Loading…
Cancel
Save