def detected_new_file(new_file_fp, recipients):
"""Original model function for handling created files."""
logging.debug('Entered detected_new_file.')
gpg = gnupg.GPG()
# Write out temporary GPG file to temp directory.
with tempfile.NamedTemporaryFile(delete=False) as enc_tmp_file:
edata = gpg.encrypt_file(new_file_fp,
recipients,
always_trust=True,
armor=False,
output=enc_tmp_file.name)
enc_file_name = enc_tmp_file.name
logging.info('Finished encrypting the original file.')
# Get the SHA1 of the encrypted file.
file_sha1 = _sha1_of_file(enc_file_name)
# Filepath
filepath = new_file_fp.name # '/home/username/Lockbox/this/is/my/file.path'
oldfilepath = new_file_fp.name
for basepath in FLAGS.basepath:
if filepath.startswith(basepath):
filepath = re.sub(basepath, '', filepath)
if oldfilepath == filepath:
logging.error('Basepath not found for %s.' % filepath)
else:
logging.debug('Detected basepath: %s.' % basepath)
logging.debug('Shortened filepath: %s.' % filepath)
# Take SHA1 of path relative to the basepath.
sha_filepath = _sha1_of_string(filepath)
# Encrypt the exact value of the basepath.
enc_filepath = gpg.encrypt(
filepath, recipients, always_trust=True, armor=False)
sha_enc_fp = _sha1_of_string(enc_filepath.data)
logging.info("Upload to S3: SHA1(PGP(filepath)): '%s' data: '%s'" % \
(sha_enc_fp, enc_filepath.data))
# Scaffolding for testing.
sdb_conn = connect_sdb()
data_domain = get_domain(sdb_conn, 'group1')
lock_domain = get_domain(sdb_conn, 'group1_locks')
success, lock = acquire_domain_object_lock(lock_domain, sha_filepath)
if not success:
logging.error("Houston, we didn't get a lock for the object.")
# Meat and potatoes.
add_path(data_domain, sha_filepath, sha_enc_fp) # enc_filepath)
add_object_delta(data_domain, sha_filepath, file_sha1)
# Scaffolding for testing.
release_domain_object_lock(lock_domain, lock)
_print_all_domain(data_domain)
sdb_conn.delete_domain('group1')
sdb_conn.delete_domain('group1_locks')
return enc_file_name
def print_domain(domain_name):
connection = connect_sdb()
if not connection.lookup(domain_name, validate=True):
return
domain = get_domain(connection, domain_name)
_print_all_domain(domain)
