tailbone-harvest/tailbone_harvest/views/harvest/time_entries.py

# -*- coding: utf-8; -*-
################################################################################
#
#  Rattail -- Retail Software Framework
#  Copyright © 2010-2023 Lance Edgar
#
#  This file is part of Rattail.
#
#  Rattail is free software: you can redistribute it and/or modify it under the
#  terms of the GNU General Public License as published by the Free Software
#  Foundation, either version 3 of the License, or (at your option) any later
#  version.
#
#  Rattail is distributed in the hope that it will be useful, but WITHOUT ANY
#  WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
#  FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
#  details.
#
#  You should have received a copy of the GNU General Public License along with
#  Rattail.  If not, see <http://www.gnu.org/licenses/>.
#
################################################################################
"""
Harvest Time Entry views
"""

from rattail_harvest.db.model import HarvestTimeEntry
from rattail_harvest.harvest.config import get_harvest_url

from .master import HarvestMasterView


class HarvestTimeEntryView(HarvestMasterView):
    """
    Master view for Harvest Time Entries
    """
    model_class = HarvestTimeEntry
    url_prefix = '/harvest/time-entries'
    route_prefix = 'harvest.time_entries'

    grid_columns = [
        'id',
        'spent_date',
        'user',
        'client',
        'project',
        'task',
        'hours',
        'notes',
    ]

    def configure_grid(self, g):
        super(HarvestTimeEntryView, self).configure_grid(g)

        g.set_type('hours', 'duration_hours')

        g.set_sort_defaults('spent_date', 'desc')

        g.set_link('id')
        g.set_link('user')
        g.set_link('client')
        g.set_link('notes')

    def configure_form(self, f):
        super(HarvestTimeEntryView, self).configure_form(f)

        # make sure id is first field
        f.remove('id')
        f.insert(0, 'id')

        # user
        f.remove('user_id')
        f.set_renderer('user', self.render_harvest_user)

        # client
        f.remove('client_id')
        f.set_renderer('client', self.render_harvest_client)

        # project
        f.remove('project_id')
        f.set_renderer('project', self.render_harvest_project)

        # task
        f.remove('task_id')
        f.set_renderer('task', self.render_harvest_task)

        f.set_type('notes', 'text')

        f.set_type('billable_rate', 'currency')
        f.set_type('cost_rate', 'currency')

    def get_xref_buttons(self, entry):
        buttons = super(HarvestTimeEntryView, self).get_xref_buttons(entry)
        model = self.model

        # harvest
        url = get_harvest_url(self.rattail_config)
        if url:
            url = '{}/time/day/{}/{}'.format(
                url,
                entry.spent_date.strftime('%Y/%m/%d'),
                entry.user_id)
            buttons.append(self.make_xref_button(url=url,
                                                 text="View in Harvest"))

        return buttons


def defaults(config, **kwargs):
    base = globals()

    HarvestTimeEntryView = kwargs.get('HarvestTimeEntryView', base['HarvestTimeEntryView'])
    HarvestTimeEntryView.defaults(config)


def includeme(config):
    defaults(config)