from dataclasses import dataclass

from typing import Optional

from uuid import uuid4 as uuid

COLLECTION_NAME = 'things'

@dataclass

class BaseThingsData:

'''Base attributes of Things type'''

pass

@dataclass

class ThingsData(BaseThingsData):

'''Things data'''

id: Optional[str] = None

@dataclass

class ThingsItemKeys:

'''Things DDB item keys'''

pk: str

sk: str

def get_data(self):

return { k:v for (k, v) in self.__dict__.items() if k not in ThingsItemKeys.__dict__.keys() }

@dataclass

class ThingsItem(ThingsData, ThingsItemKeys):

'''Things DDB item'''

id: str

def get_data(self):

return { k:v for (k, v) in self.__dict__.items() if k not in ['sk', 'pk'] }

def create_keys() -> ThingsItemKeys:

'''Create keys for DDB'''

key = '{}#{}'.format(COLLECTION_NAME, str(uuid()))

return ThingsItemKeys(**{'pk': key, 'sk': key})

def get_keys_from_id(_id: str) -> ThingsItemKeys:

'''Get keys for DDB'''

key = '{}#{}'.format(COLLECTION_NAME, _id)

return ThingsItemKeys(**{'pk': key, 'sk': key})

def get_id_from_keys(keys: ThingsItemKeys) -> str:

'''Get id from keys'''

return keys.pk.split('#')[1]