-
Notifications
You must be signed in to change notification settings - Fork 0
/
sqlighter.py
36 lines (27 loc) · 923 Bytes
/
sqlighter.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
# Modules
import sqlite3
from datetime import datetime
"""
Class: "SQLighter"
"""
class SQLighter:
def __init__(self, db_name):
self.connect = sqlite3.connect(db_name)
self.cursor = self.connect.cursor()
def add_task(self, user_id, name_task):
with self.connect:
self.cursor.execute(
"INSERT INTO `tasks` (`user_id`, `task`, `time`) VALUES (?, ?, ?)",
(user_id, name_task, datetime.now().ctime()),
)
def show_tasks(self, user_id):
with self.connect:
return self.cursor.execute(
"SELECT * FROM `tasks` WHERE `user_id` = ?", (user_id,)
).fetchall()
def delete_task(self, user_id, task):
with self.connect:
self.cursor.execute(
"DELETE FROM `tasks` WHERE `user_id` = ? AND `task` = ?",
(user_id, task),
)