diff options
Diffstat (limited to 'models')
| -rw-r--r-- | models/user.py | 14 | 
1 files changed, 13 insertions, 1 deletions
| diff --git a/models/user.py b/models/user.py index ea90f39..4d4887f 100644 --- a/models/user.py +++ b/models/user.py @@ -1,8 +1,9 @@ +"""Define a user."""  from os import urandom  from itsdangerous import URLSafeTimedSerializer -from common.security import pwd_context, secret_key +from common.security import pwd_context  class User(object): @@ -57,3 +58,14 @@ class User(object):          else:              raise ValueError("No user") + +    def to_dict(self): +        """Return a dict representation of the user.""" +        return_dict = dict() +        for key, item in self.__dict__.items(): +            try: +                return_dict[key] = item.to_dict() +            except AttributeError: +                return_dict[key] = item + +        return return_dict | 
