def get_from(user_config, user_name):
url = user_config['server_url'] + '/{}/retrieve/'.format(user_name)
r = requests.get(url)
req_obj = r.json()
if req_obj['status'] == 'SUCCESS':
gpg = gnupg.GPG(gnupghome=user_config['gnupghome'])
remote_user = keys.get_key_for_user(gpg, user_config, user_name)
encrypted_file_data = base64.b64decode(req_obj["data"])
decrypted_data = gpg.decrypt(encrypted_file_data)
# verify that data was signed
if decrypted_data.signature_id is not None:
print('Signature verified with ',decrypted_data.trust_text)
file_name = req_obj["file_name"]
with open(file_name,"wb") as fp:
fp.write(decrypted_data.data)
print('Successfully retrieved file: ', file_name)
else:
print('Signature verification failed.')
else:
print('Failed: {}'.format(req_obj['error_message']))
def put_for(user_config, user_name, file_name):
p = re.compile('^(.+) <.*$')
gpg = gnupg.GPG(gnupghome=user_config['gnupghome'])
url = user_config['server_url'] + '/{}/store/'.format(user_config['user_name'])
remote_user_key = keys.get_key_for_user(gpg, user_config, user_name)
with open(file_name, "rb") as fp:
bts = fp.read()
encrypted_data = gpg.encrypt(bts,
remote_user_key['fingerprint'],
sign=user_config['fingerprint'],
armor=False,
#DO NOT F*****G LEAVE THIS HERE
always_trust=True
)
b64_bytes = base64.b64encode(encrypted_data.data).decode('utf-8')
payload = {
"file_name": os.path.basename(file_name),
"file_data": b64_bytes,
"file_target_user": user_name
}
headers = {'content-type': 'application/json'}
requests.post(url,data=json.dumps(payload), headers=headers)