blob: 9ad3ea7b953bdc2142ffce94e606ec6ad8135032 [file] [log] [blame]
# Copyright (c) 2013 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import json
import patching
import util
class Patch(object):
"""Helper class for lazily loading and parsing patch data."""
def __init__(self, rietveld, issue_id, patchset_id, patch_id):
self.rietveld = rietveld
self.issue_id = issue_id
self.patchset_id = patchset_id
self.patch_id = patch_id
@util.lazy_property
def raw(self):
return self.rietveld.post_data(
'download/issue%s_%s_%s.diff' %
(self.issue_id, self.patchset_id, self.patch_id))
@util.lazy_property
def lines(self):
return patching.ParsePatchToLines(self.raw.splitlines())
class Review(object):
"""Represents a code review.
Information from rietveld can be obtained via the following properties:
- |issue_id| is the issue identifier.
- |issue_data| contains issue meta data as retrieved from rietveld. The data
is pulled lazily from the rietveld API on first access.
- |patchsets| has lazily-pulled patchset meta data, indexed by patchset IDa.
The subclass may then do its processing and trigger any actions. In
particular, the |rietveld| object may be used to update rietveld issue state.
"""
def __init__(self, rietveld, issue_id):
self.rietveld = rietveld
self.issue_id = issue_id
@util.lazy_property
def issue_data(self):
json_data = self.rietveld.post_data('api/%s?messages=true' % self.issue_id)
data = json.loads(json_data)
data['messages'] = [util.ObjectDict(msg) for msg in data['messages']]
return util.ObjectDict(data)
@util.lazy_property
def patchsets(self):
def retrieve_patchset(ps):
json_patchset_data = self.rietveld.post_data('api/%s/%s' %
(self.issue_id, ps))
patchset_data = json.loads(json_patchset_data)
# Amend the files property so it can lazily load and return patch data.
for file_data in patchset_data.get('files', {}).values():
file_data['patch'] = Patch(self.rietveld, self.issue_id, ps,
file_data['id'])
return util.ObjectDict(patchset_data)
return util.LazyDict(retrieve_patchset)
@util.lazy_property
def latest_patchset(self):
return self.patchsets[self.issue_data.patchsets[-1]]