import storage

class WordFrequencyStorage(storage.Storage):
    __COMMAND_CREATE_WORD_FREQUENCIES = """CREATE TABLE IF NOT EXISTS WordFrequencies (
        idtext INTEGER,
        word TEXT,
        frequency REAL
        )"""

    def _createDatabase(self):
        c = self.connect()
        c.execute(self.__COMMAND_CREATE_WORD_FREQUENCIES)
        self.commit()

    def _destroyDatabase(self):
        c = self.connect()
        c.execute('DROP TABLE IF EXISTS WordFrequencies')
        self.commit()

    def store(self, idtext, frequencies):
        c = self.connect()
        for word, freq in frequencies:
            c.execute('INSERT INTO WordFrequencies VALUES(?, ?, ?)', (idtext, word, freq))
        self.commit()