Skip to content

Commit

Permalink
python: osdp: Add method in KeyStore to persisting a key
Browse files Browse the repository at this point in the history
Signed-off-by: Siddharth Chandrasekaran <[email protected]>
  • Loading branch information
sidcha committed May 28, 2024
1 parent d70d994 commit 6e33133
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions python/osdp/key_store.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,10 +28,7 @@ def gen_key(key_len=16):
key.append(random.randint(0, 255))
return bytes(key)

def store_key(self, name):
if name not in self.keys:
raise RuntimeError
key = self.keys[name]
def _store_key(self, key):
with open(self.key_file(name), "w") as f:
f.write(key.hex())

Expand All @@ -46,6 +43,11 @@ def new_key(self, name, key_len=16, force=True):
self.keys[name] = self.gen_key(key_len)
return self.keys[name]

def commit_key(self, name):
if name not in self.keys:
raise RuntimeError
self._store_key(self.keys[name])

def update_key(self, name, key):
if name not in self.keys:
raise RuntimeError
Expand All @@ -63,4 +65,4 @@ def load_key(self, name, key_len=16):

def __del__(self):
if self.temp_dir:
self.temp_dir.cleanup()
self.temp_dir.cleanup()

0 comments on commit 6e33133

Please sign in to comment.