o
g> @ s4 d dl Zd dlZd dlZdd Zdd Zdd ZdS ) Nc
C sl z
| | }|j jW S tjjy Y dS ty Y dS ty5 } z d| W Y d }~S d }~ww )NUnknown
Invalid IPzError: )countrynamegeoip2errorsZAddressNotFoundError
ValueError Exception)ipreaderresponsee r Z C:\xampp\htdocs\Programación de servicios y procesos\jocarsa-ivory-main\funciones\pais.py get_country s
r c
C s: d}d}t j|r$t|d }| }W d n1 s w Y ng }t }d}d}d} t|D ]4\}
} | |krBd}|
}q3| |krNd}|
} n|rg| drg| dd } |
| q3t| | }
|
svt d dS g }|dur| dur|
D ] } | d
| d q|d| | || d }t d t
|
d
n2|d ddg}|
D ] } | d
| d q| d | |d |d g | }t dt
|
d z%t|d
}|| W d n1 sw Y t d| d W dS ty } zt d| d| W Y d}~dS d}~ww )z
Adds the blocked IPs to the .htaccess file within specific markers using Apache 2.4+ syntax.
Only adds new IPs that are not already present to prevent duplication.
z # BEGIN Blocked IPs by Countryz # END Blocked IPs by CountryrNFTzRequire not ipz-No new IPs to add to the country-based block.z Require not ip
zAdding z; new IPs to the existing country-based blocked IPs section.z
z Require all granted
z
z6Creating a new country-based blocked IPs section with z IPs.wzSuccessfully updated z! with new blocked IPs by country.zError writing to : )ospathexistsopen readlinesset enumeratestrip
startswithsplitaddprintappendlen
writelinesr )blacklisted_ips
htaccess_pathZ start_markerZ
end_markerflinesZ existing_ipsZ within_blockZblock_start_indexZblock_end_indexindexliner
Znew_ipsZ block_rulesZ new_linesr
r r r block_ips_htaccess sh
"r+ c
C s d} d}g d}d}zDt j|rM|d }t|d$}t|d}|| W d n1 s2w Y W d n1 sAw Y td| W n tyh } ztd | W Y d }~d S d }~ww zt| d }| } W d n1 s~w Y W n+ t y td
| d Y d S ty } ztd | W Y d }~d S d }~ww i }
| D ]-} z!|
d
d } t
| | |
vrd|
| < n|
| d7 < W q t jy Y qw ztj|}
W n, t y td| d Y d S ty } ztd| W Y d }~d S d }~ww i }g }i }|
D ]4\} }t| |
}|dvrO||v r:||| < q!||vrE|||< q!|| |7 < q!|| q!|
tt| dd dd}tt| dd dd}td | D ]\}}t| d| q|td | D ]\} }t| d| qtdt| z*td d}|D ] } || d qW d n 1 sw Y td W n ty } z
td | W Y d }~nd }~ww z1td d }| D ]\} }|| d!| d qW d n 1 sw Y td" W n ty; } z
td#| W Y d }~nd }~ww |rHt| | d S td$ d S )%Nz/var/log/apache2/access.logzGeoLite2-Country.mmdb)ZChinaZUkraineZ Singaporez/var/www/html/.htaccessz.backup_countryr r z Backup of .htaccess created at z$Error creating backup of .htaccess: zError: The file z does not exist.z*An error occurred while reading the file: -r z"Error: The GeoLite2 database file z7An error occurred while opening the GeoLite2 database: )r r Errorc S | d S Nr- r itemr r r zmain..T)keyreversec S r/ r0 r r1 r r r r3 r4 zDesired Countries:r z
Blacklisted Countries:z
Unknown/Invalid: zunknown_ips_country.logr z-Unknown IPs logged to unknown_ips_country.logz#Error writing unknown IPs to file: zblacklisted_ips_country.logz
- Count: z@Blacklisted IPs by country logged to blacklisted_ips_country.logz'Error writing blacklisted IPs to file: z'No blacklisted IPs by country to block.)r r r r writereadr! r r FileNotFoundErrorr r ipaddressZ IPv4AddressZAddressValueErrorr ZdatabaseReaderitemsr r" closedictsortedr# r+ keys)Z
log_file_pathZ mmdb_pathZnon_desired_countriesr& Z backup_pathZoriginalZbackupr
ZarchivoZlineasZdiccionario_ipsZlinear
r Zdiccionario_paisesZ unknown_ipsr% countr Zordenado_paisesZordenado_blacklistedr' r r r mainY s
rB )Zgeoip2.databaser r: r r r+ rB r r r r s E