Merge pull request #17 from schoolofcode-me/added_user_resource

Added a user resource to show user info and to delete user by id.

Author: jslvtr

section11/app.py

@@ -3,7 +3,7 @@
 from flask_jwt_extended import JWTManager
 
 from db import db
-from resources.user import UserRegister, UserLogin, TokenRefresh
+from resources.user import UserRegister, UserLogin, User, TokenRefresh
 from resources.item import Item, ItemList
 from resources.store import Store, StoreList
 
@@ -99,6 +99,7 @@ def create_tables():
 api.add_resource(ItemList, '/items')
 api.add_resource(UserRegister, '/register')
 api.add_resource(UserLogin, '/login')
+api.add_resource(User, '/user/<int:user_id>')
 api.add_resource(TokenRefresh, '/refresh')
 
 if __name__ == '__main__':

section11/models/user.py

@@ -12,6 +12,12 @@ def __init__(self, username, password):
         self.username = username
         self.password = password
 
+    def json(self):
+        return {
+            'id': self.id,
+            'username': self.username
+        }
+
     def save_to_db(self):
         db.session.add(self)
         db.session.commit()

section11/resources/user.py

@@ -58,6 +58,27 @@ def post(self):
         return {"message": "Invalid Credentials!"}, 401
 
 
+class User(Resource):
+    """
+    This resource can be useful when testing our Flask app. We may not want to expose it to public users, but for the
+    sake of demonstration in this course, it can be useful when we are manipulating data regarding the users.
+    """
+    @classmethod
+    def get(cls, user_id: int):
+        user = UserModel.find_by_id(user_id)
+        if not user:
+            return {'message': 'User Not Found'}, 404
+        return {'user': user.json()}, 200
+
+    @classmethod
+    def delete(cls, user_id: int):
+        user = UserModel.find_by_id(user_id)
+        if not user:
+            return {'message': 'User Not Found'}, 404
+        user.delete_from_db()
+        return {'message': 'User deleted.'}, 200
+
+
 class TokenRefresh(Resource):
     @jwt_refresh_token_required
     def post(self):