Skip to content

Commit 75859f5

Browse files
committed
feat: add reset param support
1 parent e550af2 commit 75859f5

File tree

2 files changed

+13
-6
lines changed

2 files changed

+13
-6
lines changed

src/api.py

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,17 +15,24 @@ class ApiInterface:
1515
__PREVIOUS_DEVICE_URL = __DEVICE_URL + "/previous"
1616
__PLAY_URL = __BASE_URL + "/play"
1717

18-
def request(self, method: str, url: str, data=None) -> urequests.Response:
18+
def request(self, method: str, url: str, data=None, params=None) -> urequests.Response:
1919
if data is None:
2020
data = {}
21+
if params is None:
22+
params = {}
23+
24+
query_string = '&'.join([f"{key}={value}" for key, value in params.items()])
25+
if query_string:
26+
url = f"{url}?{query_string}"
2127

2228
if method == "GET":
29+
if data:
30+
raise ValueError("GET requests cannot have data, use params instead")
31+
2332
resp = urequests.request(method, url, headers=self.__HEADERS)
2433
else:
2534
resp = urequests.request(method, url, headers=self.__HEADERS, data=json.dumps(data))
2635

27-
print(resp.status_code)
28-
print(resp.text)
2936
return resp
3037

3138
def check_health(self) -> bool:
@@ -34,8 +41,8 @@ def check_health(self) -> bool:
3441
except:
3542
return False
3643

37-
def get_current_device(self) -> dict:
38-
return self.request("GET", self.__DEVICE_URL).json()
44+
def get_current_device(self, reset: bool = False) -> dict:
45+
return self.request("GET", self.__DEVICE_URL, params={"reset": reset}).json()
3946

4047
def next_device(self) -> dict:
4148
return self.request("POST", self.__NEXT_DEVICE_URL).json()

src/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def __display_current_device(self) -> None:
7474
self.__display.clear()
7575
self.__display.print("Devices:", line=1)
7676

77-
current_device_data = self.__api.get_current_device()
77+
current_device_data = self.__api.get_current_device(reset=True)
7878
self.__change_device(current_device_data)
7979

8080
def __change_device(self, device_data: dict) -> None:

0 commit comments

Comments
 (0)