AssignmentDue/task.py at master · Xuanghdu/AssignmentDue · GitHub

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
# from base import Base
from datetime import datetime

# class Task(Base):


class Task():

    __slots__ = 'taskname', 'description', 'group', 'due_date', 'divisions', \
        'dependencies', 'timestamp', 'add_date', 'last_modified'

    def __init__(self, taskname, description, group, due_date=[1999, 12, 31],
                 divisions=[], dependencies=[], add_date=None):
        # super.__init__()
        self.taskname = taskname
        self.description = description
        self.group = group
        self.due_date = due_date
        self.divisions = divisions
        self.dependencies = dependencies
        now = datetime.now()
        self.timestamp = now.timestamp()
        if add_date == None:
            self.add_date = now.utctimetuple()[:5]
        else:
            self.add_date = add_date
        self.last_modified = self.add_date

    def __iter__(self):
        for division in self.divisions:
            for user in division:
                yield user

    def add_dependency(self, user1, user2):
        '''(Task, User, User) -> None'''
        index1, index2 = -1, -1
        for i in range(len(self.divisions)):
            if user1 in self.divisions[i]:
                index1 = i
                break
        for i in range(len(self.divisions)):
            if user2 in self.divisions[i]:
                index2 = i
                break
        if index1 >= 0 and index2 >= 0:
            # both users exists in the task
            if index1 == index2:
                # already in the same set
                pass
            else:
                # merge two sets
                self.divisions[index1] = \
                    self.divisions[index1].union(self.divisions[index2])
                del self.divisions[index2]
        elif index1 >= 0:
            # user1 in the task but user2 does not
            self.divisions[index1].add[user2]
        elif index2 >= 0:
            # user2 in the task but user1 does not
            self.divisions[index2].add[user1]
        else:
            # neither user is in the task
            self.divisions.append({user1, user2})

    # Remove???
    def complete_task(self, user):
        '''(Task, User) -> None'''
        index = -1
        for i in range(self.divisions):
            if user in self.divisions[i]:
                index = i
                break
        if index < 0:
            raise Exception('User should not have access of the task')
        del self.divisions[index]

    def __str__(self):
        return 'taskname: %s\ndescription: %s\ndue date: %s\nadd date: %S' % \
            (self.taskname, self.description,
             '/'.join(self.due_date), self.add_date)

    # def load_json_dict(self, load_json_dict):
    #     self.group = load_json_dict["group"]
    #     self.content = load_json_dict["content"]
    #     self.due_date = load_json_dict["due_date"]
    #     self.divisions = load_json_dict["divisions"]

    # def dump_json_dict(self):
    #     return {"group": self.group,
    #             "content": self.content,
    #             "due_date": self.due_date,
    #             "divisions", self.divisions}