54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
import os, sqlite3
|
|
|
|
class DBConnect:
|
|
def __init__(self):
|
|
try:
|
|
self.db = None
|
|
self.conn = None
|
|
print("DBConnect initialized...")
|
|
except Exception as e:
|
|
print("Exception raised:")
|
|
print(e)
|
|
raise
|
|
|
|
def createConnection(self):
|
|
self.db = sqlite3.connect('static/db/database.db')
|
|
self.conn = self.db.cursor()
|
|
|
|
def closeConnection(self):
|
|
self.conn.close()
|
|
|
|
|
|
def createEntry(self, args):
|
|
try:
|
|
# Example Insert
|
|
# sql = "INSERT INTO bills (Title, Amount, MoYr, Duedate, Optional) Values(?,?,?,?,?)";
|
|
# ps = (TITLE, float(AMOUNT), MOYR, DATE, OPTIONAL, )
|
|
self.conn.execute(sql, ps)
|
|
self.db.commit()
|
|
except Exception as e:
|
|
print("Exception raised:")
|
|
print(e)
|
|
|
|
def updateEntry(self, args):
|
|
try:
|
|
# Example Update
|
|
# sql = "UPDATE bills SET Title = ?, Amount = ?, MoYr = ?, Duedate = ?, Optional = ? WHERE id = ?";
|
|
# ps = (TITLE, float(AMOUNT), MOYR, DATE, OPTIONAL, int(ID), )
|
|
self.conn.execute(sql, ps)
|
|
self.db.commit()
|
|
except Exception as e:
|
|
print("Exception raised:")
|
|
print(e)
|
|
|
|
def deleteEntry(self, ID):
|
|
try:
|
|
# Example Delete
|
|
# sql = "DELETE FROM bills WHERE id = ?";
|
|
# ps = (int(ID), )
|
|
self.conn.execute(sql, ps)
|
|
self.db.commit()
|
|
except Exception as e:
|
|
print("Exception raised:")
|
|
print(e)
|