diff --git a/client.py b/client.py index 34ed950..8eb7715 100644 --- a/client.py +++ b/client.py @@ -10,22 +10,24 @@ class Client: self.session = requests.Session() def add_row(self, **kwargs): - addr = f"http://{self.host}:{self.port}/" + addr = self._make_addr() resp = self.session.patch(addr, json=kwargs) - resp.raise_for_status() return ResponseWrapper(resp) def get(self): - addr = f"http://{self.host}:{self.port}/" + addr = self._make_addr() resp = self.session.get(addr) - resp.raise_for_status() return ResponseWrapper(resp) + def _make_addr(self): + return f"http://{self.host}:{self.port}/" + class ResponseWrapper: def __init__(self, resp): + resp.raise_for_status() self.resp = resp def __repr__(self):