Skip to content

Commit c33e6f1

Browse files
committed
feat: 支持设置代理服务器进行扫描
1 parent 29b4c5d commit c33e6f1

1 file changed

Lines changed: 17 additions & 0 deletions

File tree

Network/Scanner.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,20 @@
11
# -*- coding: UTF-8 -*-
2+
import socket
23
from base64 import b64decode
34
from copy import deepcopy
45
from io import BytesIO
56
from queue import Queue, Empty
67
from threading import Thread, Lock
78
from time import sleep
89
from typing import List, Any, Dict, Union
10+
11+
import ping3
912
from PIL import Image, ImageTk
1013

1114
from Libs import Vars
15+
import socks
16+
17+
from Libs.Vars import user_settings_loader
1218

1319
ServerPinger: Any = None
1420
Address: Any = None
@@ -162,6 +168,17 @@ def __init__(self, host: str = "", port: int = 25565, timeout: float = 0.7, prot
162168
self.port = port
163169
self.protocol = protocol_version
164170
self.timeout = timeout
171+
self.set_proxy()
172+
173+
@staticmethod
174+
def set_proxy():
175+
if user_settings_loader.configs["use_proxy"]:
176+
kind_map = {"sock5": socks.SOCKS5,
177+
"sock4": socks.SOCKS4}
178+
socks.set_default_proxy(kind_map[user_settings_loader.configs["proxy_kind"]],
179+
user_settings_loader.configs["proxy_host"], user_settings_loader.configs["proxy_port"])
180+
from mcstatus.protocol import connection
181+
connection.socket.socket = socks.socksocket
165182

166183
def get_server_info(self) -> dict:
167184
"""

0 commit comments

Comments
 (0)