def run(self):
self.parse_options()
if self.options.run_editor:
self.open_editor()
if not os.path.exists(self.options.logfile):
self.create_file(self.options.logfile)
self.worklog = WorkLog()
if self.options.grep_pattern is not None:
self.worklog.set_filter_regex(re.compile(self.options.grep_pattern))
for dt, text in job_reader(self.options.logfile):
self.worklog.append_log_event(dt, text)
if 0 < len(self.args):
self.append_event(" ".join(self.args))
elif self.worklog.end():
if self.worklog.end().date() == datetime.date.today():
self.worklog.append_assumed_interval(datetime.datetime.now())
if self.options.categorized_report:
self.apply_categorization()
self.worklog.compute_stats(WorkStatsFactory("PL"))
if self.options.aggregate_range:
cls = AggregateRangeDisplay
elif self.options.aggregate_day:
cls = AggregateSessionDisplay
else:
cls = ChronologicalSessionDisplay
session_display = cls(self.worklog, DayRange(self.options.range),
self.options.split_breaks)
if self.options.percentage and (self.options.aggregate_range or self.options.aggregate_day):
session_display.set_unit(ReportTimePercent(self.options.split_breaks))
session_display.display()
class DidApplication:
def run(self):
self.parse_options()
if self.options.run_editor:
self.open_editor()
if not os.path.exists(self.options.logfile):
self.create_file(self.options.logfile)
self.worklog = WorkLog()
for dt, text in job_reader(self.options.logfile):
self.worklog.append_log_event(dt, text)
if 0 < len(self.args):
self.append_event(" ".join(self.args))
elif self.worklog.end():
if self.worklog.end().date() == datetime.date.today():
self.worklog.append_assumed_interval(datetime.datetime.now())
if self.options.categorized_report:
self.apply_categorization()
self.worklog.compute_stats(WorkStatsFactory("PL"))
if self.options.aggregate_range:
cls = AggregateRangeDisplay
elif self.options.aggregate_day:
cls = AggregateSessionDisplay
else:
cls = ChronologicalSessionDisplay
session_display = cls(self.worklog, DayRange(self.options.range))
if self.options.percentage and (self.options.aggregate_range or self.options.aggregate_day):
session_display.set_unit(ReportTimePercent())
session_display.display()
def apply_categorization(self):
filename = self.get_config_dir() + "/categorization"
rx = re.compile("^s/((?:[^\\\\/]|\\\\.)+)/((?:[^\\\\/]|\\\\.)*)/\s*$")
try:
f = open(filename, "r")
for line in f:
if re.match("^\s*(#|$)", line):
continue
m = rx.match(line)
if not m:
print "Invalid line in categorization file \"%s\": %s" \
% (filename, line)
sys.exit(1)
pattern = forward_slash_unescape(m.group(1))
subst = forward_slash_unescape(m.group(2))
cat_rx = re.compile(pattern)
self.worklog.map_names(lambda x: cat_rx.sub(subst, x))
except IOError as err:
print "Error opening/reading from file '{0}': {1}".format(
err.filename, err.strerror)
def parse_options(self):
parser = OptionParser(usage="%prog [options] [CURRENT-TASK]")
parser.add_option("-f", "--log-file",
metavar="FILE",
dest="logfile",
default=self.get_config_dir() + "/joblog",
action="store",
help="set the task database file")
parser.add_option("-e", "--edit", action="store_true",
dest="run_editor",
help="open the task database file in an editor")
parser.add_option("-r", "--range",
dest="range",
default="0",
action="store",
help="print log for days within given range")
parser.add_option("-d", "--aggregate-day",
action="store_true",
dest="aggregate_day",
help="display jobs aggregated for each day")
parser.add_option("-a", "--aggregate",
action="store_true",
dest="aggregate_range",
help="display jobs aggregated for the complete range")
parser.add_option("-c", "--categorized",
action="store_true",
dest="categorized_report",
help="apply categorizing regexes to job names")
parser.add_option("-p", "--percentage",
action="store_true",
dest="percentage",
help="print times as percentage of total work time")
(options, args) = parser.parse_args()
self.options = options
self.args = args
def open_editor(self):
editors = []
if os.environ.has_key('VISUAL'):
editors.append(os.environ['VISUAL'])
if os.environ.has_key('EDITOR'):
editors.append(os.environ['EDITOR'])
editors.extend(["/usr/bin/vim", "/usr/bin/nano", "/usr/bin/pico",
"/usr/bin/vi", "/usr/bin/mcedit"])
for editor in editors:
if os.path.exists(editor):
subprocess.call([editor, self.options.logfile])
sys.exit()
def get_config_dir(self):
if 'HOME' in os.environ:
return os.environ['HOME'] + "/.config/did"
return "."
def mkdir_p(self, path):
try:
os.makedirs(path)
except OSError as exc:
if exc.errno != errno.EEXIST:
raise
def create_file(self, filename):
dirpart, unused_filepart = os.path.split(filename)
if dirpart != '':
self.mkdir_p(dirpart)
file(filename, 'a').close()
def append_event(self, name):
if name == ".":
# Last break interval
last_break = self.worklog.last_break_interval()
if last_break is None:
name = ".break"
else:
name = last_break.name()
elif name == ",":
# Last work interval
last_work = self.worklog.last_work_interval()
if last_work is None:
name = "work"
else:
name = last_work.name()
writer = JobListWriter(self.options.logfile)
now = datetime.datetime.now()
writer.append(now, name)
self.worklog.append_log_event(now, name)
