sangge/tpre-python
2
1
Fork 1
Code Issues 6 Pull Requests Packages Projects 1 Activity

Merge pull request 'main' (#22) from sangge/mimajingsai:main into main

Browse Source 
Reviewed-on: dqy/mimajingsai#22
This commit is contained in:
dqy 2023-10-24 19:20:52 +08:00
commit 156c65570d
1 changed files with 5 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/node.py
View File

@ -5,6 +5,7 @@ import socket
import asyncio import asyncio
from pydantic import BaseModel from pydantic import BaseModel
from tpre import * from tpre import *
import os
@asynccontextmanager @asynccontextmanager
@ -17,7 +18,7 @@ async def lifespan(app: FastAPI):
app = FastAPI(lifespan=lifespan) app = FastAPI(lifespan=lifespan)
server_address = "http://10.20.14.232:8000/server" server_address = "http://10.20.14.232:8000/server"
id = 0 id = 0
ip = "10.16.21.163" ip = ""
client_ip_src = "" # 发送信息用户的ip client_ip_src = "" # 发送信息用户的ip
client_ip_des = "" # 接收信息用户的ip client_ip_des = "" # 接收信息用户的ip
processed_message = () # 重加密后的数据 processed_message = () # 重加密后的数据
@ -35,22 +36,14 @@ def send_ip():
id = requests.get(url) id = requests.get(url)
# 用socket获取本机ip # 用环境变量获取本机ip
def get_local_ip(): def get_local_ip():
# 创建一个套接字对象
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
# 连接到一个外部的服务器，这将自动绑定到本地IP地址
s.connect(("8.8.8.8", 80))
# 获取本地IP地址
local_ip = s.getsockname()[0]
s.close()
global ip global ip
ip = local_ip ip = os.environ.get("HOST_IP", "IP not set")
def init(): def init():
# get_local_ip() get_local_ip()
global id
send_ip() send_ip()
task = asyncio.create_task(send_heartbeat_internal()) task = asyncio.create_task(send_heartbeat_internal())
print("Finish init") print("Finish init")