flask-api-example/app.py
Andros Fenollosa 267b9b0a02 Add marshmallow
2018-02-05 23:00:30 +01:00

117 lines
2.3 KiB
Python

# -*- coding: utf-8 -*-
# Librarys
# =========================
import os
from flask import Flask, jsonify, request
from flask_restplus import Resource, Api
from dotenv import load_dotenv, find_dotenv
from models import db, User, News, Comment
from flask_marshmallow import Marshmallow
# Extensions initialization
# =========================
load_dotenv(find_dotenv())
app = Flask(__name__)
ma = Marshmallow(app)
api = Api(app)
# Configurations
# =========================
app.config['SECRET_KEY'] = os.environ.get('SECRET_KEY')
app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get('DATABASE_URI')
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db.init_app(app)
PRE_URL = '/api/v1/'
# Schemas
# =========================
# User
class UserSchema(ma.Schema):
class Meta:
# Fields to expose
fields = ('username', 'mail')
user_schema = UserSchema()
users_schema = UserSchema(many=True)
# Routes
# =========================
# Signup
@api.route(PRE_URL + 'signup')
class Signup(Resource):
def post(self):
return {'hello': 'world'}
# Login
@api.route(PRE_URL + 'login')
class Login(Resource):
def post(self):
return {'hello': 'world'}
# Logout
@api.route(PRE_URL + 'login')
class Logout(Resource):
def get(self):
return {'hello': 'world'}
# User
@api.route(PRE_URL + 'user')
class UserList(Resource):
def get(self):
all_users = User.query.all()
return users_schema.jsonify(all_users)
@api.route(PRE_URL + 'user/<int:id>')
class UserSingle(Resource):
def get(self, id):
all_users = User.query.get(id)
return user_schema.jsonify(all_users)
# News
@api.route(PRE_URL + 'news')
class NewsList(Resource):
def get(self):
my_news = News.query.all()
return serializeQuery(my_news)
def post(self):
return request.form
@api.route(PRE_URL + 'news/<int:id>')
class News_single(Resource):
def get(self, id):
return {'hello': id}
# Comment
@api.route(PRE_URL + 'news/<int:id>/comments')
class Comments(Resource):
def get(self, id):
return {'hello': 'world'}
def post(self, id):
return {'hello': 'world'}
# Run
# =========================
if __name__ == '__main__':
app.run(debug=True if os.environ.get('DEBUG') == 'True' else False)