TechEmpower.FrameworkBenchmarks/frameworks/Python/cherrypy/app.py

import os
import sys
from functools import partial
from operator import attrgetter
from random import randint
import json

import cherrypy
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column
from sqlalchemy.types import String, Integer

Base = declarative_base()

if sys.version_info[0] == 3:
  xrange = range

def getQueryNum(queryString):
  try:
    int(queryString)
    return int(queryString)
  except ValueError:
    return 1

class Fortune(Base):
  __tablename__ = "fortune"

  id = Column(Integer, primary_key = True)
  message = Column(String)

class World(Base):
  __tablename__ = "world"

  id = Column(Integer, primary_key = True)
  randomNumber = Column(Integer)

  def serialize(self):
    return {
      'id' : self.id,
      'randomNumber' : self.randomNumber
    }

class CherryPyBenchmark(object):

  @cherrypy.expose
  @cherrypy.tools.json_out()
  def json(self):
    cherrypy.response.headers["Content-Type"] = "application/json"
    json_message = {"message" : "Hello, world!"}
    return json_message


  @cherrypy.expose
  def plaintext(self):
    return "Hello, world!"

  @cherrypy.expose
  @cherrypy.tools.json_out()
  def db(self):
    cherrypy.response.headers["Content-Type"] = "application/json"
    wid = randint(1, 10000)
    world = cherrypy.request.db.query(World).get(wid).serialize()
    return world

  @cherrypy.expose
  @cherrypy.tools.json_out()
  def queries(self, queries=1):
    num_queries = getQueryNum(queries)
    if num_queries < 1:
      num_queries = 1
    if num_queries > 500:
      num_queries = 500

    rp = partial(randint, 1, 10000)
    get = cherrypy.request.db.query(World).get
    worlds = [get(rp()).serialize() for _ in xrange(num_queries)]
    return worlds

  @cherrypy.expose
  @cherrypy.tools.json_out()
  def updates(self, queries=1):
    cherrypy.response.headers["Content-Type"] = "application/json"
    num_queries = getQueryNum(queries)
    if num_queries < 1:
      num_queries = 1
    if num_queries > 500:
      num_queries = 500

    worlds = []
    rp = partial(randint, 1, 10000)
    ids = [rp() for _ in xrange(num_queries)]
    ids.sort() # To avoid deadlock
    for id in ids:
      world = cherrypy.request.db.query(World).get(id)
      world.randomNumber = rp()
      worlds.append(world.serialize())
    return worlds

if __name__ == "__main__":
  # Register the SQLAlchemy plugin
  from saplugin import SAEnginePlugin
  DBDRIVER = 'mysql'
  DBHOSTNAME = os.environ.get('DBHOST', 'localhost')  
  DATABASE_URI = '%s://benchmarkdbuser:benchmarkdbpass@%s:3306/hello_world?charset=utf8' % (DBDRIVER, DBHOSTNAME)
  SAEnginePlugin(cherrypy.engine, DATABASE_URI).subscribe()

  # Register the SQLAlchemy tool
  from satool import SATool
  cherrypy.tools.db = SATool()
  cherrypy.quickstart(CherryPyBenchmark(), '', {'/': {'tools.db.on': True}})