snagging the python function and client scripts

src/check_status.py

@@ -0,0 +1,26 @@
+import logging
+import boto3
+from transferrer.common import should_download_file, shoot_number_to_folder_path, get_source_bucket
+
+logger = logging.getLogger(__name__)
+
+
+def check_shoot_restore_status(bucket, shoot_number):
+    check_folder_restore_status(bucket, shoot_number_to_folder_path(shoot_number))
+
+
+def check_folder_restore_status(bucket, s3_folder: str):
+    logger.info(s3_folder)
+    for obj in bucket.objects.filter(Prefix=s3_folder,  OptionalObjectAttributes=[
+        'RestoreStatus',
+    ]):
+        if should_download_file(obj.key):
+            status = obj.restore_status
+        else:
+            status = "ignored"
+        logger.info(f"{obj.key}\t{status}")
+
+
+if __name__ == "__main__":
+    import sys
+    check_shoot_restore_status(get_source_bucket(boto3.Session()), sys.argv[1])

src/lambda_function.py

@@ -0,0 +1,35 @@
+import os
+import json
+import boto3
+from transferrer.transfer import transfer_shoot
+import logging
+logging.basicConfig(level=logging.INFO)
+
+logger = logging.getLogger()
+logger.setLevel(logging.INFO)
+
+ACCESSION_NUMBER = os.getenv("ACCESSION_NUMBER", "2754")
+
+
+def lambda_handler(event, context):
+    shoots = [single_handler(shoot_id) for shoot_id in get_shoot_ids(event)]
+    return {
+        'statusCode': 200,
+        'body': shoots
+    }
+
+
+def single_handler(shoot_id):
+    logger.info(f"transferring {shoot_id}")
+    transfer_shoot(
+        from_session=boto3.Session(),
+        to_session=boto3.Session(),
+        shoot_number=shoot_id,
+        accession_number=ACCESSION_NUMBER
+    )
+    return shoot_id
+
+
+def get_shoot_ids(event):
+    for record in event['Records']:
+        yield json.loads(record['body'])["Message"]

src/restore.py

@@ -0,0 +1,52 @@
+import logging
+import boto3
+from botocore.exceptions import ClientError
+
+from transferrer.common import should_download_file, shoot_number_to_folder_path, get_source_bucket
+
+logger = logging.getLogger(__name__)
+
+
+def restore_s3_folder(bucket, s3_folder: str, days_to_keep=1):
+    logger.info(f"restoring folder:  {s3_folder}")
+    for obj in bucket.objects.filter(Prefix=s3_folder):
+        if should_download_file(obj.key):
+            try:
+                logger.info(f"restoring object: {obj.key}")
+                obj.restore_object(
+                    RestoreRequest={
+                        'Days': days_to_keep,
+                        'GlacierJobParameters': {
+                            'Tier': 'Bulk'
+                        }
+                    }
+                )
+            except ClientError as e:
+                if "The operation is not valid for the object's storage class" in str(e):
+                    logger.info(f"attempt to restore non-glacier object: {obj.key}")
+                elif "RestoreAlreadyInProgress" in str(e):
+                    logger.info(f"redundant attempt to restore object: {obj.key}")
+                else:
+                    raise
+        else:
+            logger.info(f"ignoring {obj.key}")
+
+
+def restore_shoot_folder(bucket, shoot_number):
+    restore_s3_folder(bucket, shoot_number_to_folder_path(shoot_number))
+
+
+def restore_shoot_folders(shoot_numbers):
+    bucket = get_source_bucket(boto3.Session())
+    for shoot_number in (n.strip() for n in shoot_numbers):
+        logger.info(f"restoring shoot {shoot_number}")
+        restore_shoot_folder(bucket, shoot_number)
+
+
+def main():
+    import sys
+    restore_shoot_folders(sys.stdin.readlines())
+
+
+if __name__ == "__main__":
+    main()

src/start_transfers.py

