PyQuantimClient 2.0.74__tar.gz → 2.0.76__tar.gz

{pyquantimclient-2.0.74 → pyquantimclient-2.0.76}/PKG-INFO

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

{pyquantimclient-2.0.74 → pyquantimclient-2.0.76}/PyQuantimClient.egg-info/PKG-INFO

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

{pyquantimclient-2.0.74 → pyquantimclient-2.0.76}/setup.py

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

{pyquantimclient-2.0.74 → pyquantimclient-2.0.76}/src/risk.py

@@ -231,13 +231,17 @@ class risk_data(quantim):
         '''
         today = dt.datetime.today().replace(hour=0, minute=0, second=0, microsecond=0)
         ref_date = (today - dt.timedelta(days=(today.weekday() - 4) % 7)).strftime("%Y%m%d") if ref_date is None else ref_date
+        files = ["var_386","results_consol"]
         try:
-            df = self.retrieve_s3_df(bucket, f'{prefix}{ref_date}/var_386_{ref_date}.csv', sep=sep, prefix=prefix)
+            dfs = {}
+            for file_i in files:
+                dfs[file_i] = self.retrieve_s3_df(bucket, f'{prefix}{ref_date}/{file_i}_{ref_date}.csv', sep=sep, prefix=prefix)
             print(f'Var 386 ready!')
+            dfs = dfs.values()
         except:
             print(f"Var 386 not available for {ref_date}!")
-            df = None
-        return df
+            dfs = None
+        return dfs
 
     def load_limits_params(self, file_path, country):
         '''
@@ -420,4 +424,22 @@ class risk_data(quantim):
         if filename.split('.')[-1]!='csv':
             raise ValueError('Extension must be csv. Please check file.')
         resp = self.upload_with_presigned_url(file_path, "condor-sura", f"inputs/portfolios/mappings/PE/{filename}")
+        return resp
+
+    def load_ret_peers_mx(self, file_path):
+        '''
+        Load Mexico peers rets to s3.
+        '''
+        # Validar extension:
+        filename = file_path.split('/')[-1]
+        if filename.split('.')[-1]!='csv':
+            raise ValueError('Extension must be csv. Please check file.')
+        # Validar separador
+        try:
+            df = pd.read_csv(file_path, sep=";", nrows=5)
+            if df.shape[1] == 1:
+                raise ValueError('Por favor valide el separador del archivo, debe ser punto y coma (;).')
+        except Exception as e:
+            raise ValueError(str(e))
+        resp = self.upload_with_presigned_url(file_path, "condor-sura", f"inputs/benchmarks/peers/MX/{filename}")
         return resp