jjUdZddlmZddlmZmZeeZGddZda edze d<defd Z y) z Findings classifier - assigns character_id to validated findings. Uses rule-based logic to match findings to specialist characters. ) get_logger) CharacterIdFindingValidatedceZdZUdZgdZgdZgdZgdZgdZgdZ de d e e d e fd Z d d d ddddddddddd Zee e fed<ded efdZde ed eee effdZy)FindingsClassifiera Assigns findings to specialist characters based on content analysis. Characters: - Aegis (Security Paladin): injections, XSS, auth, CVEs - Schema (DB Mage): N+1, indexes, migrations, locks - Pixel (UX Bard): a11y, contrast, copy, states - Atlas (Architecture Explorer): patterns, coupling, consistency - Echo (Test Priest): coverage, edge cases, mocks - Codex (Docs Scribe): outdated docs, comments )z sql injectionxsscsrfauthentication authorizationsecurity vulnerabilitycvesanitizeescapetokenpasswordsecret credential encryptionhash)zn+1querydatabaseindex migration transactionlockdeadlockormsqlselectjoin performancedbtablecolumnz foreign key constraint) accessibilitya11yariacontrastcolorfontuiuxzuser experience usabilityz screen readerkeyboardfocussemanticzalt textlabelwcag) architecturepatterndesigncouplingcohesion dependencysoliddry separation abstraction interfaceclassmodule component structure organization consistencynaming)testcoveragez edge casemockstub assertionz unit test integratione2especexpectdescribez it shouldz test casescenario validation) documentationcommentjsdocreadmedocsoutdatedz missing docs undocumented docstringzapi docszinline commenttodofixmeexplaintextkeywordsreturncL|jtfd|DS)z.Calculate how many keywords match in the text.c3,K|] }|vsd yw)N).0keyword text_lowers 8/var/www/html/prparty/backend/app/services/classifier.py z>FindingsClassifier._calculate_keyword_score..AsF:0E1Fs )lowersum)selfr^r_rgs @rh_calculate_keyword_scorez+FindingsClassifier._calculate_keyword_score>sZZ\ FHFFFaegisschemapixelatlasechocodex) r rr#r.r-r5archtestsrGtestingrSrWdocCATEGORY_TO_CHARACTERfindingc Dt|dd}||jvr4|j|}tjd|j|||S|j d|j j}|j||j|j||j|j||j|j||j|j||j|j||jd}t|j!dd }||d k(r8|j"d k(rd }n&|j"d k(rd }n|j"dk(rd}nd}tjd|j|||S)z Assign a character to a finding based on content analysis. Args: finding: Validated finding to classify Returns: Character ID that best matches the finding categorygeneralfinding_classified_by_category) finding_id character_idmodel_category rorprqrrrsrtc |dS)Nrcrd)xs rhz5FindingsClassifier.classify_finding..vs 1rn)keyrcriticalrohighrrmediumrprtfinding_classified_by_keywords)rrscores)getattrryloggerinfoidtitle descriptionrjrmAEGIS_KEYWORDSSCHEMA_KEYWORDSPIXEL_KEYWORDSATLAS_KEYWORDS ECHO_KEYWORDSCODEX_KEYWORDSmaxitemsseverity)rlrzrrr^rs rhclassify_findingz#FindingsClassifier.classify_findingTs!*i@ T77 755nEL KK0"::)-    --'"5"5!67==?2249L9LM33D$:N:NO2249L9LM2249L9LM11$8J8JK2249L9LM  6<<>~>qA , 1 $:-& !!V+& !!X-' &  ,zz%  rnfindingsc Lggggggd}|D]'}|j|}||j|)tjdt |t |dt |dt |dt |dt |dt |d |S) z Classify multiple findings and group by character. Args: findings: List of validated findings Returns: Dict mapping character_id to their assigned findings rfindings_classifiedrorprqrrrsrt)total_findingsrorprqrrrsrt)rappendrrlen)rlr classifiedrzrs rhclassify_findingsz$FindingsClassifier.classify_findingss A   5G009L | $ + +G 4 5  !x=j)*z(+,j)*j)*Z'(j)*  rnN)__name__ __module__ __qualname____doc__rrrrrrstrlistintrmrydict__annotations__rrrrrdrnrhrr s N O N NM N GSGDIG#G -4S> 5(85[5n%'(% k4 011 2%rnrN _classifierr`c.t tatS)z)Get or create global classifier instance.)rrrdrnrhget_classifierrs(* rn) rapp.logging_configr app.modelsrrrrrrrrrdrnrhrsH*4 H eeR*. $ &-*rn