gam7 7.3.4__py3-none-any.whl

gam/atom/http_interface.py

@@ -0,0 +1,144 @@
+#
+# Copyright (C) 2008 Google Inc.
+#
+# Licensed under the Apache License 2.0;
+
+
+"""This module provides a common interface for all HTTP requests.
+
+  HttpResponse: Represents the server's response to an HTTP request. Provides
+      an interface identical to httplib.HTTPResponse which is the response
+      expected from higher level classes which use HttpClient.request.
+
+  GenericHttpClient: Provides an interface (superclass) for an object 
+      responsible for making HTTP requests. Subclasses of this object are
+      used in AtomService and GDataService to make requests to the server. By
+      changing the http_client member object, the AtomService is able to make
+      HTTP requests using different logic (for example, when running on 
+      Google App Engine, the http_client makes requests using the App Engine
+      urlfetch API). 
+"""
+
+# __author__ = 'api.jscudder (Jeff Scudder)'
+
+import io
+
+USER_AGENT = '%s GData-Python/2.0.18'
+
+
+class Error(Exception):
+    pass
+
+
+class UnparsableUrlObject(Error):
+    pass
+
+
+class ContentLengthRequired(Error):
+    pass
+
+
+class HttpResponse(object):
+    def __init__(self, body=None, status=None, reason=None, headers=None):
+        """Constructor for an HttpResponse object.
+
+        HttpResponse represents the server's response to an HTTP request from
+        the client. The HttpClient.request method returns a httplib.HTTPResponse
+        object and this HttpResponse class is designed to mirror the interface
+        exposed by httplib.HTTPResponse.
+
+        Args:
+          body: A file like object, with a read() method. The body could also
+              be a string, and the constructor will wrap it so that
+              HttpResponse.read(self) will return the full string.
+          status: The HTTP status code as an int. Example: 200, 201, 404.
+          reason: The HTTP status message which follows the code. Example:
+              OK, Created, Not Found
+          headers: A dictionary containing the HTTP headers in the server's
+              response. A common header in the response is Content-Length.
+        """
+        if body:
+            if hasattr(body, 'read'):
+                self._body = body
+            else:
+                self._body = io.StringIO(body)
+        else:
+            self._body = None
+        if status is not None:
+            self.status = int(status)
+        else:
+            self.status = None
+        self.reason = reason
+        self._headers = headers or {}
+
+    def getheader(self, name, default=None):
+        if name in self._headers:
+            return self._headers[name]
+        else:
+            return default
+
+    def read(self, amt=None):
+        if not amt:
+            return self._body.read()
+        else:
+            return self._body.read(amt)
+
+
+class GenericHttpClient(object):
+    debug = False
+
+    def __init__(self, http_client, headers=None):
+        """
+
+        Args:
+          http_client: An object which provides a request method to make an HTTP
+              request. The request method in GenericHttpClient performs a
+              call-through to the contained HTTP client object.
+          headers: A dictionary containing HTTP headers which should be included
+              in every HTTP request. Common persistent headers include
+              'User-Agent'.
+        """
+        self.http_client = http_client
+        self.headers = headers or {}
+
+    def request(self, operation, url, data=None, headers=None):
+        all_headers = self.headers.copy()
+        if headers:
+            all_headers.update(headers)
+        return self.http_client.request(operation, url, data=data,
+                                        headers=all_headers)
+
+    def get(self, url, headers=None):
+        return self.request('GET', url, headers=headers)
+
+    def post(self, url, data, headers=None):
+        return self.request('POST', url, data=data, headers=headers)
+
+    def put(self, url, data, headers=None):
+        return self.request('PUT', url, data=data, headers=headers)
+
+    def delete(self, url, headers=None):
+        return self.request('DELETE', url, headers=headers)
+
+
+class GenericToken(object):
+    """Represents an Authorization token to be added to HTTP requests.
+
+    Some Authorization headers included calculated fields (digital
+    signatures for example) which are based on the parameters of the HTTP
+    request. Therefore the token is responsible for signing the request
+    and adding the Authorization header.
+    """
+
+    def perform_request(self, http_client, operation, url, data=None,
+                        headers=None):
+        """For the GenericToken, no Authorization token is set."""
+        return http_client.request(operation, url, data=data, headers=headers)
+
+    def valid_for_scope(self, url):
+        """Tells the caller if the token authorizes access to the desired URL.
+
+        Since the generic token doesn't add an auth header, it is not valid for
+        any scope.
+        """
+        return False

gam/atom/mock_http.py

