class UserLogout(Resource):
"""docstring for UserLogout."""
def __init__(self, arg):
super(UserLogout, self).__init__()
self.blacklist = BlacklistHelper()
@jwt_required
def post(self):
""" User logout """
current_user = get_jwt_identity()
code, msg = self.blacklist.revoke_token(current_user)
return {"status": "success", "msg": msg}, code
def __init__(self):
super(UserService, self).__init__()
self.collection = "users"
self.blacklist = BlacklistHelper()
self.utils = Utils()
self.mongo = MongoDB()
class UserService:
""" doc string for UserService """
def __init__(self):
super(UserService, self).__init__()
self.collection = "users"
self.blacklist = BlacklistHelper()
self.utils = Utils()
self.mongo = MongoDB()
def user_list(self):
users = self.mongo.find(self.collection)
if users:
for user in users:
del user["password"]
return users
else:
return []
def add_user(self, user_obj):
""" user_obj - user object """
user = self.mongo.find(self.collection, {"email": user_obj["email"]})
if not user:
return self.mongo.save(self.collection, user_obj)
else:
return f'User with {user_obj["email"]} already existed.'
def get_user(self, user_id):
""" Get User profile by id. ex _id: """
res = self.mongo.find_by_id(self.collection, user_id)
if res:
del res["password"]
return ("success", res, "ok", 200)
else:
return ("error", [], "Something went wrong.", 400)
def update_user(self, _id, user_obj):
user = self.mongo.find(self.collection, {"email": user_obj["email"]})
if not user:
query = {"$set": user_obj}
res, res_obj = self.mongo.update(self.collection, _id, query)
if res:
del res_obj["password"]
return ("success", res_obj, "ok", 200)
else:
return ("error", "", "Something went wrong.", 400)
else:
return (
"error",
"",
f'Email {user_obj["email"]} address already in use.',
400,
)
def delete_user(self, user_id):
return self.mongo.delete(self.collection, user_id)
def login(self, email):
""" email as input """
user = self.mongo.find(self.collection, {"email": email})
if user:
user = user[0]
return user
else:
return None
def save_tokens(self, user_tokens):
self.blacklist.add_token_to_database(user_tokens["access"],
app.config["JWT_IDENTITY_CLAIM"])
self.blacklist.add_token_to_database(user_tokens["refresh"],
app.config["JWT_IDENTITY_CLAIM"])
def __init__(self, arg):
super(UserRegister, self).__init__(arg)
self.jwt_service = JWTService()
self.blacklist = BlacklistHelper()
self.utils = Utils()
self.user_service = UserService()
def __init__(self, arg):
super(UserLogout, self).__init__()
self.blacklist = BlacklistHelper()
class UserService():
""" doc string for UserService """
def __init__(self):
super(UserService, self).__init__()
self.collection = 'users'
self.blacklist = BlacklistHelper()
self.utils = Utils()
self.mongo = MongoDB()
def user_list(self):
users = self.mongo.find(self.collection)
if users:
for user in users:
del user['password']
return users
else:
return []
def add_user(self, user_obj):
""" user_obj - user object """
user = self.mongo.find(self.collection, {'email': user_obj['email']})
if not user:
return self.mongo.save(self.collection, user_obj)
else:
return f'User with {user_obj["email"]} already existed.'
def get_user(self, user_id):
""" Get User profile by id. ex _id: """
res = self.mongo.find_by_id(self.collection, user_id)
if res:
del res['password']
return ('success', res, 'ok', 200)
else:
return ('error', [], 'Something went wrong.', 400)
def update_user(self, _id, user_obj):
user = self.mongo.find(self.collection, {'email': user_obj['email']})
if not user:
query = {'$set': user_obj}
res, res_obj = self.mongo.update(self.collection, _id, query)
if res:
del res_obj['password']
return ('success', res_obj, 'ok', 200)
else:
return ('error', '', 'Something went wrong.', 400)
else:
return ('error', '',
f'Email {user_obj["email"]} address already in use.', 400)
def delete_user(self, user_id):
return self.mongo.delete(self.collection, user_id)
def login(self, email):
""" email as input """
user = self.mongo.find(self.collection, {'email': email})
if user:
user = user[0]
return user
else:
return None
def save_tokens(self, user_tokens):
self.blacklist.add_token_to_database(user_tokens['access'],
app.config['JWT_IDENTITY_CLAIM'])
self.blacklist.add_token_to_database(user_tokens['refresh'],
app.config['JWT_IDENTITY_CLAIM'])