Bläddra i källkod

feat: hidden api key enhancement. (#1468)

Garfield Dai 1 år sedan
förälder
incheckning
c1e7193c4b
1 ändrade filer med 7 tillägg och 1 borttagningar
  1. 7 1
      api/fields/api_based_extension_fields.py

+ 7 - 1
api/fields/api_based_extension_fields.py

@@ -5,7 +5,13 @@ from libs.helper import TimestampField
 
 class HiddenAPIKey(fields.Raw):
     def output(self, key, obj):
-        return obj.api_key[:3] + '***' + obj.api_key[-3:]
+        api_key = obj.api_key
+        # If the length of the api_key is less than 8 characters, show the first and last characters
+        if len(api_key) <= 8:
+            return api_key[0] + '******' + api_key[-1]
+        # If the api_key is greater than 8 characters, show the first three and the last three characters
+        else:
+            return api_key[:3] + '******' + api_key[-3:]
 
 
 api_based_extension_fields = {