Hopiu/django-defender
1
0
Fork 0
mirror of https://github.com/jazzband/django-defender.git synced 2026-03-16 22:10:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove_prefix method working same for all python versions

Browse source
This commit is contained in:
Dashgin 2022-06-30 16:27:36 +04:00 committed by Ken Cochrane
parent d90dfa8db7
commit a4b3f9f332
2 changed files with 6 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
defender/tests.py
View file

@ -1176,3 +1176,6 @@ class TestUtils(DefenderTestCase):
"defender:blocked:ip:192.168.24.24", "defender:blocked:"), "ip:192.168.24.24")
self.assertEqual(utils.remove_prefix(
"defender:blocked:username:johndoe", "defender:blocked:"), "username:johndoe")
self.assertEqual(utils.remove_prefix(
"defender:blocked:username:johndoe", "blocked:username:"),
"defender:blocked:username:johndoe")

10
defender/utils.py
View file

@ -129,14 +129,10 @@ def get_username_blocked_cache_key(username):
def remove_prefix(string, prefix):
if string.startswith(prefix):
return string[len(prefix):]
return string
# backwards compatibility for str.removeprefix for python < 3.9
if sys.version_info < (3, 9):
if string.startswith(prefix):
return string[len(prefix) :]
return string
return string.removeprefix(prefix)
def strip_keys(key_list):