Initial commit

Controllers/Login.py

@@ -0,0 +1,41 @@
+import datetime
+from flask import request, jsonify
+from flask_restful import Resource, abort
+from Models.User import User
+from app import app, jwt
+from flask_jwt_extended import create_access_token, get_jwt_identity, jwt_required, JWTManager, current_user, create_refresh_token
+
+# Register a callback function that takes whatever object is passed in as the
+# identity when creating JWTs and converts it to a JSON serializable format.
+@jwt.user_identity_loader
+def user_identity_lookup(user):
+    return user.id
+
+
+# Register a callback function that loads a user from your database whenever
+# a protected route is accessed. This should return any python object on a
+# successful lookup, or None if the lookup failed for any reason (for example
+# if the user has been deleted from the database).
+@jwt.user_lookup_loader
+def user_lookup_callback(_jwt_header, jwt_data):
+    identity = jwt_data["sub"]
+    return User.query.filter_by(id=identity).one_or_none()
+
+class Login(Resource):
+    def get(self, ):
+        user = User.query.filter_by(email=request.json['email']).first_or_404()
+
+        if not user or not user.check_password(request.json['password']):
+            abort(401, message='Unauthorized')
+        access_token = create_access_token(identity=user)
+        refresh_token = create_refresh_token(identity=user)
+
+        return jsonify(access_token=access_token, refresh_token=refresh_token)
+
+
+class Refresh(Resource):
+    @jwt_required(refresh=True)
+    def get(self, ):
+        identity = get_jwt_identity()
+        access_token = create_access_token(identity=identity)
+        return jsonify(access_token=access_token)

Controllers/Post.py

@@ -0,0 +1,66 @@
+from flask import request, jsonify
+from Models.Post import Post
+from Models.Tag import Tag
+from Models.Schema import post_schema, posts_schema
+from flask_restful import Resource, abort
+from app import db
+from flask_jwt_extended import create_access_token, get_jwt_identity, jwt_required, JWTManager, current_user
+
+class PostListResource(Resource):
+    @jwt_required()
+    def get(self):
+        posts = Post.query.all()
+        return posts_schema.dump(posts)
+
+    @jwt_required()
+    def post(self):
+        tags_array= []
+        for tag_id in request.json['tags']:
+            tags_array.append(Tag.query.filter_by(id=tag_id).first())
+
+        new_post = Post(
+            title=request.json['title'],
+            content=request.json['content'],
+            author_id=current_user.id,
+            author=current_user,
+            tags=tags_array
+        )
+        db.session.add(new_post)
+        db.session.commit()
+        return post_schema.dump(new_post)
+
+
+class PostResource(Resource):
+    @jwt_required()
+    def get(self, post_id):
+        post = Post.query.get_or_404(post_id)
+        return post_schema.dump(post)
+
+    @jwt_required()
+    def put(self, post_id):
+        post = Post.query.get_or_404(post_id)
+
+        post.title = request.json['title']
+        post.content = request.json['content']
+
+        db.session.commit()
+        return post_schema.dump(post)
+
+    @jwt_required()
+    def patch(self, post_id):
+        post = Post.query.get_or_404(post_id)
+
+        if 'title' in request.json:
+            post.title = request.json['title']
+        if 'content' in request.json:
+            post.content = request.json['content']
+
+        db.session.commit()
+        return post_schema.dump(post)
+
+    @jwt_required()
+    def delete(self, post_id):
+        post = Post.query.get_or_404(post_id)
+        db.session.delete(post)
+        db.session.commit()
+        return '', 204

Controllers/Tag.py

@@ -0,0 +1,22 @@
+from flask import request
+from Models.Schema import tag_schema, tags_schema
+from Models.Tag import Tag
+from flask_restful import Resource, abort
+from app import db
+from werkzeug.security import generate_password_hash
+from flask_jwt_extended import create_access_token, get_jwt_identity, jwt_required, JWTManager, current_user
+
+class TagListResource(Resource):
+    @jwt_required()
+    def get(self):
+        tags = Tag.query.all()
+        return tags_schema.dump(tags)
+
+    @jwt_required()
+    def post(self):
+        new_tag = Tag(
+            name=request.json['name']
+        )
+        db.session.add(new_tag)
+        db.session.commit()
+        return tag_schema.dump(new_tag)

Controllers/User.py

@@ -0,0 +1,63 @@
+from flask import request
+from Models.User import User
+from Models.Schema import user_schema, users_schema
+from flask_restful import Resource, abort
+from app import db
+from werkzeug.security import generate_password_hash
+from flask_jwt_extended import create_access_token, get_jwt_identity, jwt_required, JWTManager, current_user
+
+class UserListResource(Resource):
+    @jwt_required()
+    def get(self):
+        users = User.query.all()
+        return users_schema.dump(users)
+
+    @jwt_required()
+    def post(self):
+        new_user = User(
+            name=request.json['name'],
+            email=request.json['email'],
+            password=generate_password_hash(request.json['password'])
+        )
+        db.session.add(new_user)
+        db.session.commit()
+        return user_schema.dump(new_user)
+
+
+class UserResource(Resource):
+    @jwt_required()
+    def get(self, user_id):
+        user = User.query.get_or_404(user_id)
+        return user_schema.dump(post)
+
+    @jwt_required()
+    def put(self, user_id):
+        user = User.query.get_or_404(user_id)
+
+        user.name = request.json['name']
+        user.email = request.json['email']
+        user.password = generate_password_hash(request.json['password'])
+
+        db.session.commit()
+        return user_schema.dump(post)
+
+    @jwt_required()
+    def patch(self, user_id):
+        user = User.query.get_or_404(user_id)
+
+        if 'name' in request.json:
+            user.name = request.json['name']
+        if 'email' in request.json:
+            user.email = request.json['email']
+        if 'password' in request.json:
+            user.password = generate_password_hash(request.json['password'])
+
+        db.session.commit()
+        return user_schema.dump(post)
+
+    @jwt_required()
+    def delete(self, user_id):
+        user = User.query.get_or_404(user_id)
+        db.session.delete(user)
+        db.session.commit()
+        return '', 204