PyQuantimClient 2.0.71__tar.gz → 2.0.73__tar.gz

{pyquantimclient-2.0.71 → pyquantimclient-2.0.73}/PKG-INFO

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

{pyquantimclient-2.0.71 → pyquantimclient-2.0.73}/PyQuantimClient.egg-info/PKG-INFO

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

{pyquantimclient-2.0.71 → pyquantimclient-2.0.73}/setup.py

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

{pyquantimclient-2.0.71 → pyquantimclient-2.0.73}/src/risk.py

@@ -95,6 +95,43 @@ class risk_data(quantim):
             raise ValueError("Error: No se ha podido iniciar el proceso de var formato 386 o ya existe una ejecución en curso. Intente de nuevo en algunos minutos.")
         return None
 
+    def run_pat(self, start_date, end_date, country="CO"):
+        '''
+        Run performance colombia.
+        '''
+        # Validación de formato
+        try:
+            dt.datetime.strptime(start_date, "%Y-%m-%d")
+        except:
+            raise ValueError(f"start_date '{start_date}' no tiene el formato string yyyy-mm-dd")
+
+        try:
+            dt.datetime.strptime(end_date, "%Y-%m-%d")
+        except:
+            raise ValueError(f"end_date '{end_date}' no tiene el formato string yyyy-mm-dd")
+
+        data = {
+            'jobname': 'job_proc_pat',
+            "add_args": {
+                "--countries": f'["{country}"]',
+                "--start_date": start_date,
+                "--end_date": end_date
+            }
+        }
+
+        try:
+            resp = self.api_call('run_glue_job', method="post", data=data, verify=False)
+            msg = (
+                f"La ejecucion performance para {country} entre las fechas {start_date} - {end_date} se ha activado exitosamente"
+                if resp['status'] == 'RUNNING'
+                else "Error: No se ha podido iniciar el proceso performance o ya existe una ejecución en curso. Intente de nuevo en algunos minutos."
+            )
+            print(msg)
+        except:
+            raise ValueError("Error: No se ha podido iniciar el proceso performance o ya existe una ejecución en curso. Intente de nuevo en algunos minutos.")
+
+        return None
+
     def get_portfolio(self, client_id=None, port_type=None, ref_date=None):
         '''
         Get portfolio
@@ -183,8 +220,8 @@ class risk_data(quantim):
                 dfs[file_i] = self.retrieve_s3_df(bucket, f'{prefix}{ref_date}/{file_i}_{ref_date}.csv', sep=sep, prefix = prefix)
                 print(f'{file_i} ready!')
             dfs = dfs.values()
-        except:
-            print(f"Files not available for {ref_date}!")
+        except Exception as e:
+            print(f"Error descargando archivos para la fecha {ref_date}: {e}")
             dfs = None
         return dfs