sql

#-*- coding: utf8 -*-

import sys
import sqlite3

class sqliteDB():
    def __init__(self):
        self.db = sqlite3.connect('./data_quality.db')

    def add_regexp(self, param, table, regexp):
        crtable = ('%s(%s)' % (param, table))
        self.db.execute('create table if not exists %s' %  (crtable))
        data = (u'%s' % (unicode(regexp)))
        sql = ('insert into %s (%s) values ("%s")' % (param, table, data))
        self.db.execute('%s' % (sql))
        self.db.commit()

    def take_all(self, param, table):
        ttable = ('%s from %s' % (str(table), str(param)))
        cur = self.db.execute('select %s' % (ttable))
        rows = []
        data = []
        for row in cur:
            rows.append(map(lambda a: a.encode('utf-8'), row))

        for s in rows:
            data.append(s[0])
        print data
        return data