host.nfc_tool.battery_data

 1"""Battery data model for NFC tags.
 2
 3Data is stored on tags as a single NDEF Text record with an INI-style
 4section header ('bat:') followed by key=value lines.
 5"""
 6
 7from dataclasses import dataclass
 8
 9
10@dataclass
11class BatteryData:
12    """Parsed battery information from an NFC tag."""
13    sn: str = ''
14    year: str = ''
15    note: str = ''
16
17    @classmethod
18    def from_text(cls, text):
19        """Parse bat: section with key=value lines.
20
21        Tolerant of unknown keys and missing bat: prefix.
22        """
23        if 'bat:' in text:
24            text = text.split('bat:', 1)[1]
25        fields = {}
26        for line in text.strip().split('\n'):
27            if '=' in line:
28                k, v = line.split('=', 1)
29                fields[k.strip()] = v.strip()
30        return cls(
31            sn=fields.get('sn', ''),
32            year=fields.get('year', ''),
33            note=fields.get('note', ''),
34        )
35
36    def to_text(self):
37        """Serialize to bat: section format for NDEF Text record."""
38        lines = [f"bat:", f"sn={self.sn}", f"year={self.year}"]
39        if self.note:
40            lines.append(f"note={self.note}")
41        return '\n'.join(lines)
42
43    def is_valid(self):
44        """Battery data is valid if it has a serial number."""
45        return bool(self.sn.strip())
46
47    def is_battery_tag(self):
48        """Check if text was recognized as battery data."""
49        return bool(self.sn)