@@ -0,0 +1,14 @@
+import sys
+import boto3
+
+
+def post_messages(session, environment, shoot_numbers):
+    sns = session.resource("sns")
+    topic = sns.Topic(f"arn:aws:sns:eu-west-1:404315009621:transfer-shoots-{environment}")
+    for shoot_number in shoot_numbers:
+        print(f"requesting transfer of {shoot_number}")
+        topic.publish(Message=shoot_number.strip())
+
+
+if __name__ == "__main__":
+    post_messages(boto3.Session(), sys.argv[1], sys.stdin.readlines())

src/touch.py

@@ -0,0 +1,42 @@
+"""
+Like unix touch, this updates objects in s3 without substantive change.
+
+Call with a newline-separated list of keys, thus
+
+echo 'born-digital-accessions/2754_CP000179.zip \n born-digital-accessions/2754_CP000181.zip' | touch.py
+
+This is intended for use when Archivematica has ephemerally failed to accept some of the shoot zips.
+
+A fortnightly list of failed zips is published to Slack, but you could also generate it by looking elsewhere.
+"""
+import sys
+
+import boto3
+import datetime
+
+BUCKETS = {
+    "staging": "wellcomecollection-archivematica-staging-transfer-source",
+    "production": "wellcomecollection-archivematica-transfer-source"
+}
+
+
+def touch_object(session, bucket, key):
+    print(f"touching: {bucket}/{key}")
+    session.client('s3').copy_object(
+        Bucket=bucket,
+        CopySource={'Bucket': bucket,  'Key': key},
+        Key=key,
+        Metadata={
+            'touched': datetime.datetime.now().isoformat()
+        },
+        MetadataDirective="REPLACE"
+    )
+
+
+def touch_objects(session, bucket,  object_keys):
+    for object_key in object_keys:
+        touch_object(session, bucket, object_key.strip())
+
+
+if __name__ == '__main__':
+    touch_objects(boto3.Session(), BUCKETS[sys.argv[1]],  sys.stdin.readlines())

src/transferrer/common.py

@@ -5,12 +5,11 @@
 SOURCE_BUCKET = "wellcomecollection-editorial-photography"
 
 
-def get_source_bucket(max_connections=10):
-    return get_source_client(max_connections).Bucket(SOURCE_BUCKET)
+def get_source_bucket(session: boto3.session.Session, max_connections=10):
+    return get_source_client(session, max_connections).Bucket(SOURCE_BUCKET)
 
 
-def get_source_client(max_connections):
-    session = boto3.Session()
+def get_source_client(session: boto3.session.Session, max_connections):
     return session.resource('s3', config=botocore.config.Config(
         max_pool_connections=max_connections
     ))

src/transferrer/download.py

@@ -2,6 +2,7 @@
 import logging
 from concurrent.futures import ThreadPoolExecutor
 from functools import partial
+import boto3
 
 from transferrer.common import should_download_file, shoot_number_to_folder_path, get_source_bucket
 
@@ -12,11 +13,11 @@
 THREADS = 10
 
 
-def download_shoot(shoot_number, local_dir):
+def download_shoot(session: boto3.session.Session, shoot_number, local_dir):
     # Allowing enough connections for each thread to have two of them
     # prevents the `urllib3.connectionpool:Connection pool is full` warning
     # and allows for better connection reuse.
-    download_shoot_folder(get_source_bucket(max_connections=THREADS * 2), shoot_number, local_dir)
+    download_shoot_folder(get_source_bucket(session, max_connections=THREADS * 2), shoot_number, local_dir)
 
 
 def download_shoot_folder(bucket, shoot_number, local_dir):
@@ -51,4 +52,4 @@ def download_s3_file(object_summary, *, local_dir: str, s3_folder: str):
     import sys
     shoot_number = sys.argv[1]
     download_folder = os.path.join("download", shoot_number)
-    download_shoot(sys.argv[1], download_folder)
+    download_shoot(boto3.Session(), sys.argv[1], download_folder)

src/transferrer/make_zip.py

@@ -56,12 +56,11 @@ def create_born_digital_folder(source_files, target_directory, accession_id):
 
 def move_files_to_objects_folder(source_files, target_directory):
     """
-    Move the contents of source directory into the objects subfolder in the target directory
-
+    Move the contents of source directory into the target directory
     """
