-
Notifications
You must be signed in to change notification settings - Fork 0
/
simpletask.py
155 lines (121 loc) · 3.96 KB
/
simpletask.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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
# create a simple taskmanager using sqlalchemy
from dataclasses import dataclass
from typing import Callable, List
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String
# create engine
engine = create_engine("sqlite:///task.db", echo=True)
# create session
Session = sessionmaker(bind=engine)
session = Session()
# create base class
Base = declarative_base()
class Task(Base):
"""
This class represents the Task table in the database.
"""
__tablename__ = "task"
id = Column(Integer, primary_key=True)
subject = Column(String, max_length=50)
task = Column(String)
priority = Column(Integer)
def __repr__(self):
return "<Task(task='%s', priority='%s')>" % (self.task, self.priority)
# create table
Base.metadata.create_all(engine)
class AddTask:
"""
This class is responsible for adding tasks to the database.
"""
def __init__(self):
self.user_input_subject = input("Enter subject: ")
self.user_input_task = input("Enter task: ")
self.user_input_priority = input("Enter priority: ")
if (
self.user_input_subject == ""
or self.user_input_task == ""
or self.user_input_priority == ""
):
print("No input entered")
else:
self.task = Task(
subject=self.user_input_subject,
task=self.user_input_task,
priority=self.user_input_priority,
)
session.add(self.task)
session.commit()
def __repr__(self):
return "<UserInput(subject='%s', task='%s', priority='%s')>" % (
self.user_input_subject,
self.user_input_task,
self.user_input_priority,
)
class ShowAllTasks:
"""
This class is responsible for showing all tasks in the database.
"""
def __init__(self):
self.tasks = session.query(Task).all()
for task in self.tasks:
print(
f"\nTask id : {task.id} \nSubject : {task.subject}\nTask : {task.task} \nPriority : {task.priority} \n\n"
)
def __repr__(self):
return "<UserQuery(subject='%s', task='%s', priority='%s')>" % (self.tasks)
class DeleteTask:
"""
This class is responsible for deleting tasks from the database.
"""
def __init__(self):
self.task_id = input("Enter task id to delete: ")
if self.task_id == "":
print("No task id entered")
exit()
else:
self.task = session.query(Task).filter_by(id=self.task_id).first()
session.delete(self.task)
session.commit()
def __repr__(self):
return "<UserDelete(task_id='%s', task='%s')>" % (self.task_id, self.task)
@dataclass
class Command:
"""
This class represents a command that the user can execute.
"""
name: str
action: Callable[[], None]
class UserInput:
"""
This class is responsible for processing user input.
"""
def __init__(self):
self.commands = self.initialize_commands()
def initialize_commands(self) -> List[Command]:
return [
Command("0", self.exit),
Command("1", ShowAllTasks),
Command("2", AddTask),
Command("3", DeleteTask),
]
def exit(self):
session.close()
exit()
def process_input(self):
user_input = input(
" \n 0 to exit :\n 1 to show all tasks \n 2 to add a task :\n 3 to delete a task : "
)
for command in self.commands:
if user_input == command.name:
command.action()
break
else:
print("Invalid input")
def __repr__(self):
return "<UserInput(user_input='%s')>" % (self.user_input)
if __name__ == "__main__":
user_input = UserInput()
while True:
user_input.process_input()