Redo fix. Fix for actions structure to not error out when asking for next_id when the actions list is empty.

This commit is contained in:
ebolam
2022-03-05 10:31:28 -05:00
parent 5f92cbc231
commit 221f264fa7
2 changed files with 13 additions and 7 deletions

View File

@ -19,10 +19,16 @@ class KoboldStoryRegister(collections.OrderedDict):
return self.popitem()[1]
def get_first_key(self) -> int:
return next(iter(self))
if len(self) == 0:
return -1
else:
return next(iter(self))
def get_last_key(self) -> int:
return next(reversed(self))
if len(self) == 0:
return -1
else:
return next(reversed(self))
def __getitem__(self, k: int) -> str:
return super().__getitem__(k)