# -*- coding: utf-8 -*-
import os
from operator import itemgetter
from gluon.storage import Storage
from gluon.dal import DAL, Field, Row

DBHOST = 'tfb-database'
DATABASE_URI = 'mysql://benchmarkdbuser:benchmarkdbpass@%s:3306/hello_world' % DBHOST

class Dal(object):
    def __init__(self, table=None, pool_size=8):
        self.db = DAL(DATABASE_URI, migrate_enabled=False, pool_size=pool_size)
        if table == 'World':
            self.db.define_table('World', Field('randomNumber', 'integer'))
        elif table == 'Fortune':
            self.db.define_table('Fortune', Field('message'))

    def get_world(self, wid):
        # Setting `cacheable=True` improves performance by foregoing the creation
        # of some non-essential attributes. It does *not* actually cache the
        # database results (it simply produces a Rows object that *could be* cached).
        return self.db(self.db.World.id == wid).select(cacheable=True)[0].as_dict()

    def update_world(self, wid, randomNumber):
        self.db(self.db.World.id == wid).update(randomNumber=randomNumber)

    def get_fortunes(self, new_message):
        fortunes = self.db(self.db.Fortune).select(cacheable=True)
        fortunes.records.append(Row(new_message))
        return fortunes.sort(itemgetter('message'))

class RawDal(Dal):
    def __init__(self):
        super(RawDal, self).__init__()

    def get_world(self, wid):
        return self.db.executesql('SELECT * FROM World WHERE id = %s',
                                  placeholders=[wid], as_dict=True)[0]

    def update_world(self, wid, randomNumber):
        self.db.executesql('UPDATE World SET randomNumber=%s WHERE id=%s',
                           placeholders=[randomNumber, wid])

    def get_fortunes(self, new_message):
        fortunes = self.db.executesql('SELECT * FROM Fortune', as_dict=True)
        fortunes.append(new_message)
        return sorted(fortunes, key=itemgetter('message'))

def num_queries(queries):
    try:
        num = int(queries)
        return 1 if num < 1 else 500 if num > 500 else num
    except ValueError:
         return 1