Current File : //opt/alt/python38/lib64/python3.8/site-packages/maxminddb/__pycache__/reader.cpython-38.pyc
U
��^j( � @ s� d Z ddlmZ z ddlZW n ek
r4 dZY nX ddlZddlmZmZ ddl m
Z
m Z m Z m
Z
mZ ddlmZ ddlmZ ddlmZ G d d
� d
e�ZG d d � d e�ZdS )
zo
maxminddb.reader
~~~~~~~~~~~~~~~~
This module contains the pure Python database reader and related classes.
� )�unicode_literalsN)�compat_ip_address� string_type)� MODE_AUTO� MODE_MMAP� MODE_FILE� MODE_MEMORY�MODE_FD)�Decoder)�InvalidDatabaseError)�
FileBufferc @ sx e Zd ZdZdZdZdZefdd�Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� ZdS )�Readerz�
Instances of this class provide a reader for the MaxMind DB format. IP
addresses can be looked up using the ``get`` method.
� s ���MaxMind.comNc C s| |t kr ts|tkrTt|d��*}tj|�� dtjd�| _| j�� | _W 5 Q R X |}n�|t t fkr|t
|�| _| j�� | _|}nn|t kr�t|d�� }|� � | _t
| j�| _W 5 Q R X |}n4|tkr�|� � | _t
| j�| _|j}ntd�|���| j�| jtd| jd ��}|dk�r&| �� td�|���|t
| j�7 }t| j|�}|�|�\}}tf |�| _t| j| jj| j �| _ d| _ d S )
a� Reader for the MaxMind DB file format
Arguments:
database -- A path to a valid MaxMind DB file such as a GeoIP2 database
file, or a file descriptor in the case of MODE_FD.
mode -- mode to open the database with. Valid mode are:
* MODE_MMAP - read from memory map.
* MODE_FILE - read database as standard file.
* MODE_MEMORY - load database into memory.
* MODE_AUTO - tries MODE_MMAP and then MODE_FILE. Default.
* MODE_FD - the param passed via database is a file descriptor, not
a path. This mode implies MODE_MEMORY.
�rbr )�accesszwUnsupported open mode ({0}). Only MODE_AUTO, MODE_FILE, MODE_MEMORY and MODE_FD are supported by the pure Python Readeri ���zCError opening database file ({0}). Is this a valid MaxMind DB file?FN) r �mmapr �open�filenoZ ACCESS_READ�_buffer�size� _buffer_sizer r r �read�lenr �name�
ValueError�format�rfind�_METADATA_START_MARKER�max�closer r
�decode�Metadata� _metadata�search_tree_size� _DATA_SECTION_SEPARATOR_SIZE�_decoder�closed) �selfZdatabase�modeZdb_file�filenameZmetadata_startZmetadata_decoder�metadata�_� r- �
/reader.py�__init__$ sV
�� �
�� �zReader.__init__c C s | j S )z7Return the metadata associated with the MaxMind DB file)r# �r( r- r- r. r+ b s zReader.metadatac C s | � |�\}}|S )z�Return the record for the ip_address in the MaxMind DB
Arguments:
ip_address -- an IP address in the standard string notation
)�get_with_prefix_len)r( �
ip_address�recordr, r- r- r. �getf s z
Reader.getc C s� t |t�rt|�}n|}zt|j�}W n tk
rB td��Y nX |jdkrh| jj dkrht
d� |���| � |�\}}|r�| �
|�|fS d|fS )z�Return a tuple with the record and the associated prefix length
Arguments:
ip_address -- an IP address in the standard string notation
z/argument 1 must be a string or ipaddress object� � zXError looking up {0}. You attempted to look up an IPv6 address in an IPv4-only database.N)�
isinstancer r � bytearray�packed�AttributeError� TypeError�versionr# �
ip_versionr r �_find_address_in_tree�_resolve_data_pointer)r( r2 ZaddressZpacked_address�pointer�
prefix_lenr- r- r. r1 p s"
��zReader.get_with_prefix_lenc C s� t |�d }| �|�}| jj}d}||k rd||k rdd||d? d|d ? @ }| �||�}|d }q"||krtd|fS ||kr�||fS td��d S )N� r � � � zInvalid node in search tree)r � _start_noder# �
node_count�
_read_noder )r( r9 Z bit_count�noderG �i�bitr- r- r. r> � s
z Reader._find_address_in_treec C s\ | j jdks|dkrdS | jr$| jS d}td�D ] }|| j jkrD qR| �|d�}q0|| _|S )Nr5 � r �` )r# r= � _ipv4_start�rangerG rH )r( ZlengthrI r, r- r- r. rF � s zReader._start_nodec C s� || j j }| j j}|dkr@||d }d| j||d � }n�|dkr�|d| }t| j||d � �}|r�d|d @ |d< q�d|�� @ d? }|�d|� n6|d kr�||d }| j||d � }ntd
�|���t �
d |�d S ) N� rD � � r6 � r �� � zUnknown record size: {0}s !I) r# �node_byte_size� record_sizer r8 �pop�insertr r �struct�unpack)r( Z node_number�indexZ base_offsetrW �offsetZ
node_bytesZmiddler- r- r. rH � s" zReader._read_nodec C s: || j j | j j }|| jkr&td��| j�|�\}}|S )Nz,The MaxMind DB file's search tree is corrupt)r# rG r$ r r r& r! )r( r@ Zresolved�datar, r- r- r. r? � s
z Reader._resolve_data_pointerc C s& t | j�ttfkr | j�� d| _dS )zBCloses the MaxMind DB file and returns the resources to the systemTN)�typer �str�bytesr r' r0 r- r- r. r � s
z Reader.closec G s | � � d S )N)r �r( �argsr- r- r. �__exit__� s zReader.__exit__c C s | j rtd��| S )Nz%Attempt to reopen a closed MaxMind DB)r' r r0 r- r- r. � __enter__� s zReader.__enter__)�__name__�
__module__� __qualname__�__doc__r% r rN r r/ r+ r4 r1 r> rF rH r? r rd re r- r- r- r. r
s >
r
c @ s8 e Zd ZdZdd� Zedd� �Zedd� �Zdd � Zd
S ) r" a� Metadata for the MaxMind DB reader
.. attribute:: binary_format_major_version
The major version number of the binary format used when creating the
database.
:type: int
.. attribute:: binary_format_minor_version
The minor version number of the binary format used when creating the
database.
:type: int
.. attribute:: build_epoch
The Unix epoch for the build time of the database.
:type: int
.. attribute:: database_type
A string identifying the database type, e.g., "GeoIP2-City".
:type: str
.. attribute:: description
A map from locales to text descriptions of the database.
:type: dict(str, str)
.. attribute:: ip_version
The IP version of the data in a database. A value of "4" means the
database only supports IPv4. A database with a value of "6" may support
both IPv4 and IPv6 lookups.
:type: int
.. attribute:: languages
A list of locale codes supported by the databse.
:type: list(str)
.. attribute:: node_count
The number of nodes in the database.
:type: int
.. attribute:: record_size
The bit size of a record in the search tree.
:type: int
c K s^ |d | _ |d | _|d | _|d | _|d | _|d | _|d | _|d | _|d | _d
S ) zACreates new Metadata object. kwargs are key/value pairs from specrG rW r= �
database_type� languages�binary_format_major_version�binary_format_minor_version� build_epoch� descriptionN) rG rW r= rj rk rl rm rn ro )r( �kwargsr- r- r. r/ ! s
zMetadata.__init__c C s
| j d S )z8The size of a node in bytes
:type: int
r6 )rW r0 r- r- r. rV / s zMetadata.node_byte_sizec C s | j | j S )z8The size of the search tree
:type: int
)rG rV r0 r- r- r. r$ 7 s zMetadata.search_tree_sizec C s0 d� dd� | j�� D ��}dj| j| jj|d�S )Nz, c s s | ]}d | V qdS )z%s=%rNr- )�.0�xr- r- r. � <genexpr>@ s z$Metadata.__repr__.<locals>.<genexpr>z {module}.{class_name}({data}))�module�
class_namer^ )�join�__dict__�itemsr rg � __class__rf rb r- r- r. �__repr__? s �zMetadata.__repr__N) rf rg rh ri r/ �propertyrV r$ rz r- r- r- r. r" � s @
r" )ri Z
__future__r r � ImportErrorrZ Zmaxminddb.compatr r Zmaxminddb.constr r r r r Zmaxminddb.decoderr
Zmaxminddb.errorsr Zmaxminddb.filer �objectr
r" r- r- r- r. �<module> s
H
Mini Shell