120 lines
3.1 KiB
Python
120 lines
3.1 KiB
Python
import secrets
|
||
import hashlib
|
||
import time
|
||
import os
|
||
|
||
def generate_hashed_session_id():
|
||
"""
|
||
生成一个基于 SHA-256 的安全 session_id。
|
||
:return: 安全的 session_id。
|
||
"""
|
||
random_bytes = secrets.token_bytes(32) # 生成 32 字节随机数
|
||
hashed = hashlib.sha256(random_bytes).hexdigest() # 计算 SHA-256 哈希值
|
||
#保存session_id
|
||
timestamp = int(time.time())
|
||
with open(f"/tmp/k3gpt/session_id_{hashed}","w+") as f:
|
||
f.write(str(timestamp))
|
||
return hashed
|
||
|
||
def valid_session_id(session_id):
|
||
"""
|
||
检查会话id是否有效
|
||
"""
|
||
try:
|
||
with open(f"/tmp/k3gpt/session_id_{session_id}","r") as f:
|
||
s_timestamp = f.read()
|
||
if int(time.time())-int(s_timestamp) < 3600*12:
|
||
return True
|
||
else:
|
||
return False
|
||
except Exception as e:
|
||
print(e)
|
||
return False
|
||
|
||
|
||
def delete_session_id(session_id):
|
||
"""
|
||
删除session
|
||
"""
|
||
try:
|
||
os.remove(f"/tmp/k3gpt/session_id_{session_id}")
|
||
except:
|
||
pass
|
||
|
||
def hashed_password(password):
|
||
"""
|
||
生成一个基于 SHA-256 的安全 session_id。
|
||
:return: 安全的 session_id。
|
||
"""
|
||
pswd=f"k3gpt_{password}".encode("utf8")
|
||
hashed = hashlib.sha256(pswd).hexdigest() # 计算 SHA-256 哈希值
|
||
return hashed
|
||
|
||
|
||
def hashed_tel(password):
|
||
"""
|
||
生成一个基于 SHA-256 的安全 session_id。
|
||
:return: 安全的 session_id。
|
||
"""
|
||
pswd=f"tel_{password}".encode("utf8")
|
||
hashed = hashlib.sha256(pswd).hexdigest() # 计算 SHA-256 哈希值
|
||
return hashed
|
||
|
||
import re
|
||
|
||
def regular_filename(filename):
|
||
"""
|
||
去除字符串中的中文符号、英文符号、空格、换行、制表符等所有特殊符号
|
||
仅保留：中文字符、英文字母、数字
|
||
"""
|
||
# \u4e00-\u9fa5：常用中文字符范围（简体+繁体基本覆盖）
|
||
# a-zA-Z0-9：英文字母和数字
|
||
cleaned = re.sub(r'[^\u4e00-\u9fa5a-zA-Z0-9.]', '_', filename)
|
||
return cleaned
|
||
|
||
|
||
|
||
#处理思考大模型的信息
|
||
def found_llm_think_data(data):
|
||
think=""
|
||
if data.find("<think>")>=0 and data.find("</think>")==-1:
|
||
#还未思考结束
|
||
think = data[8:]
|
||
data = ""
|
||
elif data.find("<think>")>=0 and data.find("</think>")>0:
|
||
#思考结束
|
||
end = data.find("</think>")
|
||
think = data[8:end]
|
||
data = data[end+8:]
|
||
|
||
|
||
begin = data.find("{")
|
||
end = data.rfind("}")
|
||
if data.find("```json") >=0:
|
||
#找到json数据，只返回json数据
|
||
data = data[begin:end+1]
|
||
elif begin>0 and end >0:
|
||
#含有json数据
|
||
data = data[begin:end+1]
|
||
return think,data
|
||
|
||
from cryptography.fernet import Fernet
|
||
|
||
key = b'aDSK382C_Ep5hAvMkW0UPttdicyog-JoxaV_-CStutE='
|
||
|
||
def encrypt_tel(tel):
|
||
f = Fernet(key)
|
||
token = f.encrypt(tel.encode("utf-8"))
|
||
return token.decode("utf-8")
|
||
|
||
def decrypt_tel(token):
|
||
f = Fernet(key)
|
||
try:
|
||
tel = f.decrypt(token.encode("utf-8"))
|
||
except:
|
||
return "***********"
|
||
return tel.decode("utf-8")
|
||
|
||
if __name__=="__main__":
|
||
h = hashed_password("12345678")
|
||
print(h) |