elʝ$g d dl mZ d dlmc mZ d dlmZmZ d dlm Z m
Z
d dl m Z d dl
mZ d dlmZmZ d dlmZmZ d d lmZ d
Z G d d ej ZdS )
) DoesNotExistN)ValidationErrorlookup)generate_warningsrun_in_executor_decorator)Scope)
CountryAPI)WhitelistCacheUpdate IpsetUpdate)Country CountryListsend_to_socketc fd}|S )Nc fd}|S )Nc ` K | g|R i | d {V }t di d {V |S )Nmethod)msgr )selfargskwargsresultfuncr s O/opt/imunify360/venv/lib/python3.11/site-packages/im360/simple_rpc/countries.pywrapperzDnotify_resident_part.._notify_resident_part..wrapper se 46t666v66666666F h%7888 8 8 8 8 8 8 8M )r r r s ` r _notify_resident_partz3notify_resident_part.._notify_resident_part s) r r )r r s ` r notify_resident_partr s$ ! r c e Zd Zej Z ej ddd edd Z ej ddd dd Z
ej ddd d Z ej ddd
e e
j d Z ej d dd edd
Z ej d dd e ej d Z ej d dd e ej dd Z ej d dd
e ej d Zed Zd Zd Zd Zd ZdS )CountriesEndpoints blacklistcountrylistNc d t j di |t j d||t j gd|fS N)limitoffsetby_listr )r
fetch_countfetchBLACKr r' r( r s r blacklist_country_listz)CountriesEndpoints.blacklist_country_list! U
# - -f - -
$*+
r addc T K | t j || d {V S N) _country_addr
r, r itemscomments r blacklist_country_addz(CountriesEndpoints.blacklist_country_add. s3 &&{'8%IIIIIIIIIr editc T K | t j || d {V S r2 )
_country_editr
r, r4 s r blacklist_country_editz)CountriesEndpoints.blacklist_country_edit2 s3 '' (95'JJJJJJJJJr deletec R K | t j | d {V S r2 )_country_deleter
r, r r5 s r blacklist_country_deletez+CountriesEndpoints.blacklist_country_delete6 3 ))+*;UCCCCCCCCCr whitelistc d t j di |t j d||t j gd|fS r& )r
r* r+ WHITEr- s r whitelist_country_listz)CountriesEndpoints.whitelist_country_list; r/ r c T K | t j || d {V S r2 )r: r
rD r4 s r whitelist_country_editz)CountriesEndpoints.whitelist_country_editH s5 '' (95'JJJJJJJJJr c T K | t j || d {V S r2 )r3 r
rD r4 s r whitelist_country_addz(CountriesEndpoints.whitelist_country_addM s5 &&{'8%IIIIIIIIIr c R K | t j | d {V S r2 )r> r
rD r? s r whitelist_country_deletez+CountriesEndpoints.whitelist_country_deleteR rA r c V t j | n# t $ r Y dS w xY wdS )N)codeFT)r getr )r country_codes r _get_countryz CountriesEndpoints._get_countryW sD K\ * * * * * 55 ts
&&c K g }|D ]3}| | d {V s| | 4|r5t d d | d S )N)rO zCountry does not exist {}z, )rP appendr formatjoin)r country_listwarning_countriesrO s r _check_countries_existsz*CountriesEndpoints._check_countries_exists_ s ( 7 7L ++ +FFFFFFFF
7!(( 666 !+22II /00
r c K | | d {V t j || d {V \ }}t ||||ddd S )N)r6 z){}/{} country(s) were successfully editedz Noop: unable to edit {} in {} Country {} is already in {} list
dest_listnameall_listsuccess_warningfailure_warningin_another_list_warning)rW r r8 r r listnamer5 r6 affected not_affecteds r r: z CountriesEndpoints._country_editl s **5111111111'1ug'N'N'N!N!N!N!N!N!N , " G ;$F
r c K | | d{V t j ||| d{V \ }}|t j k r|rt j d{V t ||||ddd S )z0Add Country to appropriate ipset and CountryListN)ra r6 z#{}/{} ip(s) were successfully addedz Noop: unable to add {} from {}rY rZ )rW r blockr
r, webshield update_country_blacklist_configr r` s r r3 z CountriesEndpoints._country_addy s **5111111111'1'7 Hg(
(
(
"
"
"
"
"
"
, {( ( (X (;== = = = = = = = " A <$F
r c K | | d {V t j || d {V \ }}|t j k r|rt j d {V t ||||ddd S )N)ra z%{}/{} ip(s) were successfully deletedz!Noop: unable to delete {} from {}rY rZ )rW r unblockr
r, rf rg r )r ra r5 rb rc s r r> z"CountriesEndpoints._country_delete s **5111111111'1'9 H(
(
(
"
"
"
"
"
"
, {( ( (X (;== = = = = = = = " C ?$F
r )NNr2 )__name__
__module__ __qualname__r IM360SCOPEr bindr r. r7 r; r r DEFAULT_METHODr@ rE r
rG rI rK rP rW r: r3 r> r r r r! r! sv KEV[i00
10
V[i//J J J 0/J V[i00K K 10K V[i22+455D D 65 32D V[i00
10
V[i00.=>>K K ?> 10K V[i//.=>>J J J ?> 0/J V[i22.=>>D D ?> 32D
(
r r! )peeweer im360.subsys.webshieldsubsysrf defence360agent.rpc_toolsr r defence360agent.rpc_tools.utilsr r defence360agent.utilsr
im360.api.ipsr im360.contracts.messagesr
r im360.model.countryr r
im360.simple_rpc.resident_socketr r
RootEndpointsr! r r r r| s* * * * * * * * * * = = = = = = = = ( ' ' ' ' ' $ $ $ $ $ $ F F F F F F F F 4 4 4 4 4 4 4 4 ; ; ; ; ; ; ! ! !
-
r