PyQuantimClient 2.0.25__tar.gz → 2.0.27__tar.gz

{pyquantimclient-2.0.25 → pyquantimclient-2.0.27}/PKG-INFO

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

{pyquantimclient-2.0.25 → pyquantimclient-2.0.27}/PyQuantimClient.egg-info/PKG-INFO

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

{pyquantimclient-2.0.25 → pyquantimclient-2.0.27}/setup.py

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

{pyquantimclient-2.0.25 → pyquantimclient-2.0.27}/src/api.py

@@ -71,12 +71,14 @@ class quantim:
             resp = api_call_response.text
         return resp
 
-    def retrieve_s3_df(self, bucket, key, sep=','):
+    def retrieve_s3_df(self, bucket, key, sep = ',', res_url = False):
         '''
         Get series
         '''
-        data = {'bucket':bucket, 'key':key, 'sep':sep}
+        data = {'bucket':bucket, 'key':key, 'sep':sep, 'res_url':res_url}
         resp = self.api_call('retrieve_data_s3', method="post", data=data, verify=False)
+        if res_url:
+            return resp
         df = pd.DataFrame(resp)
         return df
 

{pyquantimclient-2.0.25 → pyquantimclient-2.0.27}/src/risk.py

@@ -173,8 +173,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], ['LIMITES_GEN', 'LIMITES_GEN_AGG', 'EMISIONES', 'Agregacion'])):
-            raise ValueError('You can only load LIMITES_GEN.csv, LIMITES_GEN_AGG.csv, EMISIONES.csv or Agregacion.csv. Please check file name.')
+        if not np.any(np.in1d(filename.split('.')[-2], ['LIMITES_GEN', 'LIMITES_GEN_AGG', 'EMISIONES', 'Agregacion', 'emisores_aprobados'])):
+            raise ValueError('You can only load LIMITES_GEN.csv, LIMITES_GEN_AGG.csv, EMISIONES.csv, Agregacion.csv or emisores_aprobados.csv. Please check file name.')
 
         resp = self.upload_with_presigned_url(file_path, "condor-sura", f"inputs/risk/static/{filename}")
         return resp
@@ -276,3 +276,16 @@ class risk_data(quantim):
         ref_dates = resp['ref_dates']
 
         return metrics, ref_dates
+    
+    def get_bd_pat(self, country, date, bucket="condor-pat", sep='|'):
+        '''
+        Get Value at Risk results.
+        '''
+        prefix = f"performance/{country}/{date.strftime('%Y/%m/%d')}/pat_{date.strftime('%Y%m%d')}.csv"
+        try:
+            url = self.retrieve_s3_df(bucket, prefix, sep = sep, res_url = True)
+            df = pd.read_csv(url, sep = sep)
+        except:
+            print("Archivo no disponible")
+            return None
+        return df