Spaces:
Running
Running
File size: 1,942 Bytes
cb7ff7d 88f9105 c1c7334 cb7ff7d d680b24 ebaa573 d680b24 ebaa573 d680b24 ebaa573 d680b24 ebaa573 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
from streamlit_local_storage import LocalStorage
from local_storage.entities import SavedDetails, PersonalDetails, LocationDetails, ContractorDetails
from enums.enums import DetailsType
class LocalStorageManager:
def __init__(self):
self.ls: LocalStorage = LocalStorage()
def get_detail(self, key: str = None):
for detail_type in DetailsType.values():
detail = self.ls.getItem(detail_type.name).get(key)
if detail:
if detail_type.value == DetailsType.PERSONAL_DETAILS.value:
return PersonalDetails(**detail)
elif detail_type.value == DetailsType.LOCATION_DETAILS.value:
return LocationDetails(**detail)
elif detail_type.value == DetailsType.CONTRACTOR_DETAILS.value:
return ContractorDetails(**detail)
return None
return None
def save_details(self, details: SavedDetails):
if isinstance(details, PersonalDetails):
type_ = DetailsType.PERSONAL_DETAILS
elif isinstance(details, LocationDetails):
type_ = DetailsType.LOCATION_DETAILS
elif isinstance(details, ContractorDetails):
type_ = DetailsType.CONTRACTOR_DETAILS
else:
raise ValueError("Unexpected type: {}", type(details))
existing_data = (self.ls.getItem(type_.name) or {})
existing_data[details.key] = details.to_json()
self.ls.setItem(type_.name, existing_data, type_.name)
def get_details(self, type_: DetailsType) -> list[SavedDetails]:
if type_.value == DetailsType.PERSONAL_DETAILS.value:
cls = PersonalDetails
elif type_.value == DetailsType.LOCATION_DETAILS.value:
cls = LocationDetails
else:
cls = ContractorDetails
details = self.ls.getItem(type_.name) or {}
return [cls(**v) for k,v in details.items()]
|