SwissFEL/stand
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor

Browse Source
This commit is contained in:
augustin_s
2022-05-19 22:55:59 +02:00
parent 391a232f2c
commit 3d929ad54c
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
client.py
View File

@ -10,22 +10,24 @@ class Client:
self.session = requests.Session() self.session = requests.Session()
def add_row(self, **kwargs): def add_row(self, **kwargs):
addr = f"http://{self.host}:{self.port}/" addr = self._make_addr()
resp = self.session.patch(addr, json=kwargs) resp = self.session.patch(addr, json=kwargs)
resp.raise_for_status()
return ResponseWrapper(resp) return ResponseWrapper(resp)
def get(self): def get(self):
addr = f"http://{self.host}:{self.port}/" addr = self._make_addr()
resp = self.session.get(addr) resp = self.session.get(addr)
resp.raise_for_status()
return ResponseWrapper(resp) return ResponseWrapper(resp)
def _make_addr(self):
return f"http://{self.host}:{self.port}/"
class ResponseWrapper: class ResponseWrapper:
def __init__(self, resp): def __init__(self, resp):
resp.raise_for_status()
self.resp = resp self.resp = resp
def __repr__(self): def __repr__(self):