| 22 | |
| 23 | |
| 24 | class RESTResponse(io.IOBase): |
| 25 | |
| 26 | def __init__(self, resp) -> None: |
| 27 | self.response = resp |
| 28 | self.status = resp.status |
| 29 | self.reason = resp.reason |
| 30 | self.data = None |
| 31 | |
| 32 | def read(self): |
| 33 | if self.data is None: |
| 34 | self.data = self.response.data |
| 35 | return self.data |
| 36 | |
| 37 | def getheaders(self): |
| 38 | """Returns a dictionary of the response headers.""" |
| 39 | return self.response.headers |
| 40 | |
| 41 | def getheader(self, name, default=None): |
| 42 | """Returns a given response header.""" |
| 43 | return self.response.headers.get(name, default) |
| 44 | |
| 45 | |
| 46 | class RESTClientObject: |