Bug fixes

Based on the first day of dogfooding for myself
This commit is contained in:
2024-11-30 23:42:35 -08:00
parent 5db4e76de8
commit a067953467
7 changed files with 36 additions and 39 deletions

View File

@@ -18,21 +18,24 @@ class User:
def __eq__(self, other):
return self.user_id == other.user_id
def __hash__(self):
return hash(self.user_id)
@classmethod
def from_item(cls, primary_key: int, region: Region, subscribed_alerts: List[str]) -> 'User':
alerts = [pdb.Alert.from_str(alert_str) for alert_str in subscribed_alerts]
return cls(primary_key, region, alerts)
@property
def primary_key(self) -> int:
return self.user_id
def primary_key(self) -> str:
return str(self.user_id)
@property
def primary_key_name(self) -> str:
return 'user_id'
@property
def key(self) -> Dict[str, int]:
def key(self) -> Dict[str, str]:
return {
self.primary_key_name: self.primary_key
}