mirror of
https://github.com/GothenburgBitFactory/timewarrior.git
synced 2025-06-26 10:54:28 +02:00
extensions: Make extensions conform with PEP8
This commit is contained in:
Thomas Lauf
parent
97ab53afeb
commit
310f48973e
4 changed files with 114 additions and 111 deletions
18
ext/csv.py
18
ext/csv.py
|
|
@ -5,23 +5,23 @@ import json
|
|||
|
||||
# Skip the configuration settings.
|
||||
for line in sys.stdin:
|
||||
if line == '\n':
|
||||
break;
|
||||
if line == '\n':
|
||||
break;
|
||||
|
||||
# Extract the JSON.
|
||||
doc = ''
|
||||
for line in sys.stdin:
|
||||
doc += line
|
||||
doc += line
|
||||
|
||||
total_active_time = 0
|
||||
|
||||
j = json.loads(doc)
|
||||
for object in j:
|
||||
line = '"%s",' % object['start']
|
||||
line +='"%s"' % (object['end'] if 'end' in object else '')
|
||||
line = '"%s",' % object['start']
|
||||
line +='"%s"' % (object['end'] if 'end' in object else '')
|
||||
|
||||
if 'tags' in object:
|
||||
for tag in object['tags']:
|
||||
line += ',"%s"' % tag
|
||||
if 'tags' in object:
|
||||
for tag in object['tags']:
|
||||
line += ',"%s"' % tag
|
||||
|
||||
print line
|
||||
print line
|
||||
|
|
|
|||
|
|
@ -3,4 +3,4 @@
|
|||
import sys
|
||||
|
||||
for line in sys.stdin:
|
||||
print line.strip()
|
||||
print line.strip()
|
||||
|
|
|
|||
|
|
@ -1,5 +1,5 @@
|
|||
#!/usr/bin/env python
|
||||
################################################################################
|
||||
###############################################################################
|
||||
#
|
||||
# Copyright 2015 - 2016, Paul Beckingham, Federico Hernandez.
|
||||
#
|
||||
|
|
@ -23,7 +23,7 @@
|
|||
#
|
||||
# http://www.opensource.org/licenses/mit-license.php
|
||||
#
|
||||
################################################################################
|
||||
###############################################################################
|
||||
|
||||
import sys
|
||||
import json
|
||||
|
|
@ -45,28 +45,28 @@ print(json.dumps(new))
|
|||
tags = [new['description']]
|
||||
|
||||
if 'project' in new:
|
||||
project = new['project']
|
||||
tags.append(project)
|
||||
if '.' in project:
|
||||
tags.extend([tag for tag in project.split('.')])
|
||||
project = new['project']
|
||||
tags.append(project)
|
||||
if '.' in project:
|
||||
tags.extend([tag for tag in project.split('.')])
|
||||
|
||||
if 'tags' in new:
|
||||
tags.extend(new['tags'])
|
||||
tags.extend(new['tags'])
|
||||
|
||||
combined = ' '.join(['"%s"' % tag for tag in tags]).encode('utf-8').strip()
|
||||
|
||||
# Started task.
|
||||
if 'start' in new and not 'start' in old:
|
||||
os.system('timew start ' + combined.decode() + ' :yes')
|
||||
os.system('timew start ' + combined.decode() + ' :yes')
|
||||
|
||||
# Stopped task.
|
||||
elif not 'start' in new and 'start' in old:
|
||||
os.system('timew stop ' + combined.decode() + ' :yes')
|
||||
os.system('timew stop ' + combined.decode() + ' :yes')
|
||||
|
||||
# TI-51 - in the taskwarrior hook, deleting a task doesn't stop the watch
|
||||
#
|
||||
# Broadened to: Any task that is active, with a non-pending status should not
|
||||
# be tracked.
|
||||
elif 'start' in new and new['status'] != 'pending':
|
||||
os.system('timew stop ' + combined.decode() + ' :yes')
|
||||
os.system('timew stop ' + combined.decode() + ' :yes')
|
||||
|
||||
|
|
|
|||
185
ext/totals.py
185
ext/totals.py
|
|
@ -1,42 +1,45 @@
|
|||
#!/usr/bin/env python
|
||||
################################################################################
|
||||
##
|
||||
## Copyright 2015 - 2016, Paul Beckingham, Federico Hernandez.
|
||||
##
|
||||
## Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
## of this software and associated documentation files (the "Software"), to deal
|
||||
## in the Software without restriction, including without limitation the rights
|
||||
## to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
## copies of the Software, and to permit persons to whom the Software is
|
||||
## furnished to do so, subject to the following conditions:
|
||||
##
|
||||
## The above copyright notice and this permission notice shall be included
|
||||
## in all copies or substantial portions of the Software.
|
||||
##
|
||||
## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
||||
## OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
## FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
||||
## THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
## LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
## OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
## SOFTWARE.
|
||||
##
|
||||
## http:##www.opensource.org/licenses/mit-license.php
|
||||
##
|
||||
################################################################################
|
||||
###############################################################################
|
||||
#
|
||||
# Copyright 2015 - 2016, Paul Beckingham, Federico Hernandez.
|
||||
#
|
||||
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
# of this software and associated documentation files (the "Software"), to deal
|
||||
# in the Software without restriction, including without limitation the rights
|
||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
# copies of the Software, and to permit persons to whom the Software is
|
||||
# furnished to do so, subject to the following conditions:
|
||||
#
|
||||
# The above copyright notice and this permission notice shall be included
|
||||
# in all copies or substantial portions of the Software.
|
||||
#
|
||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
||||
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
||||
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
# SOFTWARE.
|
||||
#
|
||||
# http://www.opensource.org/licenses/mit-license.php
|
||||
#
|
||||
###############################################################################
|
||||
|
||||
import sys
|
||||
import json
|
||||
import datetime
|
||||
|
||||
def formatSeconds(seconds):
|
||||
''' Convert seconds: 3661
|
||||
To formatted: 1:01:01
|
||||
'''
|
||||
hours = int(seconds / 3600)
|
||||
minutes = int(seconds % 3600) / 60
|
||||
seconds = seconds % 60
|
||||
return '%4d:%02d:%02d' % (hours, minutes, seconds)
|
||||
|
||||
def format_seconds(seconds):
|
||||
"""Convert seconds to a formatted string
|
||||
|
||||
Convert seconds: 3661
|
||||
To formatted: 1:01:01
|
||||
"""
|
||||
hours = int(seconds / 3600)
|
||||
minutes = int(seconds % 3600) / 60
|
||||
seconds = seconds % 60
|
||||
return '%4d:%02d:%02d' % (hours, minutes, seconds)
|
||||
|
||||
|
||||
DATEFORMAT = '%Y%m%dT%H%M%SZ'
|
||||
|
|
@ -46,87 +49,87 @@ header = 1
|
|||
configuration = dict()
|
||||
body = ''
|
||||
for line in sys.stdin:
|
||||
if header:
|
||||
if line == '\n':
|
||||
header = 0
|
||||
if header:
|
||||
if line == '\n':
|
||||
header = 0
|
||||
else:
|
||||
fields = line.strip().split(': ', 2)
|
||||
if len(fields) == 2:
|
||||
configuration[fields[0]] = fields[1]
|
||||
else:
|
||||
configuration[fields[0]] = ''
|
||||
else:
|
||||
fields = line.strip().split(': ', 2)
|
||||
if len(fields) == 2:
|
||||
configuration[fields[0]] = fields[1]
|
||||
else:
|
||||
configuration[fields[0]] = ''
|
||||
else:
|
||||
body += line
|
||||
body += line
|
||||
|
||||
# Sum the second tracked by tag.
|
||||
totals = dict()
|
||||
untagged = None
|
||||
j = json.loads(body)
|
||||
for object in j:
|
||||
start = datetime.datetime.strptime(object['start'], DATEFORMAT)
|
||||
start = datetime.datetime.strptime(object['start'], DATEFORMAT)
|
||||
|
||||
if 'end' in object:
|
||||
end = datetime.datetime.strptime(object['end'], DATEFORMAT)
|
||||
else:
|
||||
end = datetime.datetime.utcnow()
|
||||
|
||||
tracked = end - start
|
||||
|
||||
if 'tags' not in object or object['tags'] == []:
|
||||
if untagged is None:
|
||||
untagged = tracked
|
||||
if 'end' in object:
|
||||
end = datetime.datetime.strptime(object['end'], DATEFORMAT)
|
||||
else:
|
||||
untagged += tracked
|
||||
else:
|
||||
for tag in object['tags']:
|
||||
if tag in totals:
|
||||
totals[tag] += tracked
|
||||
else:
|
||||
totals[tag] = tracked
|
||||
end = datetime.datetime.utcnow()
|
||||
|
||||
tracked = end - start
|
||||
|
||||
if 'tags' not in object or object['tags'] == []:
|
||||
if untagged is None:
|
||||
untagged = tracked
|
||||
else:
|
||||
untagged += tracked
|
||||
else:
|
||||
for tag in object['tags']:
|
||||
if tag in totals:
|
||||
totals[tag] += tracked
|
||||
else:
|
||||
totals[tag] = tracked
|
||||
|
||||
# Determine largest tag width.
|
||||
max_width = len('Total')
|
||||
for tag in totals:
|
||||
if len(tag) > max_width:
|
||||
max_width = len(tag)
|
||||
if len(tag) > max_width:
|
||||
max_width = len(tag)
|
||||
|
||||
if 'temp.report.start' not in configuration:
|
||||
print 'There is no data in the database'
|
||||
exit()
|
||||
print 'There is no data in the database'
|
||||
exit()
|
||||
|
||||
start = datetime.datetime.strptime(configuration['temp.report.start'], DATEFORMAT)
|
||||
end = datetime.datetime.strptime(configuration['temp.report.end'], DATEFORMAT)
|
||||
end = datetime.datetime.strptime(configuration['temp.report.end'], DATEFORMAT)
|
||||
|
||||
if max_width > 0:
|
||||
# Compose report header.
|
||||
print '\nTotal by Tag, for %s - %s\n' % (start, end)
|
||||
# Compose report header.
|
||||
print '\nTotal by Tag, for %s - %s\n' % (start, end)
|
||||
|
||||
# Compose table header.
|
||||
if configuration['color'] == 'on':
|
||||
print '[4m%-*s[0m [4m%10s[0m' % (max_width, 'Tag', 'Total')
|
||||
else:
|
||||
print '%-*s %10s' % (max_width, 'Tag', 'Total')
|
||||
print '-' * max_width, '----------'
|
||||
# Compose table header.
|
||||
if configuration['color'] == 'on':
|
||||
print '[4m%-*s[0m [4m%10s[0m' % (max_width, 'Tag', 'Total')
|
||||
else:
|
||||
print '%-*s %10s' % (max_width, 'Tag', 'Total')
|
||||
print '-' * max_width, '----------'
|
||||
|
||||
# Compose table rows.
|
||||
grand_total = 0
|
||||
for tag in sorted(totals):
|
||||
formatted = formatSeconds(totals[tag].seconds)
|
||||
grand_total += totals[tag].seconds
|
||||
print '%-*s %10s' % (max_width, tag, formatted)
|
||||
# Compose table rows.
|
||||
grand_total = 0
|
||||
for tag in sorted(totals):
|
||||
formatted = format_seconds(totals[tag].seconds)
|
||||
grand_total += totals[tag].seconds
|
||||
print '%-*s %10s' % (max_width, tag, formatted)
|
||||
|
||||
if untagged is not None:
|
||||
formatted = formatSeconds(untagged.seconds)
|
||||
grand_total += untagged.seconds
|
||||
print '%-*s %10s' % (max_width, '', formatted)
|
||||
if untagged is not None:
|
||||
formatted = format_seconds(untagged.seconds)
|
||||
grand_total += untagged.seconds
|
||||
print '%-*s %10s' % (max_width, '', formatted)
|
||||
|
||||
# Compose total.
|
||||
if configuration['color'] == 'on':
|
||||
print ' ' * max_width, '[4m [0m'
|
||||
else:
|
||||
print ' ' * max_width, '----------'
|
||||
# Compose total.
|
||||
if configuration['color'] == 'on':
|
||||
print ' ' * max_width, '[4m [0m'
|
||||
else:
|
||||
print ' ' * max_width, '----------'
|
||||
|
||||
print '%-*s %10s' % (max_width, 'Total', formatSeconds(grand_total))
|
||||
print '%-*s %10s' % (max_width, 'Total', format_seconds(grand_total))
|
||||
|
||||
else:
|
||||
print 'No data in the range %s - %s' % (start, end)
|
||||
print 'No data in the range %s - %s' % (start, end)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue