ozbolt/luscenje_struktur
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Step recorded: generate_renders

Browse Source
This commit is contained in:
Ozbolt Menegatti 2019-08-21 12:16:10 +02:00
parent 046aef031f
commit dedc031696
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/database.py
View File

@ -11,6 +11,9 @@ class Database:
self.new = not os.path.exists(filename) self.new = not os.path.exists(filename)
self.db = sqlite3.connect(filename) self.db = sqlite3.connect(filename)
self.init("CREATE TABLE StepsDone ( step varchar(32) )")
self.commit()
def execute(self, *args, **kwargs): def execute(self, *args, **kwargs):
return self.db.execute(*args, **kwargs) return self.db.execute(*args, **kwargs)

8
src/word_stats.py
View File

@ -42,6 +42,12 @@ class WordStats:
return self.all_words return self.all_words
def generate_renders(self): def generate_renders(self):
step_name = 'generate_renders'
wc_done = self.db.execute("SELECT count(*) FROM StepsDone WHERE step=?", (step_name, )).fetchone()
if wc_done[0] == 1:
print("Skipping GenerateRenders, already complete")
return
lemmas = [lemma for (lemma, ) in self.db.execute("SELECT DISTINCT lemma FROM UniqWords")] lemmas = [lemma for (lemma, ) in self.db.execute("SELECT DISTINCT lemma FROM UniqWords")]
for lemma in progress(lemmas, 'word-count'): for lemma in progress(lemmas, 'word-count'):
num_words = defaultdict(int) num_words = defaultdict(int)
@ -51,6 +57,8 @@ class WordStats:
for msd0, freq in num_words.items(): for msd0, freq in num_words.items():
self.db.execute("INSERT INTO WordCount (lemma, msd0, frequency) VALUES (?,?,?)", self.db.execute("INSERT INTO WordCount (lemma, msd0, frequency) VALUES (?,?,?)",
(lemma, msd0, freq)) (lemma, msd0, freq))
self.db.execute("INSERT INTO StepsDone (step) VALUES (?)", (step_name, ))
self.db.commit() self.db.commit()
def common_lemma_msd(self, lemma, msd): def common_lemma_msd(self, lemma, msd):