feat(db): add db layer

This commit is contained in:
2025-01-06 20:32:56 +01:00
parent a20d69fa42
commit feb929c6f8

24
db.py Normal file
View File

@@ -0,0 +1,24 @@
import sqlite3
class DB:
def __init__(self, db_path, schema_path="schema.sql", seed_path="seed.sql"):
self.con = sqlite3.connect(db_path)
self.cur = self.con.cursor()
with open(schema_path) as f:
self.cur.executescript(f.read())
with open(seed_path) as f:
self.cur.executescript(f.read())
def __del__(self):
self.con.close()
def query(self, query, params=[]):
self.cur.execute(query, params)
return self.cur.fetchall()
def insertDisclosures(self, disclosures):
sql = "INSERT INTO disclosures (member_id, filing_year, filing, link) VALUES (?, ?, ?, ?)"
self.cur.executemany(sql, disclosures)
self.con.commit()