U Ha#@sddlZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z eeddeed d eed d d ZGdddeZGdddZdS)N)settings)ValidationError)validate_ipv46_address)to_path)CityCountry GEOIP_PATH GEOIP_CITYzGeoLite2-City.mmdb GEOIP_COUNTRYzGeoLite2-Country.mmdb)r r r c@s eZdZdS)GeoIP2ExceptionN)__name__ __module__ __qualname__rrB/tmp/pip-unpacked-wheel-3jxiddxt/django/contrib/gis/geoip2/base.pyr sr c@seZdZdZdZdZdZdZeeeeeefZ dZ dZ dZ dZ d*dd Zed d Zed d ZddZddZd+ddZddZddZddZddZd,ddZd d!Zd"d#Zd$d%Zed&d'Zed(d)ZdS)-GeoIP2rrNc Cs8||jkr||_n td||p(td}|s6tdt|}|r||pRtd}|rztjj t ||d|_ ||_ ||ptd}|rtjj t ||d|_ ||_|jstd|nr|r,tjj t ||d}|j}|dr||_ ||_n&|d r||_ ||_ n td |ntd d S) aI Initialize the GeoIP object. No parameters are required to use default settings. Keyword arguments may be passed in to customize the locations of the GeoIP datasets. * path: Base directory to where GeoIP data is located or the full path to where the city or country data files (*.mmdb) are located. Assumes that both the city and country data sets are located in this directory; overrides the GEOIP_PATH setting. * cache: The cache settings when opening up the GeoIP datasets. May be an integer in (0, 1, 2, 4, 8) corresponding to the MODE_AUTO, MODE_MMAP_EXT, MODE_MMAP, MODE_FILE, and MODE_MEMORY, `GeoIPOptions` C API settings, respectively. Defaults to 0, meaning MODE_AUTO. * country: The name of the GeoIP country data file. Defaults to 'GeoLite2-Country.mmdb'; overrides the GEOIP_COUNTRY setting. * city: The name of the GeoIP city data file. Defaults to 'GeoLite2-City.mmdb'; overrides the GEOIP_CITY setting. z Invalid GeoIP caching option: %sr zDGeoIP path must be provided via parameter or the GEOIP_PATH setting.r )moder z"Could not load a database from %s.rrz(Unable to recognize database edition: %sz-GeoIP path must be a valid file or directory.N) cache_options_cacher GEOIP_SETTINGSris_diris_filegeoip2ZdatabaseReaderstr_country _country_file_city _city_file_readermetadataZ database_typeendswith) selfpathcachecountrycityZ country_dbZcity_dbreaderZdb_typerrr__init__.s:       zGeoIP2.__init__cCs |jp |jSN)r r"r'rrrr$tszGeoIP2._readercCs|jr|jjS|jjSdSr.)r r*r"r+r/rrr_country_or_cityxszGeoIP2._country_or_citycCs|jr|jdSr.)r$closer/rrr__del__szGeoIP2.__del__cCs4|j}d|j|jf}d|jj||j|jdS)Nz[v%s.%s]zH<%(cls)s %(version)s _country_file="%(country)s", _city_file="%(city)s">)clsversionr*r+)r$r%binary_format_major_versionbinary_format_minor_version __class__r r!r#)r'metar4rrr__repr__s zGeoIP2.__repr__FcCst|tstdt|j|r6|js6|js6tdn2|rP|jsPtd|jn|rh|jshtd|j z t |Wnt k rt |}YnX|S)z*Check the query and database availability.z)GeoIP query must be a string, not type %sz*Invalid GeoIP country and city data files.z#Invalid GeoIP country data file: %sz Invalid GeoIP city data file: %s) isinstancer TypeErrortyper r r"r r!r#rrsocket gethostbyname)r'queryr*r+city_or_countryrrr _check_querys     zGeoIP2._check_querycCs|j|dd}t|j|S)z Return a dictionary of city information for the given IP address or Fully Qualified Domain Name (FQDN). Some information in the dictionary may be undefined (None). T)r+)rArr"r+r'r?Z enc_queryrrrr+sz GeoIP2.citycCs|j|dd}||dS)z9Return the country code for the given IP Address or FQDN.Tr@ country_coderAr*rBrrrrDszGeoIP2.country_codecCs|j|dd}||dS)z9Return the country name for the given IP Address or FQDN.TrC country_namerErBrrrrFszGeoIP2.country_namecCs|j|dd}t||S)z Return a dictionary with the country code and name when given an IP address or a Fully Qualified Domain Name (FQDN). For example, both '24.124.1.80' and 'djangoproject.com' are valid parameters. TrC)rArr0rBrrrr*szGeoIP2.country longitudelatitudecs0||dkrdStfdd|DSdS)Nc3s|]}|VqdSr.r).0oZcdictrr sz GeoIP2.coords..)r+tuple)r'r?ZorderingrrLrcoordss z GeoIP2.coordscCs ||S)z@Return a tuple of the (longitude, latitude) for the given query.rOr'r?rrrlon_latszGeoIP2.lon_latcCs ||dS)z@Return a tuple of the (latitude, longitude) for the given query.)rIrHrPrQrrrlat_lonszGeoIP2.lat_loncCs.||}|r&ddlm}||ddSdSdS)z/Return a GEOS Point object for the given query.r)Pointi)ZsridN)rRZdjango.contrib.gis.geosrT)r'r?ZllrTrrrgeoss    z GeoIP2.geoscCs|j}d|j|jfS)z@Return information about the GeoIP library and databases in use.zGeoIP Library: %s.%s )r$r%r5r6)r'r8rrrinfos z GeoIP2.infocCs t||Sr.)r)r3 full_pathr)rrropensz GeoIP2.open)NrNN)FFF)rG) r rrZ MODE_AUTOZ MODE_MMAP_EXTZ MODE_MMAPZ MODE_FILEZ MODE_MEMORY frozensetrr#r!r"r r-propertyr$r0r2r9rAr+rDrFr*rOrRrSrUrV classmethodrXrrrrrs< F       r)r=Zgeoip2.databaserZ django.confrZdjango.core.exceptionsrZdjango.core.validatorsrZdjango.utils._osr resourcesrrgetattrr Exceptionr rrrrrs