this repo has no description
edavis.dev
1import logging
2
3import apsw
4
5class FirehoseManager:
6 def __init__(self, fname='db/firehose.db'):
7 self.db_cnx = apsw.Connection(fname)
8 self.db_cnx.pragma('journal_mode', 'WAL')
9 with self.db_cnx:
10 self.db_cnx.execute("create table if not exists firehose(key text unique, value text)")
11
12 self.logger = logging.getLogger('firehose.manager')
13
14 def get_sequence_number(self):
15 row = self.db_cnx.execute("select * from firehose where key = 'seq'").fetchone()
16 if row is None:
17 return None
18 (key, value) = row
19 return int(value)
20
21 def set_sequence_number(self, value):
22 self.logger.debug(f'setting sequence number = {value}')
23
24 with self.db_cnx:
25 self.db_cnx.execute(
26 "insert into firehose (key, value) values ('seq', :value) on conflict(key) do update set value = :value",
27 dict(value=value)
28 )
29
30 self.db_cnx.pragma('wal_checkpoint(RESTART)')