PyQuantimClient 2.0.60__tar.gz → 2.0.61__tar.gz

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

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

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

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

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

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

{pyquantimclient-2.0.60 → pyquantimclient-2.0.61}/src/returns.py

@@ -32,4 +32,28 @@ class returns(quantim):
             resp = {'success':False, 'message':'Check permissions!'}
 
         exp_ret, views_df, valid_dates_df = pd.DataFrame(resp['expected_rets']), pd.Series(resp['views_abs']) if resp['views_abs'] is not None else None, pd.Series(resp['valid_dates'])
-        return exp_ret, views_df, valid_dates_df
+        return exp_ret, views_df, valid_dates_df
+    
+    def views_transform(self, ref_curr, gaps_df, optim_assets_df, cons_df=None, since_date="2008-01-01", rets_period="monthly"):
+        '''
+        Gaps.
+        '''
+        gaps = gaps_df.to_dict(orient="records")
+        optim_assets = optim_assets_df.to_dict(orient="records")
+        cons = cons_df.to_dict(orient="records") if cons_df is not None else None
+
+        data = {'ref_curr':ref_curr, 'gaps':gaps, "optim_assets":optim_assets, "cons":cons,
+                 "since_date":since_date, "rets_period":rets_period}
+
+        for attempt in range(2):
+            try:
+                resp = self.api_call('views_transform', method="post", data=data, verify=False)
+                break  # success → exit loop
+            except Exception as e:
+                if attempt == 1:  # last attempt
+                    resp = {'success': False, 'message': 'Cannot optimize gaps!', 'error': str(e)}
+        if getattr(resp, "status", None) == "error":
+            resp = {'success': False, 'message': 'Cannot optimize gaps!', 'error': resp.msg}
+        else:
+            gaps_consol_df, te_df, invalid_assets, ref_dates = pd.DataFrame(resp['gaps_df']).set_index("Asset"), pd.Series(resp['te']), resp['invalid_assets'], resp['ref_dates']
+        return gaps_consol_df, te_df, invalid_assets, ref_dates