PyQuantimClient 2.0.12__tar.gz → 2.0.14__tar.gz

{pyquantimclient-2.0.12 → pyquantimclient-2.0.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyQuantimClient
-Version: 2.0.12
+Version: 2.0.14
 Summary: Python client to access quantIM services
 Author: Daniel Velasquez
 Author-email: daniel.velasquez@sura-im.com

{pyquantimclient-2.0.12 → pyquantimclient-2.0.14}/PyQuantimClient.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyQuantimClient
-Version: 2.0.12
+Version: 2.0.14
 Summary: Python client to access quantIM services
 Author: Daniel Velasquez
 Author-email: daniel.velasquez@sura-im.com

{pyquantimclient-2.0.12 → pyquantimclient-2.0.14}/setup.py

@@ -4,7 +4,7 @@ setup(
    name='PyQuantimClient',
    packages=['PyQuantimClient'],
    package_dir={'PyQuantimClient': 'src'},
-   version='2.0.12',
+   version='2.0.14',
    description='Python client to access quantIM services',
    author='Daniel Velasquez',
    author_email='daniel.velasquez@sura-im.com',

{pyquantimclient-2.0.12 → pyquantimclient-2.0.14}/src/api.py

@@ -1,7 +1,17 @@
 # -*- coding: utf-8 -*-
+import requests, json, warnings, mimetypes
 import pandas as pd
 import numpy as np
-import requests, json
+
+warnings.filterwarnings('ignore')
+
+mimetypes.add_type('application/vnd.openxmlformats-officedocument.spreadsheetml.sheet','.xlsx')
+mimetypes.add_type('application/vnd.ms-excel', '.xls')
+mimetypes.add_type('application/vnd.ms-excel.sheet.macroEnabled.12', '.xlsm')
+mimetypes.add_type('text/csv','.csv')
+mimetypes.add_type('application/vnd.openxmlformats-officedocument.wordprocessingml.document','.docx')
+mimetypes.add_type('application/msword', '.doc')
+mimetypes.add_type('application/pdf', '.pdf')
 
 class quantim:
     def __init__(self, username, password, secretpool, env="pdn", api_url=None):
@@ -105,9 +115,12 @@ class quantim:
             print(f"Error with presigned url: {e}")
             return False
 
+        content_type, _ = mimetypes.guess_type(local_file_path)
+        put_headers = {'Content-Type':content_type}
+
         with open(local_file_path, 'rb') as file:
             try:
-                response = requests.put(presigned_url, data=file, verify=False)
+                response = requests.put(presigned_url, data=file, headers=put_headers, verify=False)
                 if response.status_code == 200:
                     print("File uploaded successfully")
                     return True
@@ -117,3 +130,18 @@ class quantim:
             except Exception as e:
                 print(f"Error uploading file: {e}")
                 return False
+
+    def send_email(self, to, subject=None, content=None, vfrom="quantim@surainvestments.com", toname=None, fromname=None, cc=None, ccname=None, bcc=None, bccname=None, replyto=None, replytoname=None):                    
+        """
+        Send email using sendgrid.
+        """
+        if subject is None and content is None:
+            raise ValueError("Both subject and content cannot be None!")
+
+        data = {'to':to,'toname':toname,'from':vfrom,'fromname':fromname,'subject':subject or "", 'content':content or "",'cc':cc,'ccname':ccname,'bcc':bcc,'bccname':bccname,'replyto':replyto,'replytoname':replytoname}
+        try:
+            resp = self.api_call('sendgrid', method="post", data=data, verify=False)
+        except Exception as e:
+            print(f"Error with presigned url: {e}")
+            return False
+        return resp