Added UserModel.delete_from_db(), modified get user by id response format.

Author: CristianoYL

section11/models/user.py

@@ -18,14 +18,18 @@ def json(self):
             'username': self.username
         }
 
-    def save_to_db(self):
-        db.session.add(self)
-        db.session.commit()
-
     @classmethod
     def find_by_username(cls, username):
         return cls.query.filter_by(username=username).first()
 
     @classmethod
     def find_by_id(cls, _id):
         return cls.query.filter_by(id=_id).first()
+
+    def save_to_db(self):
+        db.session.add(self)
+        db.session.commit()
+
+    def delete_from_db(self):
+        db.session.delete(self)
+        db.session.commit()

section11/resources/user.py

@@ -72,7 +72,7 @@ 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
+        return user.json(), 200
 
     @classmethod
     def delete(cls, user_id: int):