Added set_rain_collector

Author: MarcoGos

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "PyVantagePro_MarcoGos"
-version = "0.3.19"
+version = "0.3.20"
 authors = [
   { name="Salem Harrache", email="[email protected]" },
   { name="Marco Gosselink", email="[email protected]" },

pyvantagepro/__init__.py

@@ -9,9 +9,6 @@
     :license: GNU GPL v3.
 
 '''
-# Make sure the logger is configured early:
-from .logger import LOGGER, active_logger
-from .device import VantagePro2
 
-VERSION = '0.3.19'
+VERSION = '0.3.20'
 __version__ = VERSION

pyvantagepro/device.py

@@ -154,9 +154,10 @@ def read_from_eeprom(self, hex_address, size):
 
     def write_to_eeprom(self, hex_address, size, data):
         '''Writes to EEPROM the `size` number of bytes starting at the
-        `hex_address` with the given `data`. `data` needs to be bytes.'''
+        `hex_address` with the given `data`. `data` needs to be bytes. CRC will be automatically added.'''
         self.send("EEBWR %s %.2d\n" % (hex_address, size), self.ACK)
-        self.send(data)
+        data_with_crc = VantageProCRC(data).data_with_checksum
+        self.send(data_with_crc, self.ACK)
 
     def gettime(self):
         '''Returns the current datetime of the console.'''
@@ -307,6 +308,14 @@ def newsetup(self) -> None:
         self.wake_up()
         self.send("NEWSETUP", self.ACK)
 
+    def set_rain_collector(self, type) -> None:
+        '''Set rain collector type. 0x00 = 0.01", 0x10 = 0.2mm, 0x20 = 0.1mm'''
+        setup_bits = struct.unpack(b"B", self.read_from_eeprom("2B", 1))[0]  # type: ignore
+        setup_bits = (setup_bits & 0xCF) | type
+        data = struct.pack(b'>B', setup_bits)
+        self.write_to_eeprom("2B", 1, data)
+        self.newsetup()
+
     @cached_property
     def archive_period(self):
         '''Returns number of minutes in the archive period.'''