abf9f03e88
Added string generator function to Task Added addTask compatibility with optional args Added some exception handling to addTask
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
from task import *
|
|
class Chat(object):
|
|
def __init__(self, ident, name):
|
|
self.id = ident
|
|
self.tasks = []
|
|
self.name = name
|
|
|
|
def addTask(self, name, repeat=0, hour=0, day=0, month=0, year=0):
|
|
t = Task(name, repeat, hour, day, month, year)
|
|
self.tasks.append(t)
|
|
print('task ' + self.tasks[-1].name + ' added')
|
|
return t
|
|
|
|
def getTasks(self, y=0, d=0, m=0, y2=0, d2=0, m2=0):
|
|
dateNow = datetime.now()
|
|
if y == 0:
|
|
y = dateNow.year
|
|
if m == 0:
|
|
m = dateNow.month
|
|
if d == 0:
|
|
d = dateNow.day
|
|
if y2 == 0:
|
|
y2 = dateNow.year
|
|
if m2 == 0:
|
|
m2 = dateNow.month
|
|
if d2 == 0:
|
|
d2 = dateNow.day
|
|
dateIni = datetime(y, m, d)
|
|
dateEnd = datetime(y2, m2, d2)
|
|
resultTasks = []
|
|
for t in self.tasks:
|
|
print(t)
|
|
if t.getDate() >= dateIni and t.getDate() <= dateEnd:
|
|
resultTasks.append(t)
|
|
return resultTasks
|
|
|
|
def getChatName(self):
|
|
return self.name
|