from shadow_drive import ShadowDriveClient
from solders.keypair import Keypair
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--keypair', metavar='keypair', type=str, required=True,
help='The keypair file to use (e.g. keypair.json, dev.json)')
args = parser.parse_args()
test_keypair_file = open(args.keypair, "r")
test_keypair_bytes = test_keypair_file.read()
keypair = Keypair(test_keypair_bytes)
print("Loaded keypair")
client = ShadowDriveClient(keypair)
print("Initialized client")
size = 2 ** 20
success = False
while not success:
try:
account, tx = client.create_account("test", size)
print("Created storage account")
success = True
except ValueError:
print("failed to create storage account")
files = ["./files/alpha.txt", "./files/not_alpha.txt"]
urls = client.upload_files(files)
print("Uploaded files")
client.delete_files(urls)
print("Deleted files")
client.close_account(account)
print("Closed account")