Skip to content

Commit

Permalink
feat(device): implement RefreshPollerMixin for Device class (#84)
Browse files Browse the repository at this point in the history
The Device class extends the RefreshPollerMixin class. We implemented the is_ready() method that enables the poll_till_ready method. This method can no be used to wait for a device until it is online.
  • Loading branch information
ankitrgadiya authored Oct 23, 2024
1 parent fd05053 commit f0ebeb3
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion rapyuta_io/clients/device.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
from rapyuta_io.utils.utils import create_auth_header, get_error, get_api_response_data, \
validate_key_value, response_validator, is_true
from rapyuta_io.utils.partials import PartialMixin
from rapyuta_io.utils.pollers import RefreshPollerMixin

DEVICE_API_ERRORS = {
400: ParameterMissingException,
Expand Down Expand Up @@ -217,7 +218,7 @@ def __str__(self):
PYTHON3 = '3'


class Device(PartialMixin, ObjDict):
class Device(PartialMixin, RefreshPollerMixin, ObjDict):
"""
Device class represents a device. Member variables of the class represent the
properties of device. \n
Expand Down Expand Up @@ -350,6 +351,9 @@ def is_online(self):
return True
return False

def is_ready(self):
return self.is_online()

def get_runtime(self):
"""
Get the device runtime
Expand Down

0 comments on commit f0ebeb3

Please sign in to comment.