Skip to content

Instantly share code, notes, and snippets.

@frankV

frankV/app.py

Forked from mmautner/app.py
Last active February 14, 2019 07:10
Show Gist options
  • Save frankV/b7ab2e3abc75cc3644eb to your computer and use it in GitHub Desktop.
Save frankV/b7ab2e3abc75cc3644eb to your computer and use it in GitHub Desktop.
Download ZIP
Example of caching API results w/ Flask-Restless and Flask-Cache
Raw
app.py
import json
import hashlib
from flask import Flask, request
import flask.ext.sqlalchemy
import flask.ext.cache
import flask.ext.restless
from flask.ext.restless import ProcessingException
app = Flask(__name__)
app.config['DEBUG'] = True
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:////tmp/test.db'
app.config['CACHE_TYPE'] = 'redis'
app.config['CACHE_KEY_PREFIX'] = 'my_key'
db = flask.ext.sqlalchemy.SQLAlchemy(app)
cache = flask.ext.cache.Cache(app)
class Person(db.Model):
id = db.Column(db.Integer, primary_key=True)
db.create_all()
def get_cache_key():
return hashlib.md5(request.path + request.query_string).hexdigest()
def cache_preprocessor(**kwargs):
key = get_cache_key()
if cache.get(key):
print('returning cached result')
raise ProcessingException(description=cache.get(key), code=200)
def cache_postprocessor(result, **kwargs):
cache.set(get_cache_key(), json.dumps(result))
print('result cached.')
manager = flask.ext.restless.APIManager(app, flask_sqlalchemy_db=db)
manager.create_api(Person,
methods=['GET'],
preprocessors={'GET_MANY': [cache_preprocessor]},
postprocessors={'GET_MANY': [cache_postprocessor]})
app.run()
Raw
test.py
import requests
# first time is uncached:
print requests.get('http://localhost:5000/api/person').json()
# second time is cached:
print requests.get('http://localhost:5000/api/person').json()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment