24 lines
728 B
Python
24 lines
728 B
Python
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() |