diff --git a/verify/generate.py b/verify/generate.py
index ad860359..06a10732 100644
--- a/verify/generate.py
+++ b/verify/generate.py
@@ -1,6 +1,5 @@
-from substrateinterface import Keypair
-from os import getenv, environ
from datetime import datetime
+
import bittensor
# Hardcode or set the environment variable WALLET_PASS to the password for the wallet
@@ -14,14 +13,15 @@ def main(args):
timestamp = datetime.now()
timezone = timestamp.astimezone().tzname()
- message = f"On {timestamp} {timezone} {args.message}"
+ # ensure compatiblity with polkadotjs messages, as polkadotjs always wraps message
+ message = "" + f"On {timestamp} {timezone} {args.message}" + ""
signature = keypair.sign(data=message)
file_contents = f"{message}\n\tSigned by: {keypair.ss58_address}\n\tSignature: {signature.hex()}"
print(file_contents)
open("message_and_signature.txt", "w").write(file_contents)
- print(f"Signature generated and saved to message_and_signature.txt")
+ print("Signature generated and saved to message_and_signature.txt")
if __name__ == "__main__":
diff --git a/verify/verify.py b/verify/verify.py
index 64516f08..9e954b72 100644
--- a/verify/verify.py
+++ b/verify/verify.py
@@ -1,6 +1,7 @@
-from substrateinterface import Keypair
from binascii import unhexlify
+from substrateinterface import Keypair
+
def main(args):
file_data = open(args.file).read()
@@ -16,6 +17,8 @@ def main(args):
keypair = Keypair(ss58_address=address, ss58_format=42)
message = file_split[0]
+ if not message.startswith("") or not message.endswith(""):
+ raise ValueError("Message is not properly wrapped in .")
signature_line = file_split[2]
signature_prefix = "Signature: "
@@ -36,8 +39,6 @@ def main(args):
import argparse
parser = argparse.ArgumentParser(description="Verify a signature")
- parser.add_argument(
- "--file", help="The file containing the message and signature"
- )
+ parser.add_argument("--file", help="The file containing the message and signature")
args = parser.parse_args()
main(args)