diff --git a/db.py b/db.py new file mode 100644 index 0000000..6448216 --- /dev/null +++ b/db.py @@ -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()