Posts model, view, orm 적용

Author: msnodeve

app/__init__.py

@@ -4,6 +4,7 @@
 
 from flask import Flask
 from app.api.database import DB, MA
+from app.api import REST_API
 from app.constants import SQLALCHEMY_DATABASE_URI_FORMAT
 
 
@@ -17,11 +18,7 @@ def create_app()->(Flask):
     app.config['DEBUG'] = True
 
     DB.init_app(app)
+    REST_API.init_app(app)
     MA.init_app(app)
-
-    @app.route('/')
-    def root():
-        """ main page """
-        return "Hello World!"
-
+    
     return app

app/api/__init__.py

@@ -0,0 +1,12 @@
+"""
+    API config file
+"""
+
+from flask_restplus import Api
+from app.users.views import API as users_api
+from app.posts.views import API as posts_api
+
+REST_API = Api()
+
+REST_API.add_namespace(users_api, '/user')
+REST_API.add_namespace(posts_api, '/post')

app/posts/__init__.py

@@ -0,0 +1,33 @@
+"""
+    Posts model file
+"""
+
+from app.api.database import DB, MA
+from marshmallow import Schema, fields, validate
+from app.users.models import Users, UsersSchema
+from sqlalchemy.sql import text
+
+class Posts(DB.Model):
+    __tablename__ = 'posts'
+    __table_args__ = {'mysql_collate': 'utf8_general_ci'}
+
+    id = DB.Column(DB.Integer, primary_key=True)
+    author_id = DB.Column(DB.String(255), DB.ForeignKey(Users.user_id))
+    title = DB.Column(DB.String(512), nullable=False)
+    body = DB.Column(DB.String(1024), nullable=False)
+    author = DB.relationship('Users', uselist=False)
+    created = DB.Column(DB.TIMESTAMP, server_default=text("CURRENT_TIMESTAMP"), nullable=False)
+
+    def __init__(self, author_id, title, body):
+        self.author_id = author_id
+        self.title = title
+        self.body = body
+
+class PostsSchema(MA.Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+    id = fields.Integer()
+    author_id = fields.String(validate=not_blank)
+    title = fields.String(validate=not_blank)
+    body = fields.String(validate=not_blank)
+    author = fields.Nested(UsersSchema)
+    created = fields.String(validate=not_blank)

app/posts/models.py

@@ -0,0 +1,66 @@
+"""
+    Posts view file
+"""
+
+from flask_restplus import Namespace, Resource, reqparse, fields
+from flask import jsonify, make_response
+from http import HTTPStatus
+from sqlalchemy.exc import SQLAlchemyError
+from app.posts.models import Posts, PostsSchema
+from app.users.models import Users, UsersSchema
+from app.api.database import DB
+
+API = Namespace('Posts', description="Post's REST API")
+POSTS_SCHEMA = PostsSchema()
+
+@API.route('s')
+class Post(Resource):
+    parser = reqparse.RequestParser()
+    parser.add_argument('author_id', required=True, type=str, help="Post's author ID", location='json')
+    parser.add_argument('title', required=True, type=str, help="Post's title", location='json')
+    parser.add_argument('body', required=True, type=str, help="Post's body", location='json')
+
+    post_field = API.model('Post', {
+        'author_id': fields.String,
+        'title': fields.String,
+        'body': fields.String
+    })
+
+    @API.doc('get')
+    def get(self):
+        try:
+            posts = Posts.query.all()
+            body = jsonify(POSTS_SCHEMA.dump(posts, many=True).data)
+            code = HTTPStatus.OK
+        except SQLAlchemyError as err:
+            body = jsonify({'message' : str(err)})
+            code = HTTPStatus.INTERNAL_SERVER_ERROR
+        return make_response(body, code.value)
+
+    @API.expect(post_field)
+    @API.doc('post')
+    def post(self):
+        args_ = self.parser.parse_args()
+        post = Posts(author_id=args_['author_id'], title=args_['title'], body=args_['body'])
+        try:
+            DB.session.add(post)
+            DB.session.commit()
+            body = jsonify({'post', POSTS_SCHEMA.dump(post).data})
+            code = HTTPStatus.OK
+        except SQLAlchemyError as err:
+            body = jsonify({'message' : str(err)})
+            code = HTTPStatus.INTERNAL_SERVER_ERROR
+        return make_response(body, code.value)
+
+@API.route('/<int:reqno>')
+class PostItem(Resource):
+    def get(self, reqno):
+        try:
+            post = DB.session.query(Posts).outerjoin(
+                Users, Users.user_id == Posts.author_id).filter(Posts.id==reqno).first()
+            body = jsonify({'post' : POSTS_SCHEMA.dump(post).data})
+            code = HTTPStatus.OK
+        except SQLAlchemyError as err:
+            body = jsonify({'message' : str(err)})
+            code = HTTPStatus.INTERNAL_SERVER_ERROR
+        return make_response(body, code.value)

app/posts/views.py

@@ -0,0 +1,65 @@
+"""
+    User views file
+"""
+from http import HTTPStatus
+from flask import jsonify, make_response
+from sqlalchemy.exc import SQLAlchemyError
+from flask_restplus import Namespace, Resource, reqparse, fields
+from app.users.models import Users, UsersSchema
+from app.api.database import DB
+
+API = Namespace('Users', description="User's RESTPlus - API")
+USERS_SCHEMA = UsersSchema()
+
+@API.route('s')
+class UsersAuth(Resource):
+    parser = reqparse.RequestParser()
+    parser.add_argument('user_id', required=True, type=str, help="User's ID", location='json')
+    parser.add_argument('user_password', required=True, type=str, help="User's PW", location='json')
+    parser.add_argument('user_email', required=True, type=str, help="User's Email", location='json')
+
+    users_field = API.model('Sign up', {
+        'user_id' : fields.String,
+        'user_password' : fields.String,
+        'user_email' : fields.String
+    })
+
+    @API.doc('post')
+    @API.expect(users_field)
+    def post(self):
+        args_ = self.parser.parse_args()
+        user = Users(args_['user_id'], args_['user_password'], args_['user_email'])
+        try:
+            DB.session.add(user)
+            DB.session.commit()
+            body = jsonify({'users' : USERS_SCHEMA.dump(user).data})
+            code = HTTPStatus.OK
+        except SQLAlchemyError as err:
+            DB.session.rollback()
+            body = jsonify({'message' : str(err)})
+            code = HTTPStatus.INTERNAL_SERVER_ERROR
+        return make_response(body, code.value)
+
+@API.route('/auth')
+class UserAuth(Resource):
+    parser = reqparse.RequestParser()
+    parser.add_argument('user_id', required=True, type=str, help="User's ID", location='json')
+    parser.add_argument('user_password', required=True, type=str, help="User's PW", location='json')
+
+    user_login_field = API.model('Sign in', {
+        'user_id' : fields.String,
+        'user_password' : fields.String
+    })
+
+    @API.doc('post')
+    @API.expect(user_login_field)
+    def post(self):
+        args_ = self.parser.parse_args()
+        try:
+            user = Users.query.filter(Users.user_id == args_['user_id']).first()
+            body = jsonify({'user_id' : user.user_id})
+            code = HTTPStatus.OK
+        except SQLAlchemyError as err:
+            body = jsonify({'message' : str(err)})
+            code = HTTPStatus.INTERNAL_SERVER_ERROR
+        return make_response(body, code.value)

app/tests/__init__.py

@@ -7,6 +7,7 @@
 from app import create_app
 from app.api.database import DB
 from app.users.models import Users, UsersSchema
+from app.posts.models import Posts, PostsSchema
 
 APP = create_app()
 MANAGER = Manager(APP)

app/users/views.py

@@ -0,0 +1,37 @@
+"""empty message
+
+Revision ID: bb32b366dab2
+Revises: d8019933f2c3
+Create Date: 2019-07-23 15:07:27.216578
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'bb32b366dab2'
+down_revision = 'd8019933f2c3'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('posts',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('author_id', sa.String(length=255), nullable=True),
+    sa.Column('title', sa.String(length=512), nullable=False),
+    sa.Column('body', sa.String(length=1024), nullable=False),
+    sa.Column('created', sa.TIMESTAMP(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=False),
+    sa.ForeignKeyConstraint(['author_id'], ['users.user_id'], ),
+    sa.PrimaryKeyConstraint('id'),
+    mysql_collate='utf8_general_ci'
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('posts')
+    # ### end Alembic commands ###

manage.py

@@ -1,16 +1,16 @@
 """empty message
 
-Revision ID: 5ceec2627296
+Revision ID: d8019933f2c3
 Revises: 
-Create Date: 2019-07-23 10:16:27.384161
+Create Date: 2019-07-23 12:06:26.634913
 
 """
 from alembic import op
 import sqlalchemy as sa
 
 
 # revision identifiers, used by Alembic.
-revision = '5ceec2627296'
+revision = 'd8019933f2c3'
 down_revision = None
 branch_labels = None
 depends_on = None