modelswrkpi 9.0.1__tar.gz → 9.0.3__tar.gz

{modelswrkpi-9.0.1/modelswrkpi.egg-info → modelswrkpi-9.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modelswrkpi
-Version: 9.0.1
+Version: 9.0.3
 Author: Kelly Kapowski
 Requires-Python: >=3.10
 Requires-Dist: psycopg2

{modelswrkpi-9.0.1 → modelswrkpi-9.0.3}/models/bro_clicks/load_balancer/v1.py

@@ -217,8 +217,16 @@ class LoadBalancer(Db):
             {"set visa_soft_cap_alerted=true" if cc_type.lower()=='visa' else "set  soft_cap_alerted=true"}
              where crm_id= '{crm_id}'""")
 
-    def disable(self, crm_id, gateway_id):
+    def disable(self, crm_id, gateway_id, cc_type=None, **kw):
+        set_clause = 'set enable_initials=false'
+        if cc_type:
+            if cc_type.lower() == 'visa':
+                set_clause = 'visa_enabled = false'
+            elif cc_type:
+                set_clause = 'mc_enabled = false'
+
         self.engine.execute(
-            f"""Update {self.schema}.{self.table} set enable_initials=false where crm_id= '{crm_id}' and gateway_id = '{int(gateway_id)}'""")
-        self.db_p.engine.execute(
+            f"""Update {self.schema}.{self.table} {set_clause} where crm_id= '{crm_id}' and gateway_id = '{int(gateway_id)}'""")
+        if not cc_type:
+            self.db_p.engine.execute(
             f"update ui_54407332_clients.gateway_settings set enable_initials=false where crm_id='{crm_id}' and gateway_id='{gateway_id}'")

{modelswrkpi-9.0.1 → modelswrkpi-9.0.3}/models/bro_clicks/load_balancer/v7.py

@@ -39,7 +39,9 @@ class LoadBalancerV7(LoadBalancerV6):
                                                   d.soft_cap_alerted,
                                                    d.visa_soft_cap_alerted,  
                                                   d.initial_count_mtd  as  prev_mtd,
-                                                  d.visa_count_mtd  as  prev_visa_mtd                        
+                                                  d.visa_count_mtd  as  prev_visa_mtd,
+                                                 TRUE as visa_enabled,
+                                                 TRUE as mc_enabled                         
 
                   FROM {sk}.mids a 
                   LEFT JOIN {sk}.steps b on b.mid_id = a.mid_id 
@@ -105,7 +107,8 @@ class LoadBalancerV7(LoadBalancerV6):
             up.visa_dly_cap = pd.np.floor((up.visa_monthly_cap - up.visa_count_mtd) / drim)
             up.loc[up.dly_initial_cap < 0, 'dly_initial_cap'] = 0
             up.loc[up.visa_dly_cap < 0, 'visa_dly_cap'] = 0
-
+        up.visa_enabled = up.visa_count_mtd.fillna(0) < up.visa_monthly_cap.fillna(0)
+        up.mc_enabled = up.initial_count_mtd.fillna(0) < up.monthly_initial_cap.fillna(0)
         up.dly_initial_cap = up.dly_initial_cap.fillna(11)
         up.visa_dly_cap = up.visa_dly_cap.fillna(11)
         up.dly_min_approval_rate = up.dly_min_approval_rate.fillna(30)
@@ -121,14 +124,16 @@ class LoadBalancerV7(LoadBalancerV6):
     def gty_qry(self, crm_id, date, step, processor, cc_type=False, decs='', proc_excl=[], is_tds=None,
                 is_prepaid=False, is_decline_salvage=False, **kw):
         p_ex = ''
+        cc_enable_clause = ''
         if proc_excl and len(proc_excl) and not processor:
             p_ex = f"and a.processor not ilike all(ARRAY{[f'%%{p}%%' for p in proc_excl]}::text[])"
 
         if str(cc_type).lower() == 'visa':
             dyna_cap_cols = "a.visa_fill_pct fill_pct, a.visa_dly_cap dly_initial_cap, a.visa_monthly_cap monthly_initial_cap,  a.visa_soft_cap_alerted soft_cap_alerted, a.visa_count_mtd initial_count_mtd, a.visa_count initial_count, a.soft_cap_alerted sca2"
+            cc_enable_clause = 'and a.visa_enabled'
         else:
             dyna_cap_cols = "a.fill_pct, a.dly_initial_cap, a.monthly_initial_cap, a.soft_cap_alerted, initial_count_mtd, initial_count, a.visa_soft_cap_alerted sca2"
-
+            cc_enable_clause = 'and a.mc_enabled'
 
         return f""" --LEFT HERE NEED TO GET MCC!
                        select a.gateway_id::int,  priority,   a.approval_rate, a.date_added, a.processor, a.mid_id, b.mcc,
@@ -150,6 +155,7 @@ class LoadBalancerV7(LoadBalancerV6):
                                      {f"and (tds_exclude_cc_types is null or tds_exclude_cc_types not ilike '%%{cc_type}%%')" if cc_type and is_tds else ""}
                                      {f"and allow_prepaid" if is_prepaid else ""}
                                      {f"and allow_non_prepaid" if not is_prepaid else ""}
+                                     {cc_enable_clause}
                                      --and fill_pct < 1                         
 
                        --order by date_added desc, approval_rate desc, fill_pct asc limit 1
@@ -276,12 +282,9 @@ class LoadBalancerV7(LoadBalancerV6):
                     return 'out of processing'
                 r2 = res.to_dict(orient='records')[0]
                 if r2['initial_count_mtd'] >= r2['monthly_initial_cap']:
-                    if cc_type == 'visa':
-                        #self.alert_cb('hard_cap_alert', crm_id=crm_id, gateway_id=r2['gateway_id'], cc_type=cc_type)
-                        # self.disable(crm_id=crm_id, gateway_id=r2['gateway_id'])
-                        res = res.loc[res.gateway_id != r2['gateway_id']]
-                    else:
-                        self.alert_cb('hard_cap_alert', crm_id=crm_id, gateway_id=r2['gateway_id'], cc_type='Non-Visa')
+                    self.alert_cb('hard_cap_alert', crm_id=crm_id, gateway_id=r2['gateway_id'], cc_type=cc_type)
+                    self.disable(crm_id=crm_id, gateway_id=r2['gateway_id'], cc_type=cc_type)
+                    res = res.loc[res.gateway_id != r2['gateway_id']]
                     return _get_aft_sc()
                 self.set_iin(**r2)
                 r2['is_tds'] = is_tds

{modelswrkpi-9.0.1 → modelswrkpi-9.0.3/modelswrkpi.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modelswrkpi
-Version: 9.0.1
+Version: 9.0.3
 Author: Kelly Kapowski
 Requires-Python: >=3.10
 Requires-Dist: psycopg2

{modelswrkpi-9.0.1 → modelswrkpi-9.0.3}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = modelswrkpi
-version = 9.0.1
+version = 9.0.3
 author = Kelly Kapowski
 
 [options]