Move timestamp to base message class

Author: brefra

plugwise_usb/messages/__init__.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from enum import Enum
+from datetime import UTC, datetime
 from typing import Any
 
 from ..constants import MESSAGE_FOOTER, MESSAGE_HEADER, UTF8
@@ -28,6 +29,7 @@ def __init__(self, identifier: bytes) -> None:
         self._checksum: bytes | None = None
         self._args: list[Any] = []
         self._seq_id: bytes | None = None
+        self.timestamp = datetime.now(UTC)
 
     @property
     def seq_id(self) -> bytes | None:

plugwise_usb/messages/requests.py

@@ -4,7 +4,7 @@
 from asyncio import Future, TimerHandle, get_running_loop
 from collections.abc import Callable
 from copy import copy
-from datetime import UTC, datetime
+from datetime import datetime
 import logging
 
 from ..constants import (
@@ -50,7 +50,6 @@ def __init__(
         self._mac = mac
         self._send_counter: int = 0
         self._max_retries: int = MAX_RETRIES
-        self.timestamp = datetime.now(UTC)
         self._loop = get_running_loop()
         self._reply_identifier: bytes = b"0000"
         self._response: PlugwiseResponse | None = None

plugwise_usb/messages/responses.py

@@ -96,8 +96,6 @@ class NodeAwakeResponseType(int, Enum):
 class PlugwiseResponse(PlugwiseMessage):
     """Base class for response messages received by USB-Stick."""
 
-    timestamp: datetime | None = None
-
     def __init__(
         self,
         identifier: bytes,
@@ -139,7 +137,6 @@ def seq_id(self) -> bytes:
 
     def deserialize(self, response: bytes, has_footer: bool = True) -> None:
         """Deserialize bytes to actual message properties."""
-        self.timestamp = datetime.now(UTC)
         # Header
         if response[:4] != MESSAGE_HEADER:
             raise MessageError(