@@ -0,0 +1,123 @@
+#
+# Copyright (C) 2008 Google Inc.
+#
+# Licensed under the Apache License 2.0;
+
+
+
+# __author__ = 'api.jscudder (Jeff Scudder)'
+
+import atom.http_interface
+import atom.url
+
+
+class Error(Exception):
+    pass
+
+
+class NoRecordingFound(Error):
+    pass
+
+
+class MockRequest(object):
+    """Holds parameters of an HTTP request for matching against future requests.
+    """
+
+    def __init__(self, operation, url, data=None, headers=None):
+        self.operation = operation
+        if isinstance(url, str):
+            url = atom.url.parse_url(url)
+        self.url = url
+        self.data = data
+        self.headers = headers
+
+
+class MockResponse(atom.http_interface.HttpResponse):
+    """Simulates an httplib.HTTPResponse object."""
+
+    def __init__(self, body=None, status=None, reason=None, headers=None):
+        if body and hasattr(body, 'read'):
+            self.body = body.read()
+        else:
+            self.body = body
+        if status is not None:
+            self.status = int(status)
+        else:
+            self.status = None
+        self.reason = reason
+        self._headers = headers or {}
+
+    def read(self):
+        return self.body
+
+
+class MockHttpClient(atom.http_interface.GenericHttpClient):
+    def __init__(self, headers=None, recordings=None, real_client=None):
+        """An HttpClient which responds to request with stored data.
+
+        The request-response pairs are stored as tuples in a member list named
+        recordings.
+
+        The MockHttpClient can be switched from replay mode to record mode by
+        setting the real_client member to an instance of an HttpClient which will
+        make real HTTP requests and store the server's response in list of
+        recordings.
+
+        Args:
+          headers: dict containing HTTP headers which should be included in all
+              HTTP requests.
+          recordings: The initial recordings to be used for responses. This list
+              contains tuples in the form: (MockRequest, MockResponse)
+          real_client: An HttpClient which will make a real HTTP request. The
+              response will be converted into a MockResponse and stored in
+              recordings.
+        """
+        self.recordings = recordings or []
+        self.real_client = real_client
+        self.headers = headers or {}
+
+    def add_response(self, response, operation, url, data=None, headers=None):
+        """Adds a request-response pair to the recordings list.
+
+        After the recording is added, future matching requests will receive the
+        response.
+
+        Args:
+          response: MockResponse
+          operation: str
+          url: str
+          data: str, Currently the data is ignored when looking for matching
+              requests.
+          headers: dict of strings: Currently the headers are ignored when
+              looking for matching requests.
+        """
+        request = MockRequest(operation, url, data=data, headers=headers)
+        self.recordings.append((request, response))
+
+    def request(self, operation, url, data=None, headers=None):
+        """Returns a matching MockResponse from the recordings.
+
+        If the real_client is set, the request will be passed along and the
+        server's response will be added to the recordings and also returned.
+
+        If there is no match, a NoRecordingFound error will be raised.
+        """
+        if self.real_client is None:
+            if isinstance(url, str):
+                url = atom.url.parse_url(url)
+            for recording in self.recordings:
+                if recording[0].operation == operation and recording[0].url == url:
+                    return recording[1]
+            raise NoRecordingFound('No recodings found for %s %s' % (
+                operation, url))
+        else:
+            # There is a real HTTP client, so make the request, and record the
+            # response.
+            response = self.real_client.request(operation, url, data=data,
+                                                headers=headers)
+            # TODO: copy the headers
+            stored_response = MockResponse(body=response, status=response.status,
+                                           reason=response.reason)
+            self.add_response(stored_response, operation, url, data=data,
+                              headers=headers)
+            return stored_response

gam/atom/mock_http_core.py

