[Findit] Re-org code.

appengine/findit/common/rietveld.py

-# Copyright 2016 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.
-
-# TODO: In the new layout, this should move to the ./services or
-# ./services/waterfall_app directories, since it is only used by Waterfall.
-
-# TODO: we ought to abstract over the HTTP_CLIENT member (which is only
-# used by the Post method) by passing it to the constructor. That way
-# things are more losely coupled, improving modularity and reducing
-# fragility. In addition, for easier mocking, we may want to just have
-# the thing passed for HTTP_CLIENT to be ``callable``, rather than giving
-# a name to the method we use on that object.
-
-import logging
-import re
-import urlparse
-
-from common.codereview import CodeReview
-from common.http_client_appengine import HttpClientAppengine
-
-
-_RIETVELD_ISSUE_NUMBER_RE = re.compile('^/(\d+)/?.*')
-
-
-class Rietveld(CodeReview):
-  """The implementation of CodeReview interface for Rietveld."""
-  HTTP_CLIENT = HttpClientAppengine(follow_redirects=False)
-
-  def _GetXsrfToken(self, rietveld_url):
-    """Returns the xsrf token for follow-up requests."""
-    headers = {
-        'X-Requesting-XSRF-Token': '1',
-        'Accept': 'text/plain',
-    }
-    url = '%s/xsrf_token' % rietveld_url
-    status_code, xsrf_token = self.HTTP_CLIENT.Post(
-        url, data=None, headers=headers)
-    if status_code != 200:
-      logging.error('Failed to get xsrf token from %s', rietveld_url)
-      xsrf_token = None
-    return xsrf_token
-
-  def _GetRietveldUrlAndIssueNumber(self, issue_url):
-    """Parses the given Rietveld issue url.
-
-    Args:
-      issue_url(str): The url to an issue on Rietveld.
-    Returns:
-      (rietveld_url, issue_number)
-      rietveld_url(str): The root url of the Rietveld app.
-      issue_number(str): The issue number.
-    """
-    u = urlparse.urlparse(issue_url)
-    rietveld_url = 'https://%s' % u.netloc  # Enforces https.
-    issue_number = _RIETVELD_ISSUE_NUMBER_RE.match(u.path).group(1)
-    return rietveld_url, issue_number
-
-  def _EncodeMultipartFormData(self, fields):
-    """Encodes form fields for multipart/form-data"""
-    BOUNDARY = '-F-I-N-D-I-T-M-E-S-S-A-G-E-'
-    CRLF = '\r\n'
-    lines = []
-    for key, value in fields.iteritems():
-      lines.append('--' + BOUNDARY)
-      lines.append('Content-Disposition: form-data; name="%s"' % key)
-      lines.append('')
-      lines.append(str(value))
-    lines.append('--' + BOUNDARY + '--')
-    lines.append('')
-    body = CRLF.join(lines)
-    content_type = 'multipart/form-data; boundary=%s' % BOUNDARY
-    return content_type, body
-
-  def PostMessage(self, issue_url, message):
-    rietveld_url, issue_number = self._GetRietveldUrlAndIssueNumber(issue_url)
-    url = '%s/%s/publish' % (rietveld_url, issue_number)
-    xsrf_token = self._GetXsrfToken(rietveld_url)
-    if not xsrf_token:
-      return False
-    form_fields = {
-        'xsrf_token': xsrf_token,
-        'message': message,
-        'message_only': 'True',
-        'add_as_reviewer': 'False',
-        'send_mail': 'True',
-        'no_redirect': 'True',
-    }
-    content_type, body = self._EncodeMultipartFormData(form_fields)
-    headers = {
-        'Content-Type': content_type,
-        'Accept': 'text/plain',
-    }
-    status_code, content = self.HTTP_CLIENT.Post(
-        url, data=body, headers=headers)
-    return status_code == 200 and content == 'OK'