32 lines
703 B
Python
32 lines
703 B
Python
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)
|