Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 22 additions & 22 deletions pymodbus/datastore/store.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,10 @@
based on their preference.
"""
# pylint: disable=missing-type-doc
from __future__ import annotations

from abc import ABC, abstractmethod
from typing import Iterable
from typing import Any, Iterable

from pymodbus.exceptions import ParameterException

Expand All @@ -66,25 +68,12 @@ class BaseModbusDataBlock(ABC):
validate(self, address, count=1)
getValues(self, address, count=1)
setValues(self, address, values)
reset(self)
"""

def default(self, count, value=False):
"""Use to initialize a store to one value.

:param count: The number of fields to set
:param value: The default value to set to the fields
"""
self.default_value = value # pylint: disable=attribute-defined-outside-init
self.values = [ # pylint: disable=attribute-defined-outside-init
self.default_value
] * count
self.address = 0x00 # pylint: disable=attribute-defined-outside-init

def reset(self):
"""Reset the datastore to the initialized default value."""
self.values = [ # pylint: disable=attribute-defined-outside-init
self.default_value
] * len(self.values)
values: dict | list
address: int
default_value: Any

@abstractmethod
def validate(self, address:int, count=1) -> bool:
Expand Down Expand Up @@ -156,6 +145,20 @@ def create(cls):
"""
return cls(0x00, [0x00] * 65536)

def default(self, count, value=False):
"""Use to initialize a store to one value.

:param count: The number of fields to set
:param value: The default value to set to the fields
"""
self.default_value = value
self.values = [self.default_value] * count
self.address = 0x00

def reset(self):
"""Reset the datastore to the initialized default value."""
self.values = [self.default_value] * len(self.values)

def validate(self, address, count=1):
"""Check to see if the request is in range.

Expand Down Expand Up @@ -224,11 +227,10 @@ def __init__(self, values=None, mutable=True):
If values is a integer, then the value is set for the corresponding offset.

"""
self.values = {}
self.values: dict[int, Any] = {}
self._process_values(values)
self.mutable = mutable
self.default_value = self.values.copy()
self.address = next(iter(self.values.keys()), None)

@classmethod
def create(cls, values=None):
Expand Down Expand Up @@ -310,8 +312,6 @@ def setValues(self, address, values, use_as_default=False):
if address + idx not in self.values and not self.mutable:
raise ParameterException("Offset {address+idx} not in range")
self.values[address + idx] = val
if not self.address:
self.address = next(iter(self.values.keys()), None)
if use_as_default:
for idx, val in iter(self.values.items()):
self.default_value[idx] = val