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.upload_file(test_data.notary_file_name)
print(message)
except NotaryException as e:
print("Code %s " % e.error_code)
print(e.message)
print "notarizing file"
try:
print notary_obj.notarize_file(test_data.notary_file_name,test_data.getMetaData())
except NotaryException as e:
print("Code %s " % e.error_code)
print(e.message)
#test document status.
print "notarize again to see behaviour"
try:
print notary_obj.notarize_file(test_data.notary_file_name,test_data.getMetaData())
except NotaryException as e:
print("Code %s " % e.error_code)
print(e.message)
print "upload file"
try:
print notary_obj.upload_file(test_data.notary_file_name)
except NotaryException as e:
print("Code %s " % e.error_code)
print(e.message)
print "notary_status"
try:
print notary_obj.get_notarization_status(test_data.document_hash)
except NotaryException as e:
print("Code %s " % e.error_code)
print(e.message)
print "download file"
try:
print notary_obj.download_file(test_data.document_hash,test_data.storing_file_name)
except NotaryException as e:
print("Code %s " % e.error_code)
print(e.message)
