from flask import g

from ltpdarepo.form import Form
from wtforms.fields import TextField, HiddenField


class IQuery(Form):
    querystring = HiddenField()
    name = TextField("Name")


class Query(dict):
    def __init__(self, name=None, db=None, querystring=None):
        super(Query, self).__init__(name=name, db=db, querystring=querystring)
        self.__dict__ = self

    def create(self):
        curs = g.db.cursor()
        curs.execute("""INSERT INTO queries (name, db, querystring)
                        VALUES (%(name)s, %(db)s, %(querystring)s)""", dict(self))