Index: rietveld.py |
diff --git a/rietveld.py b/rietveld.py |
index 028b0b00d8ea3d20d7bd729f6b3a0f700ac95d51..bd9139182e5754f99e2990633fb216fbfb2a9282 100644 |
--- a/rietveld.py |
+++ b/rietveld.py |
@@ -278,6 +278,12 @@ class Rietveld(object): |
('xsrf_token', self.xsrf_token()), |
(flag, str(value))]) |
+ def set_flags(self, issue, patchset, flags): |
+ return self.post('/%d/edit_flags' % issue, [ |
+ ('last_patchset', str(patchset)), |
+ ('xsrf_token', self.xsrf_token()), |
+ ] + [(flag, str(value)) for flag, value in flags.iteritems()]) |
+ |
def search( |
self, |
owner=None, reviewer=None, |
@@ -744,6 +750,10 @@ class ReadOnlyRietveld(object): |
(flag, value, issue)) |
ReadOnlyRietveld._local_changes.setdefault(issue, {})[flag] = value |
+ def set_flags(self, issue, patchset, flags): |
+ for flag, value in flags.iteritems(): |
+ self.set_flag(issue, patchset, flag, value) |
+ |
def trigger_try_jobs( # pylint:disable=R0201 |
self, issue, patchset, reason, clobber, revision, builders_and_tests, |
master=None, category='cq'): |