PyQuantimClient 2.0.22__tar.gz → 2.0.24__tar.gz

{pyquantimclient-2.0.22 → pyquantimclient-2.0.24}/PKG-INFO

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

{pyquantimclient-2.0.22 → pyquantimclient-2.0.24}/PyQuantimClient.egg-info/PKG-INFO

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

{pyquantimclient-2.0.22 → pyquantimclient-2.0.24}/setup.py

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

{pyquantimclient-2.0.22 → pyquantimclient-2.0.24}/src/risk.py

@@ -80,7 +80,7 @@ class risk_data(quantim):
         
         '''
         if new_oper is not False:
-            cols=['type', 'secDesc1', 'isin', 'assetId', 'issuerId', 'issuerLongName', 'ccy', 'duration', 'avgRating', 'ratingType', 'SIM_SECTOR', 'linkedEntity', 'quantity', 'mktValue', 'maturity']
+            cols=['type', 'secDesc1', 'counterparty', 'isin', 'assetId', 'issuerId', 'issuerLongName', 'ccy', 'duration', 'avgRating', 'ratingType', 'SIM_SECTOR', 'linkedEntity', 'quantity', 'mktValue', 'maturity']
             if not set(cols).issubset(new_oper.columns):
                 raise ValueError(f"Las nuevas operaciones deben contener la siguiente informacion: {cols}. Por favor revisar")
             else:
@@ -100,6 +100,30 @@ class risk_data(quantim):
             limites_detalle = pd.DataFrame(resp['detail'])
             return limites_resumen, portfolio, limites_detalle
         return limites_resumen, portfolio
+    
+    def pretrade_register_ops(self, portfolioName, new_data, subgroup=False, sep= '|'):
+        '''
+        Append new opers to pretrade
+        
+        '''
+        if not subgroup:
+            subgroup = portfolioName
+        cols=['type', 'secDesc1', 'counterparty', 'isin', 'assetId', 'issuerId', 'issuerLongName', 'ccy', 'duration', 'avgRating', 'ratingType', 'SIM_SECTOR', 'linkedEntity', 'quantity', 'mktValue', 'maturity']
+        missing_cols = set(cols) - set(new_data.columns)
+        if missing_cols:
+            raise ValueError(f"Las nuevas operaciones deben contener las siguientes columnas faltantes: {missing_cols}. Por favor revisar.")
+        if not set(new_data["type"].unique()).issubset({"BUY", "SELL"}):
+            raise ValueError("La columna type unicamente recibe los valores 'BUY' o 'SELL'.")
+        if  not set(["portfolioName","subGroup"]).issubset(new_data.columns):
+            new_data["portfolioName"],new_data["subGroup"]=portfolioName,subgroup
+        new_data = new_data[["portfolioName","subGroup"]+cols]
+        new_data["registered"]=True
+        new_data=new_data.to_dict(orient="records")
+        fecha=dt.date.today().strftime("%Y/%m/%d")
+        data = {'bucket': 'condor-sura', 'key': f'output/pretrade/CO/trades/{fecha}/trades.csv', 'sep': sep, 'new_data': new_data}
+        resp = self.api_call('append_data', method="post", data=data, verify=False)
+        msg=resp["msg"]
+        return msg
 
     def get_cashflows(self, client_id=None, port_type=None):
         '''
@@ -157,16 +181,19 @@ class risk_data(quantim):
         resp = self.upload_with_presigned_url(file_path, "condor-sura", f"inputs/risk/static/{filename}")
         return resp
 
-    def load_vu_pas(self, file_path):
+    def load_pretrade_inputs(self, file_path):
         '''
-        Load PAs prices to s3.
+        Load pretrade inputs prices to s3.
         '''
         # Validate filename:
         filename = file_path.split('/')[-1]
-        if filename != 'BD_PAS.csv':
-            raise ValueError('You can only load a file named BD_PAS.csv. Please check file name and extension.')
+        if filename.split('.')[-1]!='csv':
+            raise ValueError('Extension must be csv. Please check file.')
+        if not np.any(np.in1d(filename.split('.')[-2], ['cupos_contrapartes', 'emisores', 'liq_min'])):
+            raise ValueError('You can only load cupos_contrapartes.csv, emisores.csv or liq_min.csv. Please check file name.')
+
 
-        resp = self.upload_with_presigned_url(file_path, "condor-sura", f"inputs/portfolios/nav/CO/{filename}")
+        resp = self.upload_with_presigned_url(file_path, "condor-sura", f"inputs/pretrade/CO/{filename}")
         return resp
 
     def get_irl(self, ref_date=None, return_cfs=False):