PyQuantimClient 2.0.34__tar.gz → 2.0.36__tar.gz

{pyquantimclient-2.0.34 → pyquantimclient-2.0.36}/PKG-INFO

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

{pyquantimclient-2.0.34 → pyquantimclient-2.0.36}/PyQuantimClient.egg-info/PKG-INFO

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

{pyquantimclient-2.0.34 → pyquantimclient-2.0.36}/setup.py

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

{pyquantimclient-2.0.34 → pyquantimclient-2.0.36}/src/bi.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-import re
+import re, os
 import datetime as dt
 from dateutil.relativedelta import relativedelta as rd
 import pandas as pd
@@ -128,4 +128,28 @@ class bi_data(quantim):
         resultado = pd.concat([asset_class, renta_fija, renta_variable, paises_emergentes, paises_desarrollados, otros])
         resultado.reset_index(inplace=True)
 
-        return resultado
+        return resultado
+
+    def get_asset_allocation(self, file, bucket="condor-sura"):
+        '''
+        Get Asset Allocation from Mstar.
+        '''
+        try:
+            file_name, file_extension = os.path.splitext(file)
+            prefix = f"output/morningstar_api/data/asset_allocation/{file}"
+            url = self.retrieve_s3_df(bucket, prefix, res_url=True)
+
+            if file_extension.lower() == '.csv':
+                df = pd.read_csv(url)
+                print(f"Archivo CSV '{file_name}' descargado correctamente.")
+            elif file_extension.lower() == '.xlsx':
+                df = pd.read_excel(url)
+                print(f"Archivo XLSX '{file_name}' descargado correctamente.")
+            else:
+                print("Tipo de archivo no soportado. Por favor, escriba un archivo CSV o XLSX.")
+                return None
+
+            return df
+        except Exception as e:
+            print(f"Error: {e}. Archivo no disponible")
+            return None

{pyquantimclient-2.0.34 → pyquantimclient-2.0.36}/src/risk.py

@@ -190,8 +190,8 @@ class risk_data(quantim):
         filename = file_path.split('/')[-1]
         if filename.split('.')[-1]!='csv':
             raise ValueError('Extension must be csv. Please check file.')
-        if not np.any(np.in1d(filename.split('.')[-2], ['cupos_contrapartes', 'emisores', 'liq_min'])):
-            raise ValueError('You can only load cupos_contrapartes.csv, emisores.csv or liq_min.csv. Please check file name.')
+        if not np.any(np.in1d(filename.split('.')[-2], ['cupos_contrapartes', 'emisores', 'liq_min', 'waivers'])):
+            raise ValueError('You can only load cupos_contrapartes.csv, emisores.csv, liq_min.csv, or waivers.csv. Please check file name.')
 
 
         resp = self.upload_with_presigned_url(file_path, "condor-sura", f"inputs/pretrade/CO/{filename}")