o Fj@sPddlZddlZejddddZddd Zdd d Zd d ZddZddZdS)N localhostiT)hostportdecode_responses<cCs`d|}t}||}t|d|t|t||it||t|}|||kdS)Nz velocity:r) request_count high_velocity)time redis_clientzremrangebyscorezaddstrexpirezcard)ipwindow_seconds thresholdkey current_time window_startrr//home/djax/ivt_ai_plugin/scoring/redis_state.pycheck_velocity s"    rcCs4d|d|}t|}t||ddt|iS)Nz duplicate::1 duplicate)r existssetexbool)r user_agentrr is_duplicaterrrcheck_duplicate&s r"cCs.t|}t|}d|d|}dt|iS)Nblocked:rblocked)rr r)rr rrrr check_blocked5s  r%c CsLt|}t|}d|d|}tj||||d|ttdddS)Nr#r,)rr risk_scorefeatures blocked_atmappingT)rr hsetjoinr )rr r'r(rrrr store_blocked@s  r.cCsTt|}t|}d|d|}t|dd}tj|||ttddd|iS)Nz suspicious:rcount)rr last_seenr*suspicious_count)rr hincrbyr,r )rr rr/rrrtrack_suspiciousSs  r4)rr)r) redisr Redisr rr"r%r.r4rrrrs