mirror of
https://github.com/hastagAB/Awesome-Python-Scripts.git
synced 2024-12-21 09:40:15 +00:00
9de4f392d3
Update README.md
36 lines
1.2 KiB
Python
Executable File
36 lines
1.2 KiB
Python
Executable File
import sqlite3
|
|
|
|
|
|
class DBHelper:
|
|
|
|
def __init__(self, dbname="todo.sqlite"):
|
|
self.dbname = dbname
|
|
self.conn = sqlite3.connect(dbname)
|
|
|
|
def setup(self):
|
|
tblstmt = "CREATE TABLE IF NOT EXISTS items (description text, due_date date ,owner text)"
|
|
itemidx = "CREATE INDEX IF NOT EXISTS itemIndex ON items (description ASC)"
|
|
ownidx = "CREATE INDEX IF NOT EXISTS ownIndex ON items (owner ASC)"
|
|
self.conn.execute(tblstmt)
|
|
self.conn.execute(itemidx)
|
|
self.conn.execute(ownidx)
|
|
self.conn.commit()
|
|
|
|
def add_item(self, item_text, due_date, owner):
|
|
stmt = "INSERT INTO items (description, due_date ,owner) VALUES (?, ? ,?)"
|
|
args = (item_text, due_date ,owner)
|
|
self.conn.execute(stmt, args)
|
|
self.conn.commit()
|
|
|
|
def delete_item(self, item_text, owner):
|
|
stmt = "DELETE FROM items WHERE description = (?) AND owner = (?)"
|
|
args = (item_text, owner )
|
|
self.conn.execute(stmt, args)
|
|
self.conn.commit()
|
|
|
|
def get_items(self, owner):
|
|
stmt = "SELECT description,due_date FROM items WHERE owner = (?)"
|
|
args = (owner, )
|
|
return [x for x in self.conn.execute(stmt, args)]
|
|
|