PyQuantimClient 1.0.61__tar.gz → 1.0.63__tar.gz

{PyQuantimClient-1.0.61 → PyQuantimClient-1.0.63}/PKG-INFO

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

{PyQuantimClient-1.0.61 → PyQuantimClient-1.0.63}/PyQuantimClient.egg-info/PKG-INFO

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

{PyQuantimClient-1.0.61 → PyQuantimClient-1.0.63}/PyQuantimClient.egg-info/SOURCES.txt

@@ -7,6 +7,7 @@ PyQuantimClient.egg-info/top_level.txt
 src/__init__.py
 src/api.py
 src/benchmarks.py
+src/bi.py
 src/data.py
 src/energy.py
 src/portfolios.py

{PyQuantimClient-1.0.61 → PyQuantimClient-1.0.63}/setup.py

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

{PyQuantimClient-1.0.61 → PyQuantimClient-1.0.63}/src/api.py

@@ -47,7 +47,12 @@ class quantim:
         else:
             print("Method not supported!")
             return None
-        return json.loads(api_call_response.text)
+        
+        try:
+            resp = json.loads(api_call_response.text)
+        except:
+            resp = api_call_response.text
+        return resp
 
     def retrieve_s3_df(self, bucket, key, sep=','):
         '''

PyQuantimClient-1.0.63/src/bi.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+import pandas as pd
+import numpy as np
+import datetime as dt
+from dateutil.relativedelta import relativedelta as rd
+from .api import quantim
+
+class bi_data(quantim):
+    def __init__(self, username, password, secretpool, env="pdn", api_url=None):
+        super().__init__(username, password, secretpool, env, api_url)
+
+    def get_positions_afps_cl(self, ref_date=None, bucket="condor-sura"):
+        '''
+        Get Value at Risk results and suport information.
+        '''
+        ref_date = dt.datetime.today().replace(day=1, hour=0, minute=0, second=0, microsecond=0) - rd(days=1) if ref_date is None else dt.datetime.strptime(ref_date, '%Y-%m-%d') 
+        key = f'{prefix}{ref_date.year}/{ref_date.strftime("%m")}/data.json'
+
+        data = {'bucket':bucket, 'key':key}
+        try:
+            resp = self.api_call('retrieve_json_s3', method="post", data=data, verify=False)
+            keys = list(resp.keys())
+            resp_dfs = {k:pd.DataFrame(resp[k]) for k in keys}
+        except:
+            print(f"Data not available for {ref_date}. Try previous month!")
+            keys, resp_dfs = None, None
+        return keys, resp_dfs
+

{PyQuantimClient-1.0.61 → PyQuantimClient-1.0.63}/src/risk.py

@@ -138,3 +138,36 @@ class risk_data(quantim):
         cfs = pd.read_csv(resp['cf_url']) if return_cfs else None
 
         return irl_report, cfs
+
+    def load_fund_series_cl(self, file_path, sep=";", encoding='utf-8', blocks=1):
+        '''
+        Load series Chile to s3.
+        '''
+        # Validate filename:
+        filename = file_path.split('/')[-1]
+        label = filename.split('.')[-2]
+        if filename.split('.')[-1]!='csv':
+            raise ValueError('Extension must be csv. Please check file.')
+        if not np.any(np.in1d(label, ['series_performance'])):
+            raise ValueError('You can only load series_performance.csv')
+            
+        if get_csv_separator(file_path)!=sep:
+            raise ValueError(f'Separator must be ({sep}). Please check file.')
+    
+        # Read new file:
+        try:
+            df_new = pd.read_csv(file_path, sep=sep, encoding=encoding)
+        except:
+            raise ValueError("Cannot read file. Check path and encoding.")
+
+        # Load data
+        li = int(np.ceil(len(df_new)/blocks))
+        for i in range(blocks):
+            payload = df_new.iloc[(i*li):(i+1)*li, :].to_dict(orient='records')
+            data = {'bucket':'condor-sura', 'file_name':f"inputs/benchmarks/performance/cl/{label}{i if i>0 else ''}.csv", 'payload':payload, 'sep':sep, 'overwrite':True, 'encoding':encoding}
+            try:
+                resp = self.api_call('load_data_s3', method="post", data=data, verify=False)
+            except:
+                resp = {'success':False, 'message':'Check permissions or file size!'}
+                return resp
+        return resp