from dataclasses import dataclass @dataclass class Device: access_key: str device_type: str name: str def to_dict(self): return { 'access_key': self.access_key, 'type': self.device_type, 'name': self.name } def convert(device_from_db) -> Device: return Device(*device_from_db) @dataclass class SimCard: phone_number: str device_access_key: str def to_dict(self): return { 'phone_number': self.phone_number, 'device_access_key': self.device_access_key } def convert(sim_from_db) -> SimCard: return SimCard(*sim_from_db)