PyQuantimClient 2.0.72__tar.gz → 2.0.74__tar.gz

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

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

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

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

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

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

{pyquantimclient-2.0.72 → pyquantimclient-2.0.74}/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
@@ -202,7 +239,7 @@ class risk_data(quantim):
             df = None
         return df
 
-    def load_limits_params(self, file_path, encoding='latin-1'):
+    def load_limits_params(self, file_path, country):
         '''
         Load limits parameters file to s3.
         '''
@@ -210,7 +247,7 @@ class risk_data(quantim):
         filename = file_path.split('/')[-1]
         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/risk/limits/CO/{filename}")
+        resp = self.upload_with_presigned_url(file_path, "condor-sura", f"inputs/risk/limits/{country}/{filename}")
         return resp
 
     def load_pretrade_inputs(self, file_path):