Fix admin search

safe-global · Mar 22, 2024 · 8e64353 · 8e64353
1 parent 97497f1
commit 8e64353
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/safe_transaction_service/account_abstraction/admin.py b/safe_transaction_service/account_abstraction/admin.py
@@ -2,18 +2,21 @@
 
 from eth_typing import ChecksumAddress, HexStr
 
+from gnosis.eth.django.admin import AdvancedAdminSearchMixin
+
 from safe_transaction_service.history import models as history_models
 
 from .models import SafeOperation, UserOperation, UserOperationReceipt
 
 
 @admin.register(UserOperation)
-class UserOperationAdmin(admin.ModelAdmin):
+class UserOperationAdmin(AdvancedAdminSearchMixin, admin.ModelAdmin):
     list_display = ("hash", "ethereum_tx", "sender", "nonce", "success")
     list_filter = [
         "receipt__success",
     ]
     search_fields = [
+        "==hash",
         "==sender",
     ]
     ordering = ["-nonce"]
@@ -23,7 +26,7 @@ def success(self, obj: UserOperation) -> bool:
         return obj.receipt.success
 
 
-@admin.register(UserOperationReceipt)
+@admin.register(AdvancedAdminSearchMixin, UserOperationReceipt)
 class UserOperationReceiptAdmin(admin.ModelAdmin):
     list_display = (
         "user_operation_hash",
@@ -36,6 +39,7 @@ class UserOperationReceiptAdmin(admin.ModelAdmin):
         "success",
     ]
     search_fields = [
+        "==user_operation__hash",
         "==user_operation__sender",
     ]
 
@@ -66,6 +70,8 @@ class SafeOperationAdmin(admin.ModelAdmin):
     list_filter = ["module_address"]
     list_select_related = ["user_operation__receipt"]
     search_fields = [
+        "==hash",
+        "==user_operation__hash",
         "==user_operation__sender",
     ]
     ordering = ["-modified"]