ok
Mini Shell
�
g\ � � � d dl Z d dlmZ d dlZd dlmZmZmZmZm Z m
Z
d dl m Z d dl
mZmZ d dlmZ d dlmZ G d� d e� � Z G d
� d e� � ZdS ) � N)� Generator)�JOIN� CharField�Check� DoesNotExist�ForeignKeyField� IntegerField��
model_to_dict)�instance�Model)�apply_order_by)�geoc � � e Zd ZdZ edd�� � Z eddd�� � Z ed�� � Z G d� d � � Ze d
� � � Z
d S ) �Countryz=Contains a single record per country, with its code and name.TF�� primary_key�null� )�
max_length�uniquer �r c � � e Zd ZdZej ZdS )� Country.Meta�countryN��__name__�
__module__� __qualname__�db_tabler �db�database� � �H/opt/imunify360/venv/lib/python3.11/site-packages/im360/model/country.py�Metar s � � � � � � ���;���r$ r& c � � |D ];}| � |� � � � � � � � �<dS )zq
Load country info from iterable of dicts to database
:param countries:
:return:
N)�insert�on_conflict_replace�execute)�cls� countriesr s r% � update_fromzCountry.update_from! sN � � !� @� @�G� �J�J�w� � � 3� 3� 5� 5� =� =� ?� ?� ?� ?� @� @r$ N) r r r �__doc__r �id�code�namer& � classmethodr- r# r$ r% r r s� � � � � � �G�G� ��t�%� 0� 0� 0�B� �9��$�U� ;� ;� ;�D� �9�%� � � �D� � � � � � � � � �@� @� �[�@� @� @r$ r c � � � e Zd ZdZdZdZ eedd�� � Z e d e
d� � g�� � Z e dd � �
� � Z
e d� � � Z G d � d
� � Z G d� d� � Ze� fd�� � Ze dd�� � Zed� � � Zedd�� � Zed� � � Zedefd�� � Zedefd�� � Z� xZS ) � CountryListz+List of Countries in user WHITE/BLACK list.�BLACK�WHITETFr z listname in ('WHITE','BLACK'))r � constraintsc �B � t t j � � � � S �N)�int�timer# r$ r% �<lambda>zCountryList.<lambda>< s � �C�� � � �4D�4D� r$ )r �defaultr c � � e Zd ZdZej ZdS )�CountryList.Meta� country_listNr r# r$ r% r& r? @ s � � � � � �!���;���r$ r& c �J � e Zd Z ej � � Z ej � � ZdS )�CountryList.SignalsN)r r r �blinker�Signal�added�deletedr# r$ r% �SignalsrB D s, � � � � � � ��� � �� �'�.�"�"���r$ rG c � �� t � � j di |��}| j j � |j |j �� � |S )N��
country_idr# )�super�createrG rE �send�listnamerJ )r+ �kwargs�obj� __class__s �r% rL zCountryList.createH sE �� � �e�g�g�n�&�&�v�&�&�� � �� � �s�|��� �G�G�G��
r$ Nc �� � d }|r@t j � � 5 }|� |� � }d d d � � n # 1 swxY w Y t � t � � � � � � t t j t j
t j k �� � }|r#|� t j
|k � � }|� t j � � }|r#|� t j |k � � }|r-|� | j � |� � � � }|r#|� t j |k � � }|S )N��on)r �reader�get_coder4 �select�distinct�joinr r �INNERr r/ �whererN �order_byr0 �comment�contains)r+ �by_country_code�
by_comment�by_ip�by_list�country_code_from_ip�
geo_reader�qs r% �
_fetch_filterzCountryList._fetch_filterN s| � � $� � � B��� � �
B��'1�':�':�5�'A�'A�$�
B�
B�
B�
B�
B�
B�
B�
B�
B�
B�
B����
B�
B�
B�
B�
� � �{� +� +�
�X�Z�Z�
�T�'�4�:�;�+>�'�*�+L�T�
N�
N�
�
� 9���� �,��7�8�8�A�
�J�J�w�|� $� $�� � 9����� ��7�8�8�A� � :����� �,�,�Z�8�8�9�9�A� � >����� �(<�<�=�=�A��s
�:�>�>c �@ � | j di |��� � � S )Nr# )rf �count)r+ � filter_argss r% � fetch_countzCountryList.fetch_countj s' � � �s� �/�/�;�/�/�5�5�7�7�7r$ c � � | j di |��}|�|� |� � }|�|� |� � }|�t || |� � }d� |D � � S )Nc �, � g | ]}t |� � ��S r# r
)�.0�rows r% �
<listcomp>z%CountryList.fetch.<locals>.<listcomp>w s � �0�0�0�s�
�c�"�"�0�0�0r$ r# )rf �offset�limitr )r+ rp rq r\ ri re s r% �fetchzCountryList.fetchn sq � � �C� � ,� ,� � ,� ,�� � ����� � �A� � �������A� � � �x��a�0�0�A�0�0�a�0�0�0�0r$ c �� � | � t � � � � t j |k � � �� � j S # t $ r Y d S w xY w)N)r )�getr rW r[ r0 rN r )r+ r s r% � get_listnamezCountryList.get_listnamey sk � � ��7�7� ���(�(�.�.�w�|�w�/F�G�G� � � ��
�� � � � ��4�4� ���s �AA �
A#�"A#r c � � t � � � � t j |k t j |k z � � � � � }|r&| j j � ||j �� � |S )NrI )
r4 �deleter[ r rN r* rG rF rM r/ )r+ r rN rF s r% �delete_countryzCountryList.delete_country� s} � �
� � � � �
�U� �$��/� �'�8�3�5�� � �W�Y�Y�
� � F� �K� � $� $�X�'�*� $� E� E� E��r$ �returnc � � t � t j � � � � � � t t j t j t j k �� � �
t j |k � � � t j � � }d� |�
� � D � � S )z*Returns generator of listed country codes.rS c 3 � K � | ]\ }|V � � d S r9 r# )rm r0 s r% � <genexpr>z,CountryList.country_codes.<locals>.<genexpr>� s$ � � � �/�/��$��/�/�/�/�/�/r$ )r4 rW r r0 rX rY r rZ r r/ r[ rN r\ �tuples)r+ rN re s r% �
country_codeszCountryList.country_codes� s� � �
� � �w�|� ,� ,�
�X�Z�Z�
�T�'�4�:�;�+>�'�*�+L�T�
N�
N�
�U�;�'�8�3�
4�
4�
�X�g�l�
#�
#�
� 0�/�A�H�H�J�J�/�/�/�/r$ )NNNN)NNN)r r r r. r5 r6 r r r r r rN r �ctimer] r& rG r2 rL rf rj rr ru rx r r~ �
__classcell__)rQ s @r% r4 r4 , s� �� � � � � �5�5�
�E� �E� �o�g�4�e�D�D�D�G� �y�
���'F�!G�!G� H�� � �H�
�L�d�,D�,D� E� E� E�E��i�T�"�"�"�G� � � � � � � � �#� #� #� #� #� #� #� #� �� � � � �[��
�HL�� � � �[��6 �8� 8� �[�8� �1� 1� 1� �[�1� �� � �[�� � �W� � � � �[� � �
0� �
0�
0�
0� �[�
0�
0�
0�
0�
0r$ r4 )r; �typingr rC �peeweer r r r r r �playhouse.shortcutsr �defence360agent.modelr r
�$defence360agent.model.simplificationr �im360.internalsr r r4 r# r$ r% �<module>r� s9 �� � � � � � � � � � � ����� � � � � � � � � � � � � � � � .� -� -� -� -� -� 1� 1� 1� 1� 1� 1� 1� 1� ?� ?� ?� ?� ?� ?� � � � � � �@� @� @� @� @�e� @� @� @�0o0� o0� o0� o0� o0�%� o0� o0� o0� o0� o0r$
Zerion Mini Shell 1.0