Merge pull request #246 from RockefellerArchiveCenter/development

Invalidates Cloudfront cache after build

update.py

@@ -66,6 +66,7 @@ def run(self, audience, branch, deploy=True):
         site.build()
         if deploy:
             site.upload(audience, branch)
+            site.invalidate_cache(audience, branch)
         else:
             logging.info(
                 f'Skipping upload to S3 for {audience} {branch} site.')
@@ -153,6 +154,27 @@ def upload(self, audience, branch):
                             '').lstrip('/'),
                         f),
                     ExtraArgs={'ContentType': mtype if mtype else 'application/json'})
+
+    def invalidate_cache(self, audience, branch):
+        distribution_id = decrypt_env_variable(
+            f'{branch.upper()}_{audience.upper()}_DISTRIBUTION')
+        cloudfront = boto3.client(
+            'cloudfront',
+            region_name=decrypt_env_variable('REGION_NAME'),
+            aws_access_key_id=decrypt_env_variable('ACCESS_KEY'),
+            aws_secret_access_key=decrypt_env_variable('SECRET_KEY'))
+        cloudfront.create_invalidation(
+            DistributionId=distribution_id,
+            InvalidationBatch={
+                'Paths': {
+                    'Quantity': 1,
+                    'Items': [
+                        '/*',
+                    ]
+                },
+                'CallerReference': datetime.now().timestamp().replace('.', '')
+            }
+        )
 
 
 def main(event=None, context=None):