dZddlmZddlmZddlmZddlZddlZddlmZm Z ddl m Z ddl m Z dd lmZdd lmZd Zd Zd efdZd efdZded eeeee efffdZded efdZdZy)zFLibrary for interacting with the Security Command Center Findings API.)absolute_import)division)unicode_literalsN)DictList)encoding)const)apis)errorscftjtjtjS)z8Returns the Security Command Center Findings API client.)r GetClientInstancer FINDINGS_API_NAMEFINDINGS_API_VERSION:lib/googlecloudsdk/api_lib/scc/iac_remediation/findings.py GetClientr&   7 7 % : : <  ( ( 66rct}t}t|}|j}d|d|_d|d|_|j j|}tj|}t||S)zMakes an API call to the Security Command Center Findings API. Args: finding_org_id: Organization ID of the finding finding_name: Canonical name of the finding. Returns: JSON response from the API call. zname : "z" zorganizations/z /sources/-) rrr"5SecuritycenterOrganizationsSourcesFindingsListRequestfilterparentorganizations_sources_findingsrr MessageToJsonValidateFinding)finding_org_idrclientmessages finding_idrequestresp json_resps r MakeApiCallr1=s ;& ](&*  J J L'zl#.'.#N#3:>'.  . . 3 3G <$$$T*)) r finding_jsonci}|ddddD]}|d|vrt||d<|ddk(rDd||dvr g||dd<|d||ddvsH||ddj|df|dd k(sod ||dvr g||dd <|d||dd vs||dd j|d|S) zFetches the IAMBindings from the finding data. Args: finding_json: JSON response from the API call to fetch the finding. Returns: IAM binding details per member. listFindingsResultsrfinding iamBindingsmemberactionADDroleREMOVE)dictappend)r2 iam_bindingsbindings rFetchIAMBindingr@Ss1,34Q7 Bgx ,(,l78$%xE! l78#45 513 WX&'.  WX-> ? F FWX&'.55gfoF  h & gh&78 846 WX&'1  WX-> ? I IWX&'188I rc|ddddS)zFetches the resource name from the finding data. Args: finding_json: JSON response from the API call to fetch the finding. Returns: Resource name for which the finding was generated. r4rresource displayNamer)r2s rFetchResourceNamerDqs + ,Q / ;M JJrcv tj|}|ddd}|d}|tj vrtj |t|}t|tjkDst|dkrtjt|y#tjxYw)ziChecks if the finding is supported or not. Args: finding_data: JSON response from the API call. r4rr5categoryN) jsonloadsr FindingNotFoundErrorr SUPPORTED_FINDING_CATEGORIESUnsupportedFindingCategoryErrorr@len SUPPORTED_IAM_MEMBER_COUNT_LIMITExcessiveMembersError) finding_datar5finding_categoryr>s rr)r)}s (::l+L01!4Y?GZ(U???  0 01A BB ., ,%@@@ \ Q   & &s<'8 99 (  % % ''s B""B8)__doc__ __future__rrrrHrtypingrrapitools.base.pyr*googlecloudsdk.api_lib.scc.iac_remediationr googlecloudsdk.api_lib.utilr .googlecloudsdk.command_lib.scc.iac_remediationr rrstrr"r1r@rDr)rrrrZsM&' %<,A< < 7s7$, S$sDI~& &'< KD KS K:r