mirror of https://github.com/jumpserver/jumpserver
perf: 修改随机数生成,避免使用 random 库
parent
28196573bb
commit
2aed3fcaea
|
@ -18,6 +18,20 @@ def random_ip():
|
||||||
return socket.inet_ntoa(struct.pack('>I', random.randint(1, 0xffffffff)))
|
return socket.inet_ntoa(struct.pack('>I', random.randint(1, 0xffffffff)))
|
||||||
|
|
||||||
|
|
||||||
|
def random_replace_char(s, chars, length):
|
||||||
|
using_index = set()
|
||||||
|
seq = list(s)
|
||||||
|
|
||||||
|
while length > 0:
|
||||||
|
index = secrets.randbelow(len(seq) - 1)
|
||||||
|
if index in using_index or index == 0:
|
||||||
|
continue
|
||||||
|
seq[index] = secrets.choice(chars)
|
||||||
|
using_index.add(index)
|
||||||
|
length -= 1
|
||||||
|
return ''.join(seq)
|
||||||
|
|
||||||
|
|
||||||
def random_string(length: int, lower=True, upper=True, digit=True, special_char=False, symbols=string_punctuation):
|
def random_string(length: int, lower=True, upper=True, digit=True, special_char=False, symbols=string_punctuation):
|
||||||
if not any([lower, upper, digit]):
|
if not any([lower, upper, digit]):
|
||||||
raise ValueError('At least one of `lower`, `upper`, `digit` must be `True`')
|
raise ValueError('At least one of `lower`, `upper`, `digit` must be `True`')
|
||||||
|
@ -31,12 +45,10 @@ def random_string(length: int, lower=True, upper=True, digit=True, special_char=
|
||||||
)
|
)
|
||||||
chars = ''.join([i[1] for i in chars_map if i[0]])
|
chars = ''.join([i[1] for i in chars_map if i[0]])
|
||||||
texts = list(secrets.choice(chars) for __ in range(length))
|
texts = list(secrets.choice(chars) for __ in range(length))
|
||||||
|
texts = ''.join(texts)
|
||||||
|
|
||||||
|
# 控制一下特殊字符的数量, 别随机出来太多
|
||||||
if special_char:
|
if special_char:
|
||||||
symbol_num = length // 16 + 1
|
symbol_num = length // 16 + 1
|
||||||
symbol_index = random.choices(list(range(1, length - 1)), k=symbol_num)
|
texts = random_replace_char(texts, symbols, symbol_num)
|
||||||
for i in symbol_index:
|
return texts
|
||||||
texts[i] = secrets.choice(symbols)
|
|
||||||
|
|
||||||
text = ''.join(texts)
|
|
||||||
return text
|
|
||||||
|
|
Loading…
Reference in New Issue