django-nativemojo 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

mojo/helpers/crypto/__init__.py

@@ -1,4 +1,4 @@
 from .aes import encrypt, decrypt
-from .utils import random_bytes, random_string
+from .utils import random_bytes, random_string, b64_encode, b64_decode
 from .hash import hash
 from .sign import generate_signature as sign, verify_signature as verify

mojo/helpers/crypto/utils.py

@@ -1,5 +1,7 @@
 from Crypto.Random import get_random_bytes
 import string
+from base64 import b64encode, b64decode
+import json
 
 
 def generate_key(bit_size=128):
@@ -24,3 +26,16 @@ def random_string(length, allow_digits=True, allow_chars=True, allow_special=Tru
         raise ValueError("At least one character set (digits, chars, special) must be allowed")
     random_bytes = get_random_bytes(length)
     return ''.join(characters[b % len(characters)] for b in random_bytes)
+
+
+def b64_encode(data):
+    if isinstance(data, dict):
+        data = json.dumps(data)
+    return b64encode(data.encode('utf-8')).decode('utf-8')
+
+
+def b64_decode(data):
+    dec = b64decode(data.encode('utf-8')).decode('utf-8')
+    if dec[0] == '{':
+        return json.loads(dec)
+    return dec

mojo/helpers/location/__init__.py

@@ -0,0 +1,2 @@
+# This file marks the 'location' directory as a Python package.
+from .geolocation import geolocate_ip

mojo/helpers/location/countries.py

@@ -0,0 +1,262 @@
+# -*- coding: utf-8 -*-
+"""
+Provides a simple lookup for converting ISO 3166-1 alpha-2 country codes to country names.
+"""
+
+COUNTRY_CODES = {
+    'AF': 'Afghanistan',
+    'AX': 'Åland Islands',
+    'AL': 'Albania',
+    'DZ': 'Algeria',
+    'AS': 'American Samoa',
+    'AD': 'Andorra',
+    'AO': 'Angola',
+    'AI': 'Anguilla',
+    'AQ': 'Antarctica',
+    'AG': 'Antigua and Barbuda',
+    'AR': 'Argentina',
+    'AM': 'Armenia',
+    'AW': 'Aruba',
+    'AU': 'Australia',
+    'AT': 'Austria',
+    'AZ': 'Azerbaijan',
+    'BS': 'Bahamas',
+    'BH': 'Bahrain',
+    'BD': 'Bangladesh',
+    'BB': 'Barbados',
+    'BY': 'Belarus',
+    'BE': 'Belgium',
+    'BZ': 'Belize',
+    'BJ': 'Benin',
+    'BM': 'Bermuda',
+    'BT': 'Bhutan',
+    'BO': 'Bolivia, Plurinational State of',
+    'BQ': 'Bonaire, Sint Eustatius and Saba',
+    'BA': 'Bosnia and Herzegovina',
+    'BW': 'Botswana',
+    'BV': 'Bouvet Island',
+    'BR': 'Brazil',
+    'IO': 'British Indian Ocean Territory',
+    'BN': 'Brunei Darussalam',
+    'BG': 'Bulgaria',
+    'BF': 'Burkina Faso',
+    'BI': 'Burundi',
+    'CV': 'Cabo Verde',
+    'KH': 'Cambodia',
+    'CM': 'Cameroon',
+    'CA': 'Canada',
+    'KY': 'Cayman Islands',
+    'CF': 'Central African Republic',
+    'TD': 'Chad',
+    'CL': 'Chile',
+    'CN': 'China',
+    'CX': 'Christmas Island',
+    'CC': 'Cocos (Keeling) Islands',
+    'CO': 'Colombia',
+    'KM': 'Comoros',
+    'CG': 'Congo',
+    'CD': 'Congo, the Democratic Republic of the',
+    'CK': 'Cook Islands',
+    'CR': 'Costa Rica',
+    'CI': "Côte d'Ivoire",
+    'HR': 'Croatia',
+    'CU': 'Cuba',
+    'CW': 'Curaçao',
+    'CY': 'Cyprus',
+    'CZ': 'Czechia',
+    'DK': 'Denmark',
+    'DJ': 'Djibouti',
+    'DM': 'Dominica',
+    'DO': 'Dominican Republic',
+    'EC': 'Ecuador',
+    'EG': 'Egypt',
+    'SV': 'El Salvador',
+    'GQ': 'Equatorial Guinea',
+    'ER': 'Eritrea',
+    'EE': 'Estonia',
+    'SZ': 'Eswatini',
+    'ET': 'Ethiopia',
+    'FK': 'Falkland Islands (Malvinas)',
+    'FO': 'Faroe Islands',
+    'FJ': 'Fiji',
+    'FI': 'Finland',
+    'FR': 'France',
+    'GF': 'French Guiana',
+    'PF': 'French Polynesia',
+    'TF': 'French Southern Territories',
+    'GA': 'Gabon',
+    'GM': 'Gambia',
+    'GE': 'Georgia',
+    'DE': 'Germany',
+    'GH': 'Ghana',
+    'GI': 'Gibraltar',
+    'GR': 'Greece',
+    'GL': 'Greenland',
+    'GD': 'Grenada',
+    'GP': 'Guadeloupe',
+    'GU': 'Guam',
+    'GT': 'Guatemala',
+    'GG': 'Guernsey',
+    'GN': 'Guinea',
+    'GW': 'Guinea-Bissau',
+    'GY': 'Guyana',
+    'HT': 'Haiti',
+    'HM': 'Heard Island and McDonald Islands',
+    'VA': 'Holy See',
+    'HN': 'Honduras',
+    'HK': 'Hong Kong',
+    'HU': 'Hungary',
+    'IS': 'Iceland',
+    'IN': 'India',
+    'ID': 'Indonesia',
+    'IR': 'Iran, Islamic Republic of',
+    'IQ': 'Iraq',
+    'IE': 'Ireland',
+    'IM': 'Isle of Man',
+    'IL': 'Israel',
+    'IT': 'Italy',
+    'JM': 'Jamaica',
+    'JP': 'Japan',
+    'JE': 'Jersey',
+    'JO': 'Jordan',
+    'KZ': 'Kazakhstan',
+    'KE': 'Kenya',
+    'KI': 'Kiribati',
+    'KP': "Korea, Democratic People's Republic of",
+    'KR': 'Korea, Republic of',
+    'KW': 'Kuwait',
+    'KG': 'Kyrgyzstan',
+    'LA': "Lao People's Democratic Republic",
+    'LV': 'Latvia',
+    'LB': 'Lebanon',
+    'LS': 'Lesotho',
+    'LR': 'Liberia',
+    'LY': 'Libya',
+    'LI': 'Liechtenstein',
+    'LT': 'Lithuania',
+    'LU': 'Luxembourg',
+    'MO': 'Macao',
+    'MG': 'Madagascar',
+    'MW': 'Malawi',
+    'MY': 'Malaysia',
+    'MV': 'Maldives',
+    'ML': 'Mali',
+    'MT': 'Malta',
+    'MH': 'Marshall Islands',
+    'MQ': 'Martinique',
+    'MR': 'Mauritania',
+    'MU': 'Mauritius',
+    'YT': 'Mayotte',
+    'MX': 'Mexico',
+    'FM': 'Micronesia, Federated States of',
+    'MD': 'Moldova, Republic of',
+    'MC': 'Monaco',
+    'MN': 'Mongolia',
+    'ME': 'Montenegro',
+    'MS': 'Montserrat',
+    'MA': 'Morocco',
+    'MZ': 'Mozambique',
+    'MM': 'Myanmar',
+    'NA': 'Namibia',
+    'NR': 'Nauru',
+    'NP': 'Nepal',
+    'NL': 'Netherlands',
+    'NC': 'New Caledonia',
+    'NZ': 'New Zealand',
+    'NI': 'Nicaragua',
+    'NE': 'Niger',
+    'NG': 'Nigeria',
+    'NU': 'Niue',
+    'NF': 'Norfolk Island',
+    'MK': 'North Macedonia',
+    'MP': 'Northern Mariana Islands',
+    'NO': 'Norway',
+    'OM': 'Oman',
+    'PK': 'Pakistan',
+    'PW': 'Palau',
+    'PS': 'Palestine, State of',
+    'PA': 'Panama',
+    'PG': 'Papua New Guinea',
+    'PY': 'Paraguay',
+    'PE': 'Peru',
+    'PH': 'Philippines',
+    'PN': 'Pitcairn',
+    'PL': 'Poland',
+    'PT': 'Portugal',
+    'PR': 'Puerto Rico',
+    'QA': 'Qatar',
+    'RE': 'Réunion',
+    'RO': 'Romania',
+    'RU': 'Russian Federation',
+    'RW': 'Rwanda',
+    'BL': 'Saint Barthélemy',
+    'SH': 'Saint Helena, Ascension and Tristan da Cunha',
+    'KN': 'Saint Kitts and Nevis',
+    'LC': 'Saint Lucia',
+    'MF': 'Saint Martin (French part)',
+    'PM': 'Saint Pierre and Miquelon',
+    'VC': 'Saint Vincent and the Grenadines',
+    'WS': 'Samoa',
+    'SM': 'San Marino',
+    'ST': 'Sao Tome and Principe',
+    'SA': 'Saudi Arabia',
+    'SN': 'Senegal',
+    'RS': 'Serbia',
+    'SC': 'Seychelles',
+    'SL': 'Sierra Leone',
+    'SG': 'Singapore',
+    'SX': 'Sint Maarten (Dutch part)',
+    'SK': 'Slovakia',
+    'SI': 'Slovenia',
+    'SB': 'Solomon Islands',
+    'SO': 'Somalia',
+    'ZA': 'South Africa',
+    'GS': 'South Georgia and the South Sandwich Islands',
+    'SS': 'South Sudan',
+    'ES': 'Spain',
+    'LK': 'Sri Lanka',
+    'SD': 'Sudan',
+    'SR': 'Suriname',
+    'SJ': 'Svalbard and Jan Mayen',
+    'SE': 'Sweden',
+    'CH': 'Switzerland',
+    'SY': 'Syrian Arab Republic',
+    'TW': 'Taiwan, Province of China',
+    'TJ': 'Tajikistan',
+    'TZ': 'Tanzania, United Republic of',
+    'TH': 'Thailand',
+    'TL': 'Timor-Leste',
+    'TG': 'Togo',
+    'TK': 'Tokelau',
+    'TO': 'Tonga',
+    'TT': 'Trinidad and Tobago',
+    'TN': 'Tunisia',
+    'TR': 'Turkey',
+    'TM': 'Turkmenistan',
+    'TC': 'Turks and Caicos Islands',
+    'TV': 'Tuvalu',
+    'UG': 'Uganda',
+    'UA': 'Ukraine',
+    'AE': 'United Arab Emirates',
+    'GB': 'United Kingdom of Great Britain and Northern Ireland',
+    'US': 'United States of America',
+    'UM': 'United States Minor Outlying Islands',
+    'UY': 'Uruguay',
+    'UZ': 'Uzbekistan',
+    'VU': 'Vanuatu',
+    'VE': 'Venezuela, Bolivarian Republic of',
+    'VN': 'Viet Nam',
+    'VG': 'Virgin Islands, British',
+    'VI': 'Virgin Islands, U.S.',
+    'WF': 'Wallis and Futuna',
+    'EH': 'Western Sahara',
+    'YE': 'Yemen',
+    'ZM': 'Zambia',
+    'ZW': 'Zimbabwe',
+}
+
+def get_country_name(code):
+    """Returns the full country name for a given alpha-2 code."""
+    if not code or not isinstance(code, str):
+        return None
+    return COUNTRY_CODES.get(code.upper(), code)

mojo/helpers/location/geolocation.py

@@ -0,0 +1,196 @@
+import requests
+import ipaddress
+import random
+from mojo.helpers.settings import settings
+from .countries import get_country_name
+
+# Lazy-load model to avoid circular imports
+_GeoLocatedIP = None
+GEOLOCATION_PROVIDERS = settings.get('GEOLOCATION_PROVIDERS', ['ipinfo'])
+
+
+def get_geo_located_ip_model():
+    global _GeoLocatedIP
+    if _GeoLocatedIP is None:
+        from mojo.apps.account.models.device import GeoLocatedIP
+        _GeoLocatedIP = GeoLocatedIP
+    return _GeoLocatedIP
+
+
+def geolocate_ip(ip_address):
+    """
+    Fetches geolocation data for a given IP address. It handles both
+    public IPs (by calling an external provider) and private IPs.
+    Returns a normalized dictionary of geolocation data.
+    """
+    # 1. Handle private/reserved IPs
+    try:
+        ip_obj = ipaddress.ip_address(ip_address)
+        if ip_obj.is_private or ip_obj.is_reserved:
+            return {
+                'provider': 'internal',
+                'country_name': 'Private Network',
+                'region': 'Private' if ip_obj.is_private else 'Reserved',
+            }
+    except ValueError:
+        return None  # Invalid IP
+
+    # 2. Handle public IPs by dispatching to a randomly selected provider
+    providers = GEOLOCATION_PROVIDERS
+    provider = random.choice(providers)
+    api_key_setting_name = f'GEOLOCATION_API_KEY_{provider.upper()}'
+    api_key = getattr(settings, api_key_setting_name, None)
+
+    provider_map = {
+        'ipinfo': fetch_from_ipinfo,
+        'ipstack': fetch_from_ipstack,
+        'ip-api': fetch_from_ipapi,
+        'maxmind': fetch_from_maxmind,
+    }
+
+    fetch_function = provider_map.get(provider)
+
+    if fetch_function:
+        return fetch_function(ip_address, api_key)
+    else:
+        # In a real app, you might want to log this or handle it differently
+        print(f"[Geolocation Error] Provider '{provider}' is not supported.")
+        return None
+
+
+def refresh_geolocation_for_ip(ip_address):
+    """
+    This function is the entry point for the background task.
+    It gets or creates a GeoLocatedIP record and refreshes it if necessary.
+    """
+    GeoLocatedIP = get_geo_located_ip_model()
+
+    # Get or create the record, then call its internal refresh logic.
+    geo_record, created = GeoLocatedIP.objects.get_or_create(ip_address=ip_address)
+
+    if created or geo_record.is_expired:
+        geo_record.refresh()
+
+    return geo_record
+
+
+def fetch_from_ipinfo(ip_address, api_key):
+    """
+    Fetches geolocation data from the ipinfo.io API and normalizes it.
+    Fails gracefully by returning None if any error occurs.
+    """
+    try:
+        url = f"https://ipinfo.io/{ip_address}"
+        if api_key:
+            url += f"?token={api_key}"
+
+        response = requests.get(url, timeout=5)
+        response.raise_for_status()  # Raise an exception for bad status codes (4xx or 5xx)
+        data = response.json()
+
+        # Normalize the data to our model's schema
+        loc_parts = data.get('loc', '').split(',')
+        latitude = float(loc_parts[0]) if len(loc_parts) == 2 else None
+        longitude = float(loc_parts[1]) if len(loc_parts) == 2 else None
+        country_code = data.get('country')
+
+        return {
+            'provider': 'ipinfo',
+            'country_code': country_code,
+            'country_name': get_country_name(country_code),
+            'region': data.get('region'),
+            'city': data.get('city'),
+            'postal_code': data.get('postal'),
+            'latitude': latitude,
+            'longitude': longitude,
+            'timezone': data.get('timezone'),
+            'data': data  # Store the raw response
+        }
+
+    except Exception as e:
+        # In a real application, you would want to log this error.
+        print(f"[Geolocation Error] Failed to fetch from ipinfo.io for IP {ip_address}: {e}")
+        return None
+
+
+def fetch_from_ipstack(ip_address, api_key):
+    """
+    Fetches geolocation data from the ipstack.com API and normalizes it.
+    """
+    if not api_key:
+        print("[Geolocation Error] ipstack provider requires an API key (GEOLOCATION_API_KEY_IPSTACK).")
+        return None
+    try:
+        url = f"http://api.ipstack.com/{ip_address}?access_key={api_key}"
+        response = requests.get(url, timeout=5)
+        response.raise_for_status()
+        data = response.json()
+
+        if data.get('success') is False:
+            error_info = data.get('error', {}).get('info', 'Unknown error')
+            print(f"[Geolocation Error] ipstack API error: {error_info}")
+            return None
+
+        country_code = data.get('country_code')
+        return {
+            'provider': 'ipstack',
+            'country_code': country_code,
+            'country_name': data.get('country_name') or get_country_name(country_code),
+            'region': data.get('region_name'),
+            'city': data.get('city'),
+            'postal_code': data.get('zip'),
+            'latitude': data.get('latitude'),
+            'longitude': data.get('longitude'),
+            'data': data
+        }
+    except Exception as e:
+        print(f"[Geolocation Error] Failed to fetch from ipstack.com for IP {ip_address}: {e}")
+        return None
+
+
+def fetch_from_ipapi(ip_address, api_key=None):
+    """
+    Fetches geolocation data from the ip-api.com API and normalizes it.
+    Note: The free tier does not require an API key.
+    """
+    try:
+        url = f"http://ip-api.com/json/{ip_address}"
+        response = requests.get(url, timeout=5)
+        response.raise_for_status()
+        data = response.json()
+
+        if data.get('status') == 'fail':
+            error_info = data.get('message', 'Unknown error')
+            print(f"[Geolocation Error] ip-api.com API error: {error_info}")
+            return None
+
+        country_code = data.get('countryCode')
+        return {
+            'provider': 'ip-api',
+            'country_code': country_code,
+            'country_name': data.get('country') or get_country_name(country_code),
+            'region': data.get('regionName'),
+            'city': data.get('city'),
+            'postal_code': data.get('zip'),
+            'latitude': data.get('lat'),
+            'longitude': data.get('lon'),
+            'timezone': data.get('timezone'),
+            'data': data
+        }
+    except Exception as e:
+        print(f"[Geolocation Error] Failed to fetch from ip-api.com for IP {ip_address}: {e}")
+        return None
+
+
+def fetch_from_maxmind(ip_address, api_key):
+    """
+    Placeholder for MaxMind GeoIP2 web service integration.
+    """
+    # MaxMind's GeoIP2 web services are best accessed via their official client library.
+    # See: https://github.com/maxmind/geoip2-python
+    # This is a placeholder for where you would integrate the geoip2.webservice.Client.
+    # You would typically fetch account_id and license_key from settings here instead of a single api_key.
+    raise NotImplementedError(
+        "MaxMind provider requires the 'geoip2' client library. "
+        "Set GEOLOCATION_API_KEY_MAXMIND_ACCOUNT_ID and GEOLOCATION_API_KEY_MAXMIND_LICENSE_KEY in your settings."
+    )

mojo/helpers/logit.py

@@ -48,6 +48,40 @@ def color_print(msg, color, end="\n"):
     ConsoleLogger.print_message(msg, color, end)
 
 
+# Convenience logging functions with automatic logger routing
+_mojo_logger = None
+_debug_logger = None
+_error_logger = None
+
+def info(*args):
+    """Log info messages to mojo.log"""
+    global _mojo_logger
+    if _mojo_logger is None:
+        _mojo_logger = get_logger("mojo", "mojo.log")
+    _mojo_logger.info(*args)
+
+def warn(*args):
+    """Log warning messages to mojo.log"""
+    global _mojo_logger
+    if _mojo_logger is None:
+        _mojo_logger = get_logger("mojo", "mojo.log")
+    _mojo_logger.warning(*args)
+
+def debug(*args):
+    """Log debug messages to debug.log"""
+    global _debug_logger
+    if _debug_logger is None:
+        _debug_logger = get_logger("debug", "debug.log", debug=True)
+    _debug_logger.info(*args)
+
+def error(*args):
+    """Log error messages to error.log"""
+    global _error_logger
+    if _error_logger is None:
+        _error_logger = get_logger("error", "error.log")
+    _error_logger.error(*args)
+
+
 # Mask sensitive data in the log
 def mask_sensitive_data(text):
     sensitive_patterns = [
@@ -142,6 +176,9 @@ class Logger:
     def warning(self, *args):
         self.logger.warning(self._build_log(*args))
 
+    def warn(self, *args):
+        self.logger.warning(self._build_log(*args))
+
     def error(self, *args):
         self.logger.error(self._build_log(*args))
 

mojo/helpers/redis/__init__.py

@@ -0,0 +1,2 @@
+from .client import get_connection
+from .adapter import RedisAdapter, reset_adapter, get_adapter