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