def __init__(self, proxy=None):
if proxy is None:
self.proxy = Client(settings.TRAC_USER, settings.TRAC_PASSWORD)
else:
self.proxy = proxy
class Importer(object):
def __init__(self, proxy=None):
if proxy is None:
self.proxy = Client(settings.TRAC_USER, settings.TRAC_PASSWORD)
else:
self.proxy = proxy
def sync(self, since):
ticket_ids = self.proxy.get_recent_changes(since)
for ticket_id in ticket_ids:
self.sync_ticket(ticket_id)
return self.associate_sprinters_with_changes(since)
def sync_ticket(self, ticket_id):
ticket_data = self.proxy.get_ticket(ticket_id)
id_, created, changed, attributes = ticket_data
defaults = {
'kind': attributes['type'],
'component': attributes['component'],
'resolution': attributes['resolution'],
'severity': attributes['severity'],
'status': attributes['status'],
}
ticket, was_created = Ticket.objects.get_or_create(
pk=ticket_id, defaults=defaults)
self.sync_ticket_changes(ticket)
def sync_ticket_changes(self, ticket):
ticket_changelog_data = self.proxy.get_ticket_changelog(ticket.pk)
last_recorded_timestamp = self.last_recorded_timestamp(ticket)
for data in ticket_changelog_data:
time, author, field, old_value, new_value, permanent = data
timestamp = self.standardize_datetime(time)
if timestamp > last_recorded_timestamp:
ticket.changes.create( # TODO:get_or_create!!!!
timestamp=timestamp, author=author, field=field,
old_value=old_value, new_value=new_value)
def standardize_datetime(self, time):
timestamp = datetime.strptime(time.value, "%Y%m%dT%H:%M:%S")
return make_aware(timestamp, utc)
def last_recorded_timestamp(self, ticket):
try:
last_recorded_change = ticket.changes.latest('timestamp')
last_recorded_timestamp = last_recorded_change.timestamp
except Change.DoesNotExist:
last_recorded_timestamp = make_aware(datetime(1970, 1, 1), utc)
return last_recorded_timestamp
def associate_sprinters_with_changes(self, since):
ticket_changes = Change.objects.filter(timestamp__gt=since)
sprinters = set()
for change in ticket_changes:
try:
sprinter = Sprinter.objects.get_by_trac_author(change.author)
sprinters.add(sprinter)
except Sprinter.DoesNotExist:
continue
frozen_ticket = change.ticket_snapshot()
sprinter.changes.create( # TODO:get_or_create
ticket_change=change, field=change.field,
ticket_id=change.ticket_id, **frozen_ticket.attrs)
return sprinters