setting ContentType with python magic

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "s3-loader"
-version = "1.1.5"
+version = "1.1.6"
 description = "wrapper script for uploading files and dirs to s3"
 authors = ["phil-bell <[email protected]>"]
 license = "GNU General Public License"
@@ -13,6 +13,7 @@ readme = "README.md"
 python = "^3.6.12"
 boto3 = "^1.16.63"
 typer = "^0.3.2"
+python-magic = "^0.4.20"
 
 [tool.poetry.dev-dependencies]
 black = {version = "^20.8b1", allow-prereleases = true}

s3_loader/main.py

@@ -1,6 +1,7 @@
 import os
 import typer
 import boto3
+import magic
 import pickle
 from botocore.exceptions import NoCredentialsError
 
@@ -36,7 +37,12 @@ def upload_file(
             aws_secret_access_key=creds["secret"],
         )
         try:
-            s3.upload_file(local_file, bucket, s3_file)
+            s3.upload_file(
+                Filename=local_file,
+                Bucket=bucket,
+                Key=s3_file,
+                ExtraArgs={"ContentType": magic.from_file(local_file, mime=True)},
+            )
             print(f"File uploaded: {s3_file}")
         except FileNotFoundError:
             print("ERROR: file not found")