nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

new commands

Browse source
This commit is contained in:
Romain Bignon 2011-04-05 18:29:23 +02:00 • committed by Romain Bignon
commit 33051bb986
1 changed files with 157 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

192
weboob/applications/boobathon/boobathon.py
View file

@ -31,11 +31,14 @@ __all__ = ['Boobathon']
class Task(object): class Task(object):
def __init__(self, id, backend, capability): STATUS_NONE = 0
self.id = id STATUS_PROGRESS = 1
STATUS_DONE = 2
def __init__(self, backend, capability):
self.backend = backend self.backend = backend
self.capability = capability self.capability = capability
self.done = False self.status = self.STATUS_NONE
self.date = None self.date = None
self.branch = u'' self.branch = u''
@ -55,9 +58,8 @@ class Event(object):
self.name = 'wiki/weboob/%s' % name self.name = 'wiki/weboob/%s' % name
self.description = None self.description = None
self.date = None self.date = None
self.start = None self.begin = None
self.end = None self.end = None
self.duration = None
self.location = None self.location = None
self.backend = backend self.backend = backend
self.members = OrderedDict() self.members = OrderedDict()
@ -66,11 +68,23 @@ class Event(object):
def get_me(self): def get_me(self):
return self.members.get(self.backend.browser.get_userid(), None) return self.members.get(self.backend.browser.get_userid(), None)
def currently_in_event(self):
if not self.date or not self.begin or not self.end:
return False
return self.begin < datetime.now() < self.end
def format_duration(self):
if not self.begin or not self.end:
return None
delta = self.end - self.begin
return '%02d:%02d' % (delta.seconds/3600, delta.seconds%3600)
def load(self): def load(self):
self.content = self.backend.get_content(self.name) self.content = self.backend.get_content(self.name)
self.members.clear() self.members.clear()
member = None member = None
task_id = 0
for line in self.content.content.split('\n'): for line in self.content.content.split('\n'):
line = line.strip() line = line.strip()
if line.startswith('h1. '): if line.startswith('h1. '):
@ -96,13 +110,11 @@ class Event(object):
key, value = m.group(1), m.group(2) key, value = m.group(1), m.group(2)
if member is None: if member is None:
if key == 'Date': if key == 'Date':
self.date = self.parse_datetime(value, '%Y-%m-%d') self.date = self.parse_date(value)
elif key == 'Start': elif key == 'Start' or key == 'Begin':
self.start = self.parse_datetime(value, '%H:%M') self.begin = self.parse_time(value)
elif key == 'End': elif key == 'End':
self.end = self.parse_datetime(value, '%H:%M') self.end = self.parse_time(value)
elif key == 'Duration':
self.duration = self.parse_datetime(value, '%H:%M')
elif key == 'Location': elif key == 'Location':
self.location = value self.location = value
else: else:
@ -121,12 +133,15 @@ class Event(object):
if not m: if not m:
print 'Unable to parse task: "%s"' % line print 'Unable to parse task: "%s"' % line
continue continue
task_id += 1 task = Task(m.group(1), m.group(2))
task = Task(task_id, m.group(1), m.group(2))
member.tasks.append(task) member.tasks.append(task)
if m.group(3) == '!/img/weboob/_progress.png!':
task.status = task.STATUS_PROGRESS
continue
mm = re.match('!/img/weboob/_done.png! (\d+):(\d+) (\w+)', m.group(3)) mm = re.match('!/img/weboob/_done.png! (\d+):(\d+) (\w+)', m.group(3))
if mm: if mm and self.date:
task.done = True task.status = task.STATUS_DONE
task.date = datetime(self.date.year, task.date = datetime(self.date.year,
self.date.month, self.date.month,
self.date.day, self.date.day,
@ -134,9 +149,20 @@ class Event(object):
int(m.group(2))) int(m.group(2)))
task.branch = m.group(3) task.branch = m.group(3)
def parse_datetime(self, value, format): def parse_date(self, value):
try: try:
return datetime.strptime(value, format) return datetime.strptime(value, '%Y-%m-%d')
except ValueError:
return None
def parse_time(self, value):
m = re.match('(\d+):(\d+)', value)
if not m:
return
try:
return self.date.replace(hour=int(m.group(1)),
minute=int(m.group(2)))
except ValueError: except ValueError:
return None return None
@ -150,7 +176,7 @@ h2. Event
%s %s
* *Date*: %s * *Date*: %s
* *Start*: %s * *Begin*: %s
* *End*: %s * *End*: %s
* *Duration*: %s * *Duration*: %s
* *Location*: %s * *Location*: %s
@ -160,9 +186,9 @@ h2. Attendees
""" % (self.title, """ % (self.title,
self.description, self.description,
self.date.strftime('%Y-%m-%d') if self.date else '_Unknown_', self.date.strftime('%Y-%m-%d') if self.date else '_Unknown_',
self.start.strftime('%H:%M') if self.start else '_Unknown_', self.begin.strftime('%H:%M') if self.begin else '_Unknown_',
self.end.strftime('%H:%M') if self.end else '_Unknown_', self.end.strftime('%H:%M') if self.end else '_Unknown_',
self.duration.strftime('%H:%M') if self.duration else '_Unknown_', self.format_duration() or '_Unknown_',
self.location or '_Unknown_') self.location or '_Unknown_')
for member in self.members.itervalues(): for member in self.members.itervalues():
@ -184,17 +210,19 @@ h2. Attendees
* *Hardware*: %s * *Hardware*: %s
%s %s
|_.Backend|_.Capabilities|_.Done|""" % (member.name, |_.Backend|_.Capabilities|_.Status|""" % (member.name,
member.id, member.id,
repository, repository,
member.hardware, member.hardware,
availabilities) availabilities)
for task in member.tasks: for task in member.tasks:
if task.done: if task.status == task.STATUS_DONE:
status = '!/img/weboob/_done.png! %s:%s %s' % (task.date.hour, status = '!/img/weboob/_done.png! %s:%s %s' % (task.date.hour,
task.date.minute, task.date.minute,
task.branch) task.branch)
elif task.status == task.STATUS_PROGRESS:
status = '!/img/weboob/_progress.png!'
else: else:
status = ' ' status = ' '
s += u""" s += u"""
@ -244,11 +272,11 @@ class Boobathon(ReplApplication):
regexp='^(\d{4}-\d{2}-\d{2})?$') regexp='^(\d{4}-\d{2}-\d{2})?$')
if self.event.date: if self.event.date:
self.event.date = datetime.strptime(self.event.date, '%Y-%m-%d') self.event.date = datetime.strptime(self.event.date, '%Y-%m-%d')
self.event.start = self.ask('Start at (HH:MM)', self.event.begin = self.ask('Begin at (HH:MM)',
default=self.event.start.strftime('%H:%M') if self.event.start else '', default=self.event.begin.strftime('%H:%M') if self.event.begin else '',
regexp='^(\d{2}:\d{2})?$') regexp='^(\d{2}:\d{2})?$')
if self.event.start: if self.event.begin:
self.event.start = datetime.strptime(self.event.start, '%H:%M') self.event.begin = datetime.strptime(self.event.begin, '%H:%M')
self.event.end = self.ask('End at (HH:MM)', self.event.end = self.ask('End at (HH:MM)',
default=self.event.end.strftime('%H:%M') if self.event.end else '', default=self.event.end.strftime('%H:%M') if self.event.end else '',
regexp='^(\d{2}:\d{2})?$') regexp='^(\d{2}:\d{2})?$')
@ -271,10 +299,12 @@ class Boobathon(ReplApplication):
for member in self.event.members.itervalues(): for member in self.event.members.itervalues():
s = u' %s %20s %s|' % ('->' if member.is_me else ' ', member.name, self.BOLD) s = u' %s %20s %s|' % ('->' if member.is_me else ' ', member.name, self.BOLD)
for task in member.tasks: for task in member.tasks:
if task.done: if task.status == task.STATUS_DONE:
s += '##' s += '##'
else: elif task.status == task.STATUS_PROGRESS:
s += '--' s += '--'
else:
s += ' '
s += '|%s' % self.NC s += '|%s' % self.NC
print s print s
@ -287,24 +317,35 @@ class Boobathon(ReplApplication):
for i, task in enumerate(member.tasks): for i, task in enumerate(member.tasks):
while len(tasks) <= i*2+1: while len(tasks) <= i*2+1:
tasks.append([]) tasks.append([])
if task.done: if task.status == task.STATUS_DONE:
status = '%s#%s' % (self.BOLD, self.NC) status = '%s#%s' % (self.BOLD, self.NC)
elif task.status == task.STATUS_PROGRESS:
status = '|'
else: else:
status = ' ' status = ' '
tasks[i*2].append('%s %s' % (status, task.backend)) tasks[i*2].append('%s %s' % (status, task.backend))
tasks[i*2+1].append('%s `-%s' % (status, task.capability[3:])) tasks[i*2+1].append('%s `-%s' % (status, task.capability[3:]))
sys.stdout.write(' ')
for name in members: for name in members:
sys.stdout.write(' %s%-20s%s' % (self.BOLD, name, self.NC)) sys.stdout.write(' %s%-20s%s' % (self.BOLD, name, self.NC))
sys.stdout.write('\n') sys.stdout.write('\n ')
for name in members: for name in members:
sys.stdout.write(' %s%-20s%s' % (self.BOLD, '-' * len(name), self.NC)) sys.stdout.write(' %s%-20s%s' % (self.BOLD, '-' * len(name), self.NC))
sys.stdout.write('\n') sys.stdout.write('\n')
for line in tasks: for i, line in enumerate(tasks):
if not i%2:
sys.stdout.write(' #%-2d' % (i/2))
else:
sys.stdout.write(' ')
for task in line: for task in line:
sys.stdout.write(' %-20s' % task) sys.stdout.write(' %-20s' % task)
sys.stdout.write('\n') sys.stdout.write('\n')
def do_edit(self, line):
self.event.load()
self.edit_event()
def do_info(self, line): def do_info(self, line):
self.event.load() self.event.load()
print self.event.title print self.event.title
@ -312,9 +353,9 @@ class Boobathon(ReplApplication):
print self.event.description print self.event.description
print '' print ''
print 'Date:', self.event.date print 'Date:', self.event.date
print 'Start:', self.event.start print 'Begin:', self.event.begin
print 'End:', self.event.end print 'End:', self.event.end
print 'Duration:', self.event.duration print 'Duration:', self.event.format_duration()
print 'Location:', self.event.location print 'Location:', self.event.location
print '' print ''
print 'There are %d members, use the "members" command to list them' % len(self.event.members) print 'There are %d members, use the "members" command to list them' % len(self.event.members)
@ -331,7 +372,7 @@ class Boobathon(ReplApplication):
print 'Hardware:', member.hardware print 'Hardware:', member.hardware
accompl = 0 accompl = 0
for task in member.tasks: for task in member.tasks:
if task.done: if task.status == task.STATUS_DONE:
accompl += 1 accompl += 1
print '%d tasks (%d accomplished)' % (len(member.tasks), accompl) print '%d tasks (%d accomplished)' % (len(member.tasks), accompl)
print '' print ''
@ -358,8 +399,89 @@ class Boobathon(ReplApplication):
else: else:
self.save_event('Left the event') self.save_event('Left the event')
def do_addtask(self, line):
self.event.load()
mem = self.event.get_me()
if not mem:
print 'You have not joined this event.'
return
backend, capability = self.parse_command_args(line, 2, 2)
for task in mem.tasks:
if (task.backend,task.capability) == (backend,capability):
print 'A task already exists for that.'
return
task = Task(backend, capability)
mem.tasks.append(task)
self.save_event('New task')
def do_start(self, line):
self.event.load()
mem = self.event.get_me()
if not mem:
print 'You have not joined this event.'
return
if len(mem.tasks) == 0:
print "You don't have any task to do."
return
if not self.event.currently_in_event():
print "You can't start a task, we are not in event."
return
if line.isdigit():
task_id = int(line)
else:
task_id = -1
last_done = -1
for i, task in enumerate(mem.tasks):
if task.status == task.STATUS_DONE:
last_done = i
elif task.status == task.STATUS_PROGRESS:
task.status = task.STATUS_NONE
if i == task_id or task.status == task.STATUS_NONE:
break
else:
print 'Task not found.'
if task.status == task.STATUS_DONE:
print 'Task is already done.'
return
task.status = task.STATUS_PROGRESS
mem.tasks.remove(task)
mem.tasks.insert(last_done, task)
self.save_event('Started a task')
def do_done(self, line): def do_done(self, line):
self.event.load() self.event.load()
mem = self.event.get_me()
if not mem:
print 'You have not joined this event.'
return
if len(mem.tasks) == 0:
print "You don't have any task to do."
return
if not self.event.currently_in_event():
print "You can't start a task, we are not in event."
return
for task in mem.tasks:
if task.status == task.STATUS_PROGRESS:
print 'Task: %s,%s' % (task.backend, task.capability)
task.status = task.STATUS_DONE
task.date = datetime.now()
task.branch = self.ask('Enter name of branch')
self.save_event('Task accomplished')
return
print "There isn't any task in progress."
def load_default_backends(self): def load_default_backends(self):
""" """