get file mime-type for S3 upload

stages/publish/s3_upload.py

@@ -5,6 +5,7 @@ import argparse
 import datetime
 from botocore.exceptions import ClientError
 import logging
+import mimetypes
 
 
 def upload_file(file_name, bucket, object_name=None):
@@ -19,8 +20,15 @@ def upload_file(file_name, bucket, object_name=None):
     access_key = os.environ["S3_ACCESS_KEY"]
     secret_key = os.environ["S3_SECRET_KEY"]
 
+    filetype = mimetypes.guess_type(file_name)
+
+    if not filetype[0]:
+        mimetype = "application/octet-stream"
+    else:
+        mimetype = filetype[0]
+
     # TODO: Add signature
-    extra_args = {"ContentType": "application/octet-stream", "ACL": "private"}
+    extra_args = {"ContentType": mimetype, "ACL": "private"}
 
     # If S3 object_name was not specified, use file_name
     if object_name is None: