mimetype?

1 files changed, 4 insertions, 1 deletions

diff --git a/cli/app/utils/cortex_utils.py b/cli/app/utils/cortex_utils.py
index 0de9596..4aa4533 100644
--- a/cli/app/utils/cortex_utils.py
+++ b/cli/app/utils/cortex_utils.py
@@ -2,6 +2,7 @@ import os
 from os.path import join
 import requests
 import urllib3
+import mimetypes
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
 from app.settings import app_cfg
@@ -96,5 +97,7 @@ def upload_bytes_to_cortex(opt_folder_id, fn, fp, mimetype, **kwargs):
 
 def upload_file_to_cortex(opt_folder_id, fn, **kwargs):
   """Upload a file from disk"""
+  base_fn = os.path.basename(fn)
+  mimetype = mimetypes.guess_type(base_fn)
   with open(fn, 'rb') as fp:
-    return upload_fp_to_cortex(opt_folder_id, fp, **kwargs)
+    return upload_fp_to_cortex(opt_folder_id, (base_fn, fp, mimetype,), **kwargs)