modelswrkpi 9.0.1__py3-none-any.whl → 9.1.0__py3-none-any.whl

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}'")

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)
@@ -113,7 +116,7 @@ class LoadBalancerV7(LoadBalancerV6):
         up.approval_rate = up.approval_rate.fillna(0)
         up.soft_cap_alerted = up.soft_cap_alerted.fillna(False)
         up.visa_soft_cap_alerted = up.visa_soft_cap_alerted.fillna(False)
-        up.drop('prev_mtd', axis=1, errors='ignore', inplace=True)
+        up.drop(['prev_mtd', 'prev_visa_mtd'], axis=1, errors='ignore', inplace=True)
         up = up.drop_duplicates(['gateway_id', 'router_id'])
         # self.engine.execute(f'truncate {self.schema}.{self.table}')
         self.upsert(up.dropna())
@@ -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.dist-info → modelswrkpi-9.1.0.dist-info}/METADATA

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

{modelswrkpi-9.0.1.dist-info → modelswrkpi-9.1.0.dist-info}/RECORD

@@ -67,8 +67,8 @@ models/bro_clicks/scrub_settings.cpython-310-x86_64-linux-gnu.so,sha256=gKo6LHc-
 models/bro_clicks/splitter.cpython-310-x86_64-linux-gnu.so,sha256=27kzt-JDllwAXMFq8Xe2Av0RIDpwFK4h_t6BvytGsak,57712
 models/bro_clicks/load_balancer/__init__.py,sha256=TMIh-RpmoCaEMTLU7geYtcDpygX5KFM8B4RRsHcKutA,432
 models/bro_clicks/load_balancer/compile.py,sha256=0mD0LCkkR2aasWk8AdxVA6ylHN28l4bV6vwkA2TUQF4,661
-models/bro_clicks/load_balancer/v1.cpython-310-x86_64-linux-gnu.so,sha256=iFWITw0tjieXWku8F2lg06n7Ar29r57AjA_ZUV35SO4,172976
-models/bro_clicks/load_balancer/v1.py,sha256=ikWbsxGFsYNTVbDGp43OvcbAGITFT9pt8JqEiq9zheI,10840
+models/bro_clicks/load_balancer/v1.cpython-310-x86_64-linux-gnu.so,sha256=Eqst8z6bGCUZ6DM8s1cPDA4LEhWVrQa82d97YlPJirc,177072
+models/bro_clicks/load_balancer/v1.py,sha256=-vCW9oGwMi4jkcVyR6yi6UA5O0xP1Q7EfhKcpTvQUc4,11115
 models/bro_clicks/load_balancer/v2.cpython-310-x86_64-linux-gnu.so,sha256=VCGHr8KSipO2E0IfJ0zSpdte-xw_xsSK5vRWujtumnU,140368
 models/bro_clicks/load_balancer/v2.py,sha256=WSM6lKtyLrzx1xdhL3ncuH4QH-FBnS5Gm0uDFAtNHXo,7303
 models/bro_clicks/load_balancer/v3.cpython-310-x86_64-linux-gnu.so,sha256=f03MbM9KyuKpctvJ-6g6pnhMbnsasPtTkFVeQlRXUeA,256496
@@ -79,8 +79,8 @@ models/bro_clicks/load_balancer/v5.cpython-310-x86_64-linux-gnu.so,sha256=gHA2J0
 models/bro_clicks/load_balancer/v5.py,sha256=jA1KPoJLcm8szMGhaGuMSH-dZ6d0dwNv1PgGm0UHsJ0,8036
 models/bro_clicks/load_balancer/v6.cpython-310-x86_64-linux-gnu.so,sha256=1X_3SFmkafbH3Ejj8q7aHUHHxNzpP6zlJXNTKehks2s,165360
 models/bro_clicks/load_balancer/v6.py,sha256=oD_VYJ_Yvm5YJAGBx4IYxQHvIbKTeRq-Y00O38adOyQ,9587
-models/bro_clicks/load_balancer/v7.cpython-310-x86_64-linux-gnu.so,sha256=iGvt5u_PYzCMhf4MBbRWhH29sR-PyBSz7yAayf4qBVI,210064
-models/bro_clicks/load_balancer/v7.py,sha256=AJUdUt8vy4g639AAKaS5VWilfSOC74ycB3Pf471UOE0,18121
+models/bro_clicks/load_balancer/v7.cpython-310-x86_64-linux-gnu.so,sha256=82CbiqQvJZjYBYSGwc45fEX8Pfpnkb6Br8ipRet0-5I,214160
+models/bro_clicks/load_balancer/v7.py,sha256=cEQQmLPBlKoSau9V-PAXhq0cHsfU661MiyNHNbE49dk,18457
 models/bro_clicks/load_balancer/v8.py,sha256=tkk6FpLgQvKCaFf2kmKsHhW_Pj4gJq4PcSe-DKcIUC4,264
 models/cb911/__init__.py,sha256=Y3tX5WQYFUswdMMUpWqa9U_fNObUAhXcVrix3eXYP4k,64
 models/cb911/alerts.cpython-310-x86_64-linux-gnu.so,sha256=LCsUfaTzdQKS6ZaKQbfLCNTZbpZqaJmEsWDTUOdHUrI,61808
@@ -136,7 +136,7 @@ models/reports/table_reports/inactive_report.cpython-310-x86_64-linux-gnu.so,sha
 models/reports/table_reports/mtd_report.cpython-310-x86_64-linux-gnu.so,sha256=dHko7wWia9y9d-gIPC8uxycUnobII6hQgairSPRU168,74096
 models/vaultx/__init__.py,sha256=PwYMmWXQMojeyFnQuQW3wNlv9dLUmX_uiDfCOHJN7XI,46
 models/vaultx/ssc_cont.cpython-310-x86_64-linux-gnu.so,sha256=69kGuEko6oJEqxopS_4BxoX5S-Twq94eFwYJFEu_4B8,94992
-modelswrkpi-9.0.1.dist-info/METADATA,sha256=eGnsJVZAHfaAtMyEqhIuUB3bPgfExeaEvjDkZm14RLI,346
-modelswrkpi-9.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-modelswrkpi-9.0.1.dist-info/top_level.txt,sha256=fOgLNy4xnMfp0ihe_kFhY_yH71kNkdEQG68z8FoLR4g,19
-modelswrkpi-9.0.1.dist-info/RECORD,,
+modelswrkpi-9.1.0.dist-info/METADATA,sha256=Q06H4MrzsFOGRBgFbNnGjoPoL3qU3Gj6J4fgMXTLwOA,346
+modelswrkpi-9.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+modelswrkpi-9.1.0.dist-info/top_level.txt,sha256=fOgLNy4xnMfp0ihe_kFhY_yH71kNkdEQG68z8FoLR4g,19
+modelswrkpi-9.1.0.dist-info/RECORD,,