mathiasertl / django-hashers-passlib Goto Github PK

from django.utils.translation import gettext_noop as _

# This file is a patch for django-hashers-passlib==0.4 and passlib==1.7.4 with Django 4.2. 
# These packages have not been updated for a long time and do not support Django 4.2.

def apply_patches():
    # Patch for passlib.ext.django.utils
    try:
        from passlib.ext.django.utils import _PasslibHasherWrapper

        # Patched safe_summary method to use the patched _ function instead of the original ugettext_noop
        def patched_safe_summary(self, encoded):
            from collections import OrderedDict
            from django.contrib.auth.hashers import mask_hash
            handler = self.passlib_handler
            items = [
                (_('algorithm'), handler.name),
            ]
            if hasattr(handler, "parsehash"):
                kwds = handler.parsehash(encoded, sanitize=mask_hash)
                for key, value in kwds.items():
                    key = self._translate_kwds.get(key, key)
                    items.append((_(key), value))
            return OrderedDict(items)

        _PasslibHasherWrapper.safe_summary = patched_safe_summary
    except ImportError as e:
        # Handle the case where passlib is not installed
        print(f"Failed to apply patch for passlib: {e}")

    # Patch for hashers_passlib
    try:
        # Check if ugettext_noop already exists, if not, add it
        from django.utils.translation import ugettext_noop
    except ImportError:
        # Dynamically add ugettext_noop pointing to gettext_noop in the django.utils.translation module
        import django.utils.translation
        django.utils.translation.ugettext_noop = _

        # Now safely import hashers_passlib since ugettext_noop has been added
        import hashers_passlib

        # Since we've successfully imported hashers_passlib, this indicates the patch has taken effect,
        # If necessary, original state can be restored here
        # However, typically, if ugettext_noop is not directly used elsewhere in the Django project, restoration may not be needed

    except Exception as e:
        print(f"Unexpected error while applying patch for hashers_passlib: {e}")

from django.apps import AppConfig

class PasslibPatchConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'passlib_patch'

    def ready(self):
        from .patches import apply_patches
        apply_patches()

from . import patches

default_app_config = 'passlib_patch.apps.PasslibPatchConfig'

from django.test import TestCase
from django.contrib.auth import get_user_model
from passlib.hash import django_pbkdf2_sha256 as handler
from wp_user.patches import apply_patches

apply_patches()

class PasslibHashersTestCase(TestCase):
    def test_passlib_functionality(self):
        hashed = handler.hash("s3cr3t")
        self.assertTrue(handler.verify("s3cr3t", hashed), "Password verification failed!")

    def test_hashers_passlib_integration(self):
        User = get_user_model()
        user = User(username='testuser')
        user.set_password('my_password')
        user.save()

        self.assertTrue(user.check_password('my_password'), "User password verification failed!")

When creating your own hasher, you need to override algorithm,
verify(), encode() and safe_summary().

def safe_summary(self, encoded):
        from django.contrib.auth.hashers import mask_hash, _, SortedDict
        algorithm, salt, hash = encoded.split('$', 2)
        assert algorithm == self.algorithm
        return SortedDict([
            (_('algorithm'), algorithm),
            #(_('salt'), mask_hash(salt, show=2)),
            (_('hash'), mask_hash(hash)),
        ])