PyQuantimClient 2.0.55__tar.gz → 2.0.57__tar.gz

{pyquantimclient-2.0.55 → pyquantimclient-2.0.57}/PKG-INFO

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

{pyquantimclient-2.0.55 → pyquantimclient-2.0.57}/PyQuantimClient.egg-info/PKG-INFO

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

{pyquantimclient-2.0.55 → pyquantimclient-2.0.57}/setup.py

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

{pyquantimclient-2.0.55 → pyquantimclient-2.0.57}/src/product.py

@@ -6,11 +6,15 @@ class product_data(quantim):
     def __init__(self, username, password, secretpool, env="pdn", api_url=None):
         super().__init__(username, password, secretpool, env, api_url)
 
-    def get_returns_report_cl(self, ref_date):
+    def get_returns_report_cl(self, ref_date, row_format=False):
         '''
         Retrive products returns report for clients in CL.
         '''
-        filename = f"prices_cl/report/{ref_date.replace('-','/')}/rentabilidades.csv"
+        if row_format:
+            filename = f"prices_cl/report/{ref_date.replace('-','/')}/rentabilidades_filas.csv"
+        else:
+            filename = f"prices_cl/report/{ref_date.replace('-','/')}/rentabilidades.csv"
+
         try:
             url = self.retrieve_s3_df(bucket="condor-reporting", key=filename, res_url=True)
             df = pd.read_csv(url, sep="|")

{pyquantimclient-2.0.55 → pyquantimclient-2.0.57}/src/risk.py

@@ -336,7 +336,7 @@ class risk_data(quantim):
         patron = r"^Positions_(\d{8})\.csv$"
         match = re.match(patron, filename)
         if not match:
-            raise ValueError('You can only update files named Positions_AAAAMMDD.csv. Please check.')
+            raise ValueError('You can only update files named Positions_YYYYMMDD.csv. Please check.')
         # validar que la fecha sea valida
         fecha_str = match.group(1)
         try:
@@ -347,4 +347,27 @@ class risk_data(quantim):
         key= f"inputs/portfolios/CO/{filename}"
         bucket="condor-sura"
         resp = self.upload_with_presigned_url(file_path, bucket, key, prefix)
+        return resp
+    
+    def load_haircuts(self, file_path):
+        '''
+        Load haircuts banrep.
+        '''
+        # Load data
+        filename = file_path.split("/")[-1]
+        # validar el nombre del archivo
+        patron = r"^haircuts_(\d{6})\.xlsx$"
+        match = re.match(patron, filename)
+        if not match:
+            raise ValueError('You can only update files named haircuts_YYYYMM.xlsx. Please check.')
+        # validar que la fecha sea valida
+        fecha_str = match.group(1)
+        try:
+            dt.datetime.strptime(fecha_str, "%Y%m")
+        except ValueError:
+            raise ValueError('The date is not valid. Format must be YYYYMM')
+        prefix="inputs/risk/varliquid/haircuts/haircuts_"
+        key= f"inputs/risk/varliquid/haircuts/{filename}"
+        bucket="condor-sura"
+        resp = self.upload_with_presigned_url(file_path, bucket, key, prefix)
         return resp