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

Merge branch 'main' of https://git.mamahaha.work/dqy/mimajingsai

Browse Source
This commit is contained in:
dqy 2023-10-24 19:23:36 +08:00
commit 569c6b1a77
1 changed files with 4 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
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
@ -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")