-    os.makedirs(os.path.join(target_directory, "objects"))
+    os.makedirs(target_directory, exist_ok=True)
     for file_abspath in source_files:
-        shutil.move(file_abspath, os.path.join(target_directory, "objects"))
+        shutil.move(file_abspath, target_directory)
 
 
 def files_in_folder(source_directory):
@@ -103,21 +102,22 @@ def create_born_digital_zips(
             accession_id = f"{shoot_number_to_accession_id(accession_number, shoot_number)}_{ix:03d}"
             yield create_born_digital_zip(
                 full_paths(source_directory, batch),
-                os.path.join(target_directory, accession_id),
+                target_directory,
                 accession_id,
             )
     else:
         accession_id = shoot_number_to_accession_id(accession_number, shoot_number)
         yield create_born_digital_zip(
             full_paths(source_directory, filenames),
-            os.path.join(target_directory, accession_id),
+            target_directory,
             accession_id,
         )
 
 
-def create_born_digital_zip(source_files, target_directory, accession_id):
-    folder = create_born_digital_folder(source_files, target_directory, accession_id)
-    return shutil.make_archive(accession_id, "zip", folder)
+def create_born_digital_zip(source_files, target_directory: str, accession_id: str):
+    accession_path = os.path.join(target_directory, accession_id)
+    folder = create_born_digital_folder(source_files, accession_path, accession_id)
+    return shutil.make_archive(accession_path, "zip", folder)
 
 
 # copied from itertools 3.12 documentation.

src/transferrer/transfer.py

@@ -0,0 +1,25 @@
+import os
+import tempfile
+from transferrer.download import download_shoot
+from transferrer.make_zip import create_born_digital_zips
+from transferrer.upload import upload
+import boto3
+
+
+def transfer_shoot(from_session, to_session, shoot_number, accession_number):
+    with tempfile.TemporaryDirectory() as tmpfolder:
+        source_folder = os.path.join(tmpfolder, "source")
+        target_folder = os.path.join(tmpfolder, "target")
+        download_shoot(from_session, shoot_number, source_folder)
+        for zipfile in create_born_digital_zips(source_folder, target_folder, accession_number, shoot_number, 250):
+            upload(to_session, zipfile)
+
+
+if __name__ == "__main__":
+    import sys
+    transfer_shoot(
+        from_session=boto3.Session(profile_name=os.environ["AWS_SOURCE_PROFILE"]),
+        to_session=boto3.Session(profile_name=os.environ["AWS_TARGET_PROFILE"]),
+        shoot_number=sys.argv[1],
+        accession_number="2754"
+    )

src/transferrer/upload.py

@@ -7,15 +7,17 @@
 
 logger = logging.getLogger(__name__)
 
+TARGET_BUCKET = os.getenv("TARGET_BUCKET", "wellcomecollection-archivematica-staging-transfer-source")
 
-def upload(s3, zipfile_path, target_bucket_name="wellcomecollection-archivematica-staging-transfer-source"):
+
+def upload(session, zipfile_path, target_bucket_name=TARGET_BUCKET):
     logger.info(f"uploading {zipfile_path} to {target_bucket_name}")
-    get_target_bucket(s3, target_bucket_name).upload_file(zipfile_path, f"born-digital-accessions/{os.path.basename(zipfile_path)}")
+    get_target_bucket(session, target_bucket_name).upload_file(zipfile_path, f"born-digital-accessions/{os.path.basename(zipfile_path)}")
 
 
-def get_target_bucket(s3, target_bucket):
-    return s3.Bucket(target_bucket)
+def get_target_bucket(session, target_bucket):
+    return session.resource('s3').Bucket(target_bucket)
 
 
 if __name__ == "__main__":
-    upload( boto3.Session(profile_name=os.environ["AWS_TARGET_PROFILE"]).resource('s3'), sys.argv[1])
+    upload(boto3.Session(profile_name=os.environ["AWS_TARGET_PROFILE"]), sys.argv[1])