Your IP : 216.73.216.213
�jvXn � @ s� d Z d d l m Z y d d l Z Wn e k
r@ d Z Yn Xd d l Z d d l m Z m Z d d l m
Z
m Z m Z m
Z
d d l m Z d d l m Z d d l m Z Gd d
� d
e � Z Gd d � d e � Z d S)
zo
maxminddb.reader
~~~~~~~~~~~~~~~~
This module contains the pure Python database reader and related classes.
� )�unicode_literalsN)�
byte_from_int�compat_ip_address)� MODE_AUTO� MODE_MMAP� MODE_FILE� MODE_MEMORY)�Decoder)�InvalidDatabaseError)�
FileBufferc @ s� e Z d Z d Z d Z d Z d Z e d d � Z d d � Z d d
� Z
d d � Z d
d � Z d d � Z
d d � Z d d � Z d S)�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 k r t s | t k rp t | d � �= } t j | j � d d t j �| _ | j j � | _ Wd QRXn� | t t f k r� t
| � | _ | j j � | _ n^ | t k r� t | d � �( } | j � | _ t
| j � | _ Wd QRXn t d j | � � � | j j | j t d | j d
� � } | d k rW| j � t d j | � � � | t
| j � 7} t | j | � } | j | � \ } } t | � | _ t | j | j j | j � | _ 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.
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.
�rbr �accessNznUnsupported open mode ({0}). Only MODE_AUTO, MODE_FILE, and MODE_MEMORY are support by the pure Python Reader� i � zCError opening database file ({0}). Is this a valid MaxMind DB file?i ���) r �mmapr �open�filenoZ ACCESS_READ�_buffer�size� _buffer_sizer r r �read�len�
ValueError�format�rfind�_METADATA_START_MARKER�max�closer
r �decode�Metadata� _metadata�search_tree_size� _DATA_SECTION_SEPARATOR_SIZE�_decoder)�selfZdatabase�modeZdb_fileZmetadata_startZmetadata_decoder�metadata�_� r+ �C/opt/alt/python35/lib64/python3.5/site-packages/maxminddb/reader.py�__init__$ s8
zReader.__init__c C s | j S)z7Return the metadata associated with the MaxMind DB file)r# )r' r+ r+ r, r) U s zReader.metadatac C sh t | � } | j d k rB | j j d k rB t d j | � � � | j | � } | rd | j | � Sd S)z�Return the record for the ip_address in the MaxMind DB
Arguments:
ip_address -- an IP address in the standard string notation
� � zXError looking up {0}. You attempted to look up an IPv6 address in an IPv4-only database.N)r �versionr# �
ip_versionr r �_find_address_in_tree�_resolve_data_pointer)r' �
ip_address�address�pointerr+ r+ r, �getY s ! z
Reader.getc C s� t | j � } t | � d } | j | � } xW t | � D]I } | | j j k rT Pd | | d ?d | d ?@} | j | | � } q; W| | j j k r� d S| | j j k r� | St d � � d S)N� r � � r zInvalid node in search tree) � bytearray�packedr � _start_node�ranger# �
node_count�
_read_noder
)r' r4 r< Z bit_count�node�i�bitr+ r+ r, r2 k s z Reader._find_address_in_treec C s� | j j d k s | d k r" d S| j r2 | j Sd } x9 t d � D]+ } | | j j k r^ P| j | d � } qE W| | _ | S)Nr. r r �` )r# r1 � _ipv4_startr> r? r@ )r' �lengthrA r* r+ r+ r, r= ~ s zReader._start_nodec C s: | | j j } | j j } | d k rT | | d } d | j | | d � } n� | d k r� t j d | j | d | d � � \ } | r� | d M} n d | @d ?} | | d } t | � | j | | d � } nI | d k r| | d } | j | | d � } n t d
j | � � � t j d | � d S)
N� r9 s � s !Br/ � �� � zUnknown record size: {0}s !Ir ) r# �node_byte_size� record_sizer �struct�unpackr r
r )r' Z node_number�indexZ base_offsetrM �offsetZ
node_bytesZmiddler+ r+ r, r@ � s( $
zReader._read_nodec C sQ | | j j | j j } | | j k r5 t d � � | j j | � \ } } | S)Nz,The MaxMind DB file's search tree is corrupt)r# r? r$ r r
r& r! )r' r6 Zresolved�datar* r+ r+ r, r3 � s
z Reader._resolve_data_pointerc C s, t | j � t t f k r( | j j � d S)zBCloses the MaxMind DB file and returns the resources to the systemN)�typer �str�bytesr )r' r+ r+ r, r � s z Reader.close)�__name__�
__module__� __qualname__�__doc__r% r rE r r- r) r7 r2 r= r@ r3 r r+ r+ r+ r, r s 1 r c @ sR e Z d Z d Z d d � Z e d d � � Z e d d � � Z d d � Z d
S) r" z"Metadata for the MaxMind DB readerc K sy | d | _ | d | _ | d | _ | d | _ | d | _ | d | _ | d | _ | d | _ | d | _ d
S) zACreates new Metadata object. kwargs are key/value pairs from specr? rM r1 �
database_type� languages�binary_format_major_version�binary_format_minor_version� build_epoch� descriptionN) r? rM r1 rZ r[ r\ r] r^ r_ )r' �kwargsr+ r+ r, r- � s
zMetadata.__init__c C s | j d S)zThe size of a node in bytesr/ )rM )r' r+ r+ r, rL � s zMetadata.node_byte_sizec C s | j | j S)zThe size of the search tree)r? rL )r' r+ r+ r, r$ � s zMetadata.search_tree_sizec C sJ d j d d � | j j � D� � } d j d | j d | j j d | � S)Nz, c s s | ] } d | Vq d S)z%s=%rNr+ )�.0�xr+ r+ r, � <genexpr>� s z$Metadata.__repr__.<locals>.<genexpr>z {module}.{class_name}({data})�module�
class_namerR )�join�__dict__�itemsr rW � __class__rV )r' �argsr+ r+ r, �__repr__� s
% zMetadata.__repr__N) rV rW rX rY r- �propertyrL r$ rk r+ r+ r+ r, r" � s
r" )rY �
__future__r r � ImportErrorrN Zmaxminddb.compatr r Zmaxminddb.constr r r r Zmaxminddb.decoderr Zmaxminddb.errorsr
Zmaxminddb.filer �objectr r" r+ r+ r+ r, �<module> s
"