PyQuantimClient 2.0.58__tar.gz → 2.0.60__tar.gz

{pyquantimclient-2.0.58 → pyquantimclient-2.0.60}/PKG-INFO

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

{pyquantimclient-2.0.58 → pyquantimclient-2.0.60}/PyQuantimClient.egg-info/PKG-INFO

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

{pyquantimclient-2.0.58 → pyquantimclient-2.0.60}/PyQuantimClient.egg-info/SOURCES.txt

@@ -12,6 +12,7 @@ src/bi.py
 src/credit.py
 src/data.py
 src/energy.py
+src/mstar.py
 src/performance.py
 src/portfolios.py
 src/preprocess_co.py

{pyquantimclient-2.0.58 → pyquantimclient-2.0.60}/setup.py

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

pyquantimclient-2.0.60/src/mstar.py

@@ -0,0 +1,17 @@
+from .api import quantim
+
+class mstar_data(quantim):
+    def __init__(self, username, password, secretpool, env="pdn", api_url=None):
+        super().__init__(username, password, secretpool, env, api_url)
+
+    def load_token(self, file_path):
+        """
+        Load token file to s3.
+        """
+        # Validate filename:
+        filename = file_path.split("/")[-1]
+        if filename.split(".")[-1]!="txt":
+            raise ValueError("Extension must be txt. Please check file.")
+
+        resp = self.upload_with_presigned_url(file_path, "condor-sura", f"inputs/morningstar_api/token/token.txt")
+        return resp

{pyquantimclient-2.0.58 → pyquantimclient-2.0.60}/src/risk.py

@@ -36,16 +36,21 @@ class risk_data(quantim):
             resp = {'success':False, 'message':'Check permissions!'}
         return resp
 
-    def get_limits(self, portfolio):
+    def get_limits(self, portfolio=None, port_name=None):
         '''
         Get limits table.
         '''
+        if portfolio is None and port_name is None:
+            raise ValueError("Both portfolio and port_name cannot be none")
 
-        data = {'portfolio':portfolio.to_dict(orient="records")}
+        if portfolio:
+            data = {'portfolio':portfolio.to_dict(orient="records")}
+        else:
+            data = {'portfolioName':port_name}
         resp = self.api_call('limits', method="post", data=data, verify=False)
-        summ, detail = pd.DataFrame(resp['summ']), pd.DataFrame(resp['detail'])
+        summ, date_port = pd.DataFrame(resp['summ']), resp['date_port']
 
-        return port_date, summ, detail
+        return summ, date_port
 
     def run_limits(self, ref_date=None):
         '''