From d09eb3c4fafe880fcc01c668dd26215800c66efe Mon Sep 17 00:00:00 2001
From: feng <1304903146@qq.com>
Date: Fri, 20 Sep 2024 18:03:43 +0800
Subject: [PATCH] perf: Lock username is not case sensitive

---
 apps/users/utils.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/apps/users/utils.py b/apps/users/utils.py
index cdb15eb23..14b36a875 100644
--- a/apps/users/utils.py
+++ b/apps/users/utils.py
@@ -110,6 +110,7 @@ class BlockUtil:
     BLOCK_KEY_TMPL: str
 
     def __init__(self, username):
+        username = username.lower()
         self.block_key = self.BLOCK_KEY_TMPL.format(username)
         self.key_ttl = int(settings.SECURITY_LOGIN_LIMIT_TIME) * 60
 
@@ -125,6 +126,7 @@ class BlockUtilBase:
     BLOCK_KEY_TMPL: str
 
     def __init__(self, username, ip):
+        username = username.lower()
         self.username = username
         self.ip = ip
         self.limit_key = self.LIMIT_KEY_TMPL.format(username, ip)
@@ -158,6 +160,7 @@ class BlockUtilBase:
 
     @classmethod
     def unblock_user(cls, username):
+        username = username.lower()
         key_limit = cls.LIMIT_KEY_TMPL.format(username, '*')
         key_block = cls.BLOCK_KEY_TMPL.format(username)
         # Redis 尽量不要用通配
@@ -166,6 +169,7 @@ class BlockUtilBase:
 
     @classmethod
     def is_user_block(cls, username):
+        username = username.lower()
         block_key = cls.BLOCK_KEY_TMPL.format(username)
         return bool(cache.get(block_key))