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

Allow import of harvest time entries w/ missing project

Browse source
This commit is contained in:
Lance Edgar 2022-01-29 19:33:03 -06:00
parent 2a0c9bf015
commit 259d3b0f33
2 changed files with 62 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

51
rattail_harvest/importing/model.py
View file

@ -24,10 +24,15 @@
rattail-harvest model importers
"""
import logging
from rattail.importing.model import ToRattail
from rattail_harvest.db import model
log = logging.getLogger(__name__)
##############################
# harvest cache models
##############################
@ -47,7 +52,53 @@ class HarvestTaskImporter(ToRattail):
class HarvestTimeEntryImporter(ToRattail):
model_class = model.HarvestTimeEntry
# flags to auto-create records for "unknown" references
auto_create_unknown_project = True
# flags to log warning vs. debug for "unknown" references
warn_for_unknown_project = True
def setup(self):
super(HarvestTimeEntryImporter, self).setup()
model = self.model
if 'project_id' in self.fields:
self.harvest_projects_by_id = self.app.cache_model(
self.session, model.HarvestProject, key='id')
def cache_query(self):
query = super(HarvestTimeEntryImporter, self).cache_query()
return query.filter(self.model_class.spent_date >= self.start_date)\
.filter(self.model_class.spent_date <= self.end_date)
def get_harvest_project(self, project_id):
if hasattr(self, 'harvest_projects_by_id'):
return self.harvest_projects_by_id.get(project_id)
model = self.model
return self.session.query(model.HarvestProject)\
.filter(model.HarvestProject.id == project_id)\
.first()
def update_object(self, entry, data, local_data=None):
entry = super(HarvestTimeEntryImporter, self).update_object(entry, data, local_data)
model = self.model
if 'project_id' in self.fields:
project_id = data['project_id']
project = self.get_harvest_project(project_id)
if not project:
logger = log.warning if self.warn_for_unknown_project else log.debug
logger("unknown project id %s for time entry id %s: %s",
project_id, entry.id, entry)
if self.auto_create_unknown_project:
project = model.HarvestProject()
project.id = project_id
project.name = "(unknown)"
self.session.add(project)
if hasattr(self, 'harvest_projects_by_id'):
self.harvest_projects_by_id[project_id] = project
elif entry.project_id:
entry.project_id = None
return entry