-# Multiple inheritance makes my inner Java nerd cry. Why can't I just
-# add arbitrary attributes to list or str objects?! Guido, we need to
-# talk.
-class TwitterJsonListResponse(TwitterResponse, list):
- __doc__ = """Twitter JSON Response
- """ + TwitterResponse.__doc__
- def __init__(self, lst, headers):
- TwitterResponse.__init__(self, headers)
- list.__init__(self, lst)
-class TwitterJsonDictResponse(TwitterResponse, dict):
- __doc__ = """Twitter JSON Response
- """ + TwitterResponse.__doc__
- def __init__(self, d, headers):
- TwitterResponse.__init__(self, headers)
- dict.__init__(self, d)
-
-class TwitterXmlResponse(TwitterResponse, str):
- __doc__ = """Twitter XML Response
- """ + TwitterResponse.__doc__
+def wrap_response(response, headers):
+ response_typ = type(response)
+ class WrappedTwitterResponse(TwitterResponse, response_typ):
+ __doc__ = TwitterResponse.__doc__
+
+ def __init__(self, response, headers):
+ response_typ.__init__(self, response)
+ TwitterResponse.__init__(self, headers)
+
+ return WrappedTwitterResponse(response, headers)
+