def main_method(cmd_str=None):
'''
main method of notary.
Parameters
----------
cmd_str takes the command line input.
Returns
-------
'''
global notary
parser = argparse.ArgumentParser()
parser.add_argument("command", choices=[ 'register', 'confirm', 'notarize', 'login', 'notarystatus',
'uploadfile', 'downloadfile'],
help="Name of the command.")
parser.add_argument("-password", type=str, help="the password used to access the wallet.")
parser.add_argument("-email", type=str, help="the email address of the registered user.")
parser.add_argument("-file", type=str, help="Fully qualified name of the file.")
parser.add_argument("-metadata", type=str, help="File containing metadata of the file to notarize.")
parser.add_argument("-confirm_url", type=str, help="Confirmation URL to confirm an account.")
parser.add_argument("-document_hash", type=str, help="Document hash of a document")
if cmd_str is None:
args = parser.parse_args()
else:
args = parser.parse_args(cmd_str)
if not args.password:
print("Password is required!")
return
if notary is None:
notary = NotaryClient("./notaryconfig.ini", args.password)
command = args.command
print "Running " + command + " command"
if command == "register":
if not args.email:
print "register command needs email address"
else:
print args.email
result = notary.register_user(args.email)
print result
return result
elif command == "notarize":
if not args.metadata:
print "notarize command needs metadata file"
return
if not args.file:
print "notarize command needs file"
return
# print args.file
# print args.metadata
metadata = json.loads(args.metadata)
return notary.notarize_file(args.file,
metadata)
elif command == "uploadfile":
if not args.file:
print "upload command needs file"
return
# print args.file
return notary.upload_file(args.file)
elif command == "downloadfile":
if not args.document_hash:
print "download command needs document hash value"
return
if not args.file:
print "download command needs file"
return
# print args.file
return notary.download_file(args.document_hash,args.file)
elif command == "login":
return notary.authenticate()
elif command == "notarystatus":
if not args.document_hash:
print "confirm command needs document_hash"
else:
print args.document_hash
status = notary.get_notarization_status(args.document_hash)
print "The Document status is"
print status
return status
else:
print "no command"
import test_data
from notary_client import NotaryClient, NotaryException
try:
notary_client = NotaryClient('./notaryconfig.ini', 'foobar')
message = notary_client.register_user(test_data.email_address)
print(message)
except NotaryException as e:
print("Code %s " % e.error_code)
print(e.message)
try:
notary_obj = NotaryClient(test_data.config_file_name, "tessfsdft123")
except simplecrypt.DecryptionException as e:
print e.message
# test wallet exists are not.
client_wallet_obj = ClientWallet("somepassword")
print "wallet exists"
print client_wallet_obj.wallet_exists()
#test wallet is registered or not.
#test wallet is confirmed or not.
#test register to server.
print "registering wallet"
try:
print notary_obj.register_user(test_data.email_address)
except NotaryException as e:
print("Code %s " % e.error_code)
print(e.message)
print "try authentication without confirmation"
try:
print notary_obj.authenticate()
except NotaryException as e:
print("Code %s " % e.error_code)
print(e.message)
print "getting register status"
try:
print notary_obj.get_account()
except NotaryException as e:
print("Code %s " % e.error_code)
print(e.message)
