aljaz/python-nostr
1
0
Fork 0
mirror of https://github.com/aljazceru/python-nostr.git synced 2025-12-19 15:24:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

add error handling to mine_vanity_key function

Browse Source
This commit is contained in:
jeffthibault
2023-01-21 14:20:08 -05:00
parent 35c500cf14
commit 9d523c8942
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
nostr/key.py
View File

@@ -109,8 +109,10 @@ class PrivateKey:
return self.raw_secret == other.raw_secret return self.raw_secret == other.raw_secret
def mine_vanity_key(prefix: str = None, suffix: str = None) -> PrivateKey: def mine_vanity_key(prefix: str = None, suffix: str = None) -> PrivateKey:
if prefix is None and suffix is None:
raise ValueError("Expected at least one of 'prefix' or 'suffix' arguments")
while True: while True:
attempts += 1
sk = PrivateKey() sk = PrivateKey()
if prefix is not None and not sk.public_key.bech32()[5:5+len(prefix)] == prefix: if prefix is not None and not sk.public_key.bech32()[5:5+len(prefix)] == prefix:
continue continue