| #!/usr/bin/env python |
| # Copyright (C) 2010 Google Inc. All rights reserved. |
| # |
| # Redistribution and use in source and binary forms, with or without |
| # modification, are permitted provided that the following conditions are |
| # met: |
| # |
| # * Redistributions of source code must retain the above copyright |
| # notice, this list of conditions and the following disclaimer. |
| # * Redistributions in binary form must reproduce the above |
| # copyright notice, this list of conditions and the following disclaimer |
| # in the documentation and/or other materials provided with the |
| # distribution. |
| # * Neither the name of Google Inc. nor the names of its |
| # contributors may be used to endorse or promote products derived from |
| # this software without specific prior written permission. |
| # |
| # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| from __future__ import with_statement |
| |
| import codecs |
| import mimetypes |
| import socket |
| import urllib2 |
| |
| from webkitpy.common.net.networktransaction import NetworkTransaction |
| |
| def get_mime_type(filename): |
| return mimetypes.guess_type(filename)[0] or 'application/octet-stream' |
| |
| |
| def _encode_multipart_form_data(fields, files): |
| """Encode form fields for multipart/form-data. |
| |
| Args: |
| fields: A sequence of (name, value) elements for regular form fields. |
| files: A sequence of (name, filename, value) elements for data to be |
| uploaded as files. |
| Returns: |
| (content_type, body) ready for httplib.HTTP instance. |
| |
| Source: |
| http://code.google.com/p/rietveld/source/browse/trunk/upload.py |
| """ |
| BOUNDARY = '-M-A-G-I-C---B-O-U-N-D-A-R-Y-' |
| CRLF = '\r\n' |
| lines = [] |
| |
| for key, value in fields: |
| lines.append('--' + BOUNDARY) |
| lines.append('Content-Disposition: form-data; name="%s"' % key) |
| lines.append('') |
| if isinstance(value, unicode): |
| value = value.encode('utf-8') |
| lines.append(value) |
| |
| for key, filename, value in files: |
| lines.append('--' + BOUNDARY) |
| lines.append('Content-Disposition: form-data; name="%s"; filename="%s"' % (key, filename)) |
| lines.append('Content-Type: %s' % get_mime_type(filename)) |
| lines.append('') |
| if isinstance(value, unicode): |
| value = value.encode('utf-8') |
| lines.append(value) |
| |
| lines.append('--' + BOUNDARY + '--') |
| lines.append('') |
| body = CRLF.join(lines) |
| content_type = 'multipart/form-data; boundary=%s' % BOUNDARY |
| return content_type, body |
| |
| |
| class TestResultsUploader: |
| def __init__(self, host): |
| self._host = host |
| |
| def _upload_files(self, attrs, file_objs): |
| url = "http://%s/testfile/upload" % self._host |
| content_type, data = _encode_multipart_form_data(attrs, file_objs) |
| headers = {"Content-Type": content_type} |
| request = urllib2.Request(url, data, headers) |
| urllib2.urlopen(request) |
| |
| def upload(self, params, files, timeout_seconds): |
| file_objs = [] |
| for filename, path in files: |
| with codecs.open(path, "rb") as file: |
| file_objs.append(('file', filename, file.read())) |
| |
| orig_timeout = socket.getdefaulttimeout() |
| try: |
| socket.setdefaulttimeout(timeout_seconds) |
| NetworkTransaction(timeout_seconds=timeout_seconds).run( |
| lambda: self._upload_files(params, file_objs)) |
| finally: |
| socket.setdefaulttimeout(orig_timeout) |