← Back to branch summary

~allison/ubuntu-app-review-board/dostuff

« back to all changes in this revision

Viewing changes to dostuff_lib/Task.py

  • Committer: Luca Marchetti
  • Date: 2012-07-09 13:35:39 UTC
  • Revision ID: ximilian@gmail.com-20120709133539-m4p5kh2jr0tylgdx
commit before release

Show diffs side-by-side

added added

removed removed

Lines of Context:
dostuff_lib/Task.py
 
1
# -*- Mode: Python; coding: utf-8; indent-tabs-mode: nil; tab-width: 4 -*-
 
2
### BEGIN LICENSE
 
3
# Copyright (C) 2012 ximilian <ximilian@gmail.com>
 
4
# This program is free software: you can redistribute it and/or modify it 
 
5
# under the terms of the GNU General Public License version 3, as published 
 
6
# by the Free Software Foundation.
 
7
# 
 
8
# This program is distributed in the hope that it will be useful, but 
 
9
# WITHOUT ANY WARRANTY; without even the implied warranties of 
 
10
# MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR 
 
11
# PURPOSE.  See the GNU General Public License for more details.
 
12
# 
 
13
# You should have received a copy of the GNU General Public License along 
 
14
# with this program.  If not, see <http://www.gnu.org/licenses/>.
 
15
### END LICENSE
 
16
 
 
17
import re
 
18
 
 
19
class Task(object):
 
20
    """ A class for task object """
 
21
 
 
22
    id = -1
 
23
    priority = ""
 
24
    data = ""
 
25
    creation_date = ""
 
26
    completion_date = ""
 
27
    project = 0
 
28
    context = 0
 
29
    keyword = 0
 
30
    raw = ""
 
31
 
 
32
    status = "beg"
 
33
 
 
34
    OPEN_BRACKET  = "("
 
35
    CLOSE_BRACKET = ")"
 
36
    COMPLETED = "x"
 
37
    PROJECT   = "+"
 
38
    CONTEXT   = "@"
 
39
    KEYWORD   = "#"
 
40
 
 
41
    REGEX_DATE = re.compile('[0-9]{4}(-[0-9]{2}){2}')
 
42
 
 
43
    def __init__(self, line):
 
44
        Task.id += 1
 
45
        self.id = Task.id
 
46
        self.project = set()
 
47
        self.context = set()
 
48
        self.keyword = set()
 
49
        self.raw = line
 
50
 
 
51
        self.Parse()
 
52
        self.data = self.data.strip()
 
53
        if self.data == "":
 
54
            self.id = -1
 
55
            Task.id -=1
 
56
 
 
57
    def __repr__(self):
 
58
        return "Task(id=%r, priority=%r, data=%r, creation_date=%r, completion_date=%r, project=%r, context=%r, keyword=%r)" % \
 
59
                (self.id, self.priority, self.data, \
 
60
                 repr(self.creation_date), repr(self.completion_date), 
 
61
                 self.project, self.context, self.keyword)
 
62
 
 
63
    def __str__(self):
 
64
        return 'id: ' + str(self.id) + '\n' \
 
65
               'pri: ' + self.priority + '\n' \
 
66
               'date: ' + self.creation_date + " _ " + self.completion_date + '\n' \
 
67
               '+: ' + str(self.project) + '\n' \
 
68
               '@: ' + str(self.context) + '\n' \
 
69
               '#: ' + str(self.keyword) + '\n' \
 
70
               '[' + self.data + ']' + '\n'
 
71
 
 
72
    def Parse(self, begin=0):
 
73
        if not begin < self.raw.__len__():
 
74
            return
 
75
 
 
76
        while (self.raw[begin] == ' '):
 
77
            begin += 1
 
78
 
 
79
        if self.status == "beg":
 
80
            begin = self.ParsePriority(0)
 
81
            begin = self.ParseCompleted(begin)
 
82
        elif self.status == "sdate":
 
83
            [self.creation_date, begin] = self.ParseDate(begin)
 
84
            self.status = ""
 
85
        elif self.status == "edate":
 
86
            [self.completion_date, begin] = self.ParseDate(begin)
 
87
            self.status = "sdate"
 
88
        elif self.raw[begin] == Task.PROJECT:
 
89
            self.ParseToList(begin+1, self.project)
 
90
            begin = self.ParseDataWord(begin)
 
91
        elif self.raw[begin] == Task.CONTEXT:
 
92
            self.ParseToList(begin+1, self.context)
 
93
            begin = self.ParseDataWord(begin)
 
94
        elif self.raw[begin] == Task.KEYWORD:
 
95
            self.ParseToList(begin+1, self.keyword)
 
96
            begin = self.ParseDataWord(begin)
 
97
        else:
 
98
            begin = self.ParseDataWord(begin)
 
99
 
 
100
        self.Parse(begin)
 
101
 
 
102
    def ParsePriority(self, begin):
 
103
        self.status = "sdate"
 
104
        if self.raw[begin] == Task.OPEN_BRACKET and self.raw[begin+2] == Task.CLOSE_BRACKET:
 
105
            self.priority = self.raw[1]
 
106
            return begin+3
 
107
        self.priority = "-"
 
108
        return begin
 
109
 
 
110
    def ParseCompleted(self, begin):
 
111
        if self.raw[begin] == Task.COMPLETED and self.raw[begin+1] == ' ':
 
112
            self.priority = Task.COMPLETED
 
113
            self.status = "edate"
 
114
            return begin+2
 
115
        return begin
 
116
 
 
117
    def ParseDate(self, begin):
 
118
        [word, end] = self.ParseWord(self.raw, begin)
 
119
        if not Task.REGEX_DATE.match(word):
 
120
            end  = begin
 
121
            word = ""
 
122
        return [word, end]
 
123
 
 
124
    def ParseToList(self, begin, working_list):
 
125
        [elem, end] = self.ParseWord(self.raw, begin)
 
126
        if not elem == "":
 
127
            working_list.add(elem)
 
128
        return end
 
129
 
 
130
    def ParseDataWord(self, begin):
 
131
        [word, end] = self.ParseWord(self.raw, begin)
 
132
        self.data += " " + word
 
133
        return end
 
134
 
 
135
    def ParseWord(self, text, begin):
 
136
        end = begin
 
137
        try:
 
138
            end = text.index(" ", begin)
 
139
        except (ValueError):
 
140
            end = text.__len__()
 
141
        word = text[begin:end]
 
142
        return [word, end]
 
143