feat(db): add db layer
This commit is contained in:
24
db.py
Normal file
24
db.py
Normal 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()
|
||||
Reference in New Issue
Block a user