rattail/tailbone
2
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity Actions

Add basic Excel download support for raw worked shifts

Browse source 
also, tweak response per python3
This commit is contained in:
Lance Edgar 2018-10-09 14:10:36 -05:00
parent 362173ef10
commit 94ba18eaee
2 changed files with 30 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tailbone/views/master.py
View file

@ -2145,8 +2145,8 @@ class MasterView(View):
os.remove(path)
response.content_length = len(response.body)
response.content_type = b'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
response.content_disposition = b'attachment; filename={}.xlsx'.format(self.get_grid_key())
response.content_type = str('application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
response.content_disposition = str('attachment; filename={}.xlsx').format(self.get_grid_key())
return response
def get_xlsx_fields(self):

28
tailbone/views/shifts/core.py
View file

@ -93,6 +93,7 @@ class WorkedShiftsView(MasterView):
"""
model_class = model.WorkedShift
url_prefix = '/shifts/worked'
results_downloadable_xlsx = True
has_versions = True
grid_columns = [
@ -125,6 +126,10 @@ class WorkedShiftsView(MasterView):
# TODO: these sorters should be automatic once we fix the schema
g.sorters['start_time'] = g.make_sorter(model.WorkedShift.punch_in)
g.sorters['end_time'] = g.make_sorter(model.WorkedShift.punch_out)
# TODO: same goes for these renderers
g.set_type('start_time', 'datetime')
g.set_type('end_time', 'datetime')
# (but we'll still have to set this)
g.set_sort_defaults('start_time', 'desc')
g.set_renderer('length', render_shift_length)
@ -157,6 +162,29 @@ class WorkedShiftsView(MasterView):
url = self.request.route_url('employees.view', uuid=employee.uuid)
return tags.link_to(text, url)
def get_xlsx_fields(self):
fields = super(WorkedShiftsView, self).get_xlsx_fields()
# add employee name
i = fields.index('employee_uuid')
fields.insert(i + 1, 'employee_name')
return fields
def get_xlsx_row(self, shift, fields):
row = super(WorkedShiftsView, self).get_xlsx_row(shift, fields)
# add custom fields
row['employee_name'] = shift.employee.person.display_name
# localize start and end times (Excel requires time with no zone)
if shift.punch_in:
row['punch_in'] = localtime(self.rattail_config, shift.punch_in, from_utc=True, tzinfo=False)
if shift.punch_out:
row['punch_out'] = localtime(self.rattail_config, shift.punch_out, from_utc=True, tzinfo=False)
return row
def includeme(config):
ScheduledShiftsView.defaults(config)