@@ -0,0 +1,313 @@
+#!/usr/bin/env python
+#
+# Copyright (C) 2009 Google Inc.
+#
+# Licensed under the Apache License 2.0;
+
+
+
+# This module is used for version 2 of the Google Data APIs.
+
+
+# __author__ = 'j.s@google.com (Jeff Scudder)'
+
+import io
+import os.path
+import pickle
+import tempfile
+
+import atom.http_core
+
+
+class Error(Exception):
+    pass
+
+
+class NoRecordingFound(Error):
+    pass
+
+
+class MockHttpClient(object):
+    debug = None
+    real_client = None
+    last_request_was_live = False
+
+    # The following members are used to construct the session cache temp file
+    # name.
+    # These are combined to form the file name
+    # /tmp/cache_prefix.cache_case_name.cache_test_name
+    cache_name_prefix = 'gdata_live_test'
+    cache_case_name = ''
+    cache_test_name = ''
+
+    def __init__(self, recordings=None, real_client=None):
+        self._recordings = recordings or []
+        if real_client is not None:
+            self.real_client = real_client
+
+    def add_response(self, http_request, status, reason, headers=None,
+                     body=None):
+        response = MockHttpResponse(status, reason, headers, body)
+        # TODO Scrub the request and the response.
+        self._recordings.append((http_request._copy(), response))
+
+    AddResponse = add_response
+
+    def request(self, http_request):
+        """Provide a recorded response, or record a response for replay.
+
+        If the real_client is set, the request will be made using the
+        real_client, and the response from the server will be recorded.
+        If the real_client is None (the default), this method will examine
+        the recordings and find the first which matches.
+        """
+        request = http_request._copy()
+        _scrub_request(request)
+        if self.real_client is None:
+            self.last_request_was_live = False
+            for recording in self._recordings:
+                if _match_request(recording[0], request):
+                    return recording[1]
+        else:
+            # Pass along the debug settings to the real client.
+            self.real_client.debug = self.debug
+            # Make an actual request since we can use the real HTTP client.
+            self.last_request_was_live = True
+            response = self.real_client.request(http_request)
+            scrubbed_response = _scrub_response(response)
+            self.add_response(request, scrubbed_response.status,
+                              scrubbed_response.reason,
+                              dict(atom.http_core.get_headers(scrubbed_response)),
+                              scrubbed_response.read())
+            # Return the recording which we just added.
+            return self._recordings[-1][1]
+        raise NoRecordingFound('No recoding was found for request: %s %s' % (
+            request.method, str(request.uri)))
+
+    Request = request
+
+    def _save_recordings(self, filename):
+        recording_file = open(os.path.join(tempfile.gettempdir(), filename),
+                              'wb')
+        pickle.dump(self._recordings, recording_file)
+        recording_file.close()
+
+    def _load_recordings(self, filename):
+        recording_file = open(os.path.join(tempfile.gettempdir(), filename),
+                              'rb')
+        self._recordings = pickle.load(recording_file)
+        recording_file.close()
+
+    def _delete_recordings(self, filename):
+        full_path = os.path.join(tempfile.gettempdir(), filename)
+        if os.path.exists(full_path):
+            os.remove(full_path)
+
+    def _load_or_use_client(self, filename, http_client):
+        if os.path.exists(os.path.join(tempfile.gettempdir(), filename)):
+            self._load_recordings(filename)
+        else:
+            self.real_client = http_client
+
+    def use_cached_session(self, name=None, real_http_client=None):
+        """Attempts to load recordings from a previous live request.
+
+        If a temp file with the recordings exists, then it is used to fulfill
+        requests. If the file does not exist, then a real client is used to
+        actually make the desired HTTP requests. Requests and responses are
+        recorded and will be written to the desired temprary cache file when
+        close_session is called.
+
+        Args:
+          name: str (optional) The file name of session file to be used. The file
+                is loaded from the temporary directory of this machine. If no name
+                is passed in, a default name will be constructed using the
+                cache_name_prefix, cache_case_name, and cache_test_name of this
+                object.
+          real_http_client: atom.http_core.HttpClient the real client to be used
+                            if the cached recordings are not found. If the default
+                            value is used, this will be an
+                            atom.http_core.HttpClient.
+        """
+        if real_http_client is None:
+            real_http_client = atom.http_core.HttpClient()
+        if name is None:
+            self._recordings_cache_name = self.get_cache_file_name()
+        else:
+            self._recordings_cache_name = name
+        self._load_or_use_client(self._recordings_cache_name, real_http_client)
+
+    def close_session(self):
+        """Saves recordings in the temporary file named in use_cached_session."""
+        if self.real_client is not None:
+            self._save_recordings(self._recordings_cache_name)
+
+    def delete_session(self, name=None):
+        """Removes recordings from a previous live request."""
+        if name is None:
+            self._delete_recordings(self._recordings_cache_name)
+        else:
+            self._delete_recordings(name)
+
+    def get_cache_file_name(self):
+        return '%s.%s.%s' % (self.cache_name_prefix, self.cache_case_name,
+                             self.cache_test_name)
+
+    def _dump(self):
+        """Provides debug information in a string."""
+        output = 'MockHttpClient\n  real_client: %s\n  cache file name: %s\n' % (
+            self.real_client, self.get_cache_file_name())
+        output += '  recordings:\n'
+        i = 0
+        for recording in self._recordings:
+            output += '    recording %i is for: %s %s\n' % (
+                i, recording[0].method, str(recording[0].uri))
+            i += 1
+        return output
+
+
+def _match_request(http_request, stored_request):
+    """Determines whether a request is similar enough to a stored request
+       to cause the stored response to be returned."""
+    # Check to see if the host names match.
+    if (http_request.uri.host is not None
+        and http_request.uri.host != stored_request.uri.host):
+        return False
+    # Check the request path in the URL (/feeds/private/full/x)
+    elif http_request.uri.path != stored_request.uri.path:
+        return False
+    # Check the method used in the request (GET, POST, etc.)
+    elif http_request.method != stored_request.method:
+        return False
+    # If there is a gsession ID in either request, make sure that it is matched
+    # exactly.
+    elif ('gsessionid' in http_request.uri.query
+          or 'gsessionid' in stored_request.uri.query):
+        if 'gsessionid' not in stored_request.uri.query:
+            return False
+        elif 'gsessionid' not in http_request.uri.query:
+            return False
+        elif (http_request.uri.query['gsessionid']
+                  != stored_request.uri.query['gsessionid']):
+            return False
+    # Ignores differences in the query params (?start-index=5&max-results=20),
+    # the body of the request, the port number, HTTP headers, just to name a
+    # few.
+    return True
+
+
+def _scrub_request(http_request):
+    """ Removes email address and password from a client login request.
+
+    Since the mock server saves the request and response in plantext, sensitive
+    information like the password should be removed before saving the
+    recordings. At the moment only requests sent to a ClientLogin url are
+    scrubbed.
+    """
+    if (http_request and http_request.uri and http_request.uri.path and
+            http_request.uri.path.endswith('ClientLogin')):
+        # Remove the email and password from a ClientLogin request.
+        http_request._body_parts = []
+        http_request.add_form_inputs(
+            {'form_data': 'client login request has been scrubbed'})
+    else:
+        # We can remove the body of the post from the recorded request, since
+        # the request body is not used when finding a matching recording.
+        http_request._body_parts = []
+    return http_request
+
+
+def _scrub_response(http_response):
+    return http_response
+
+
+class EchoHttpClient(object):
+    """Sends the request data back in the response.
+
+    Used to check the formatting of the request as it was sent. Always responds
+    with a 200 OK, and some information from the HTTP request is returned in
+    special Echo-X headers in the response. The following headers are added
+    in the response:
+    'Echo-Host': The host name and port number to which the HTTP connection is
+                 made. If no port was passed in, the header will contain
+                 host:None.
+    'Echo-Uri': The path portion of the URL being requested. /example?x=1&y=2
+    'Echo-Scheme': The beginning of the URL, usually 'http' or 'https'
+    'Echo-Method': The HTTP method being used, 'GET', 'POST', 'PUT', etc.
+    """
+
+    def request(self, http_request):
+        return self._http_request(http_request.uri, http_request.method,
+                                  http_request.headers, http_request._body_parts)
+
+    def _http_request(self, uri, method, headers=None, body_parts=None):
+        body = io.StringIO()
+        response = atom.http_core.HttpResponse(status=200, reason='OK', body=body)
+        if headers is None:
+            response._headers = {}
+        else:
+            # Copy headers from the request to the response but convert values to
+            # strings. Server response headers always come in as strings, so an int
+            # should be converted to a corresponding string when echoing.
+            for header, value in headers.items():
+                response._headers[header] = str(value)
+        response._headers['Echo-Host'] = '%s:%s' % (uri.host, str(uri.port))
+        response._headers['Echo-Uri'] = uri._get_relative_path()
+        response._headers['Echo-Scheme'] = uri.scheme
+        response._headers['Echo-Method'] = method
+        for part in body_parts:
+            if isinstance(part, str):
+                body.write(part)
+            elif hasattr(part, 'read'):
+                body.write(part.read())
+        body.seek(0)
+        return response
+
+
+class SettableHttpClient(object):
+    """An HTTP Client which responds with the data given in set_response."""
+
+    def __init__(self, status, reason, body, headers):
+        """Configures the response for the server.
+
+        See set_response for details on the arguments to the constructor.
+        """
+        self.set_response(status, reason, body, headers)
+        self.last_request = None
+
+    def set_response(self, status, reason, body, headers):
+        """Determines the response which will be sent for each request.
+
+        Args:
+          status: An int for the HTTP status code, example: 200, 404, etc.
+          reason: String for the HTTP reason, example: OK, NOT FOUND, etc.
+          body: The body of the HTTP response as a string or a file-like
+                object (something with a read method).
+          headers: dict of strings containing the HTTP headers in the response.
+        """
+        self.response = atom.http_core.HttpResponse(status=status, reason=reason,
+                                                    body=body)
+        self.response._headers = headers.copy()
+
+    def request(self, http_request):
+        self.last_request = http_request
+        return self.response
+
+
+class MockHttpResponse(atom.http_core.HttpResponse):
+    def __init__(self, status=None, reason=None, headers=None, body=None):
+        self._headers = headers or {}
+        if status is not None:
+            self.status = status
+        if reason is not None:
+            self.reason = reason
+        if body is not None:
+            # Instead of using a file-like object for the body, store as a string
+            # so that reads can be repeated.
+            if hasattr(body, 'read'):
+                self._body = body.read()
+            else:
+                self._body = body
+
+    def read(self):
+        return self._body