def get_zone_state(self, zone, page=_DEFAULT_PAGE):
if _check_delay(self.zone_update_time, self.update_delay):
response = self._update_irrigation_state(page)
if not isinstance(response, States):
self.zones = States()
return None
else:
self.zone_update_time = time.time()
return self.zones.active(zone)
def __init__(
self,
server,
password,
update_delay=20,
retry=3,
retry_sleep=10,
logger=logging.getLogger(__name__),
):
self.rainbird_client = RainbirdClient(server, password, retry,
retry_sleep, logger)
self.logger = logger
self.zones = States()
self.rain_sensor = None
self.update_delay = update_delay
self.zone_update_time = None
self.sensor_update_time = None
def _update_irrigation_state(self, page=_DEFAULT_PAGE):
mask = ("%%0%dX" % RAIBIRD_COMMANDS["ControllerResponses"]["BF"]
["activeStations"]["length"])
result = self._process_command(
lambda resp: States((mask % resp["activeStations"])[:4]),
"CurrentStationsActive",
page,
)
if isinstance(result, States):
self.zones = result
return result
class RainbirdController:
def __init__(
self,
server,
password,
update_delay=20,
retry=3,
retry_sleep=10,
logger=logging.getLogger(__name__),
):
self.rainbird_client = RainbirdClient(server, password, retry,
retry_sleep, logger)
self.logger = logger
self.zones = States()
self.rain_sensor = None
self.update_delay = update_delay
self.zone_update_time = None
self.sensor_update_time = None
def get_model_and_version(self):
return self._process_command(
lambda response: ModelAndVersion(
response["modelID"],
response["protocolRevisionMajor"],
response["protocolRevisionMinor"],
),
"ModelAndVersion",
)
def get_available_stations(self, page=_DEFAULT_PAGE):
mask = ("%%0%dX" % RAIBIRD_COMMANDS["ControllerResponses"]["83"]
["setStations"]["length"])
return self._process_command(
lambda resp: AvailableStations(mask % resp["setStations"],
page=resp["pageNumber"]),
"AvailableStations",
page,
)
def get_command_support(self, command):
return self._process_command(
lambda resp: CommandSupport(resp["support"],
echo=resp["commandEcho"]),
"CommandSupport",
command,
)
def get_serial_number(self):
return self._process_command(lambda resp: resp["serialNumber"],
"SerialNumber")
def get_current_time(self):
return self._process_command(
lambda resp: datetime.time(resp["hour"], resp["minute"], resp[
"second"]),
"CurrentTime",
)
def get_current_date(self):
return self._process_command(
lambda resp: datetime.date(resp["year"], resp["month"], resp["day"]
),
"CurrentDate",
)
def water_budget(self, budget):
return self._process_command(
lambda resp: WaterBudget(resp["programCode"], resp["seasonalAdjust"
]),
"WaterBudget",
budget,
)
def get_rain_sensor_state(self):
if _check_delay(self.sensor_update_time, self.update_delay):
self.logger.debug("Requesting current Rain Sensor State")
response = self._process_command(
lambda resp: bool(resp["sensorState"]),
"CurrentRainSensorState",
)
if isinstance(response, bool):
self.rain_sensor = response
self.sensor_update_time = time.time()
else:
self.rain_sensor = None
return self.rain_sensor
def get_zone_state(self, zone, page=_DEFAULT_PAGE):
if _check_delay(self.zone_update_time, self.update_delay):
response = self._update_irrigation_state(page)
if not isinstance(response, States):
self.zones = States()
return None
else:
self.zone_update_time = time.time()
return self.zones.active(zone)
def set_program(self, program):
return self._process_command(lambda resp: True, "ManuallyRunProgram",
program)
def test_zone(self, zone):
return self._process_command(lambda resp: True, "TestStations", zone)
def irrigate_zone(self, zone, minutes):
response = self._process_command(lambda resp: True,
"ManuallyRunStation", zone, minutes)
self._update_irrigation_state()
return response == True and self.zones.active(zone)
def stop_irrigation(self):
response = self._process_command(lambda resp: True, "StopIrrigation")
self._update_irrigation_state()
return response == True and not reduce(
(lambda x, y: x or y), self.zones.states)
def get_rain_delay(self):
return self._process_command(lambda resp: resp["delaySetting"],
"RainDelayGet")
def set_rain_delay(self, days):
return self._process_command(lambda resp: True, "RainDelaySet", days)
def advance_zone(self, param):
return self._process_command(lambda resp: True, "AdvanceStation",
param)
def get_current_irrigation(self):
return self._process_command(
lambda resp: bool(resp["irrigationState"]),
"CurrentIrrigationState",
)
def command(self, command, *args):
data = rainbird.encode(command, *args)
self.logger.debug("Request to line: " + str(data))
decrypted_data = self.rainbird_client.request(
data,
RAIBIRD_COMMANDS["ControllerCommands"]["%sRequest" %
command]["length"],
)
self.logger.debug("Response from line: " + str(decrypted_data))
if decrypted_data is None:
self.logger.warn("Empty response from controller")
return None
decoded = rainbird.decode(decrypted_data)
if (decrypted_data[:2] !=
RAIBIRD_COMMANDS["ControllerCommands"]["%sRequest" %
command]["response"]):
raise Exception(
"Status request failed with wrong response! Requested %s but got %s:\n%s"
% (
RAIBIRD_COMMANDS["ControllerCommands"][
"%sRequest" % command]["response"],
decrypted_data[:2],
decoded,
))
self.logger.debug("Response: %s" % decoded)
return decoded
def _process_command(self, funct, cmd, *args):
response = self.command(cmd, *args)
return (funct(response) if response is not None
and response["type"] == RAIBIRD_COMMANDS["ControllerResponses"]
[RAIBIRD_COMMANDS["ControllerCommands"][cmd + "Request"]
["response"]]["type"] else response)
def _update_irrigation_state(self, page=_DEFAULT_PAGE):
mask = ("%%0%dX" % RAIBIRD_COMMANDS["ControllerResponses"]["BF"]
["activeStations"]["length"])
result = self._process_command(
lambda resp: States((mask % resp["activeStations"])[:4]),
"CurrentStationsActive",
page,
)
if isinstance(result, States):
self.zones = result
return result