146 lines
5.1 KiB
Python
146 lines
5.1 KiB
Python
from http.server import ThreadingHTTPServer, SimpleHTTPRequestHandler
|
||
import os
|
||
import urllib.parse
|
||
import socket
|
||
|
||
class APIHandler(SimpleHTTPRequestHandler):
|
||
# 设置超时，避免长时间占用连接
|
||
timeout = 30
|
||
|
||
# MIME 类型映射
|
||
MIME_TYPES = {
|
||
'.html': 'text/html; charset=utf-8',
|
||
'.css': 'text/css',
|
||
'.js': 'application/javascript',
|
||
'.json': 'application/json',
|
||
'.png': 'image/png',
|
||
'.jpg': 'image/jpeg',
|
||
'.jpeg': 'image/jpeg',
|
||
'.gif': 'image/gif',
|
||
'.svg': 'image/svg+xml',
|
||
'.ico': 'image/x-icon',
|
||
'.mp3': 'audio/mpeg',
|
||
'.wav': 'audio/wav',
|
||
'.mp4': 'video/mp4',
|
||
'.txt': 'text/plain; charset=utf-8',
|
||
}
|
||
|
||
def log_message(self, format, *args):
|
||
# 自定义日志格式
|
||
print(f"[{self.log_date_time_string()}] {self.address_string()} - {format % args}")
|
||
|
||
def do_GET(self):
|
||
try:
|
||
# 解析路径和查询参数
|
||
parsed_path = urllib.parse.urlparse(self.path)
|
||
path = parsed_path.path
|
||
query = parsed_path.query
|
||
|
||
# 检查是否是API接口
|
||
api_param = None
|
||
if path.startswith('/api/'):
|
||
api_param = path[5:] # 提取 /api/ 后面的数字
|
||
|
||
# 生成带参数的index.html URL
|
||
if api_param:
|
||
# 使用HTML文件并附加api参数
|
||
self.serve_file('index.html', query=f'api={api_param}')
|
||
elif path == '/' or path == '/index.html':
|
||
# 默认访问使用 api=1
|
||
self.serve_file('index.html', query='api=1')
|
||
elif path == '/coords' or path == '/coordinate.html':
|
||
self.serve_file('coordinate.html')
|
||
else:
|
||
# 处理静态文件请求
|
||
# 移除开头的 /
|
||
filename = path.lstrip('/')
|
||
if os.path.exists(filename):
|
||
self.serve_static_file(filename)
|
||
else:
|
||
self.send_error(404, 'Not Found')
|
||
except Exception as e:
|
||
print(f"Error handling request: {e}")
|
||
self.send_error(500, 'Internal Server Error')
|
||
|
||
def serve_file(self, filename, query=None):
|
||
try:
|
||
file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), filename)
|
||
if os.path.exists(file_path):
|
||
self.send_response(200)
|
||
self.send_header('Content-type', 'text/html; charset=utf-8')
|
||
self.end_headers()
|
||
|
||
with open(file_path, 'rb') as f:
|
||
content = f.read()
|
||
|
||
# 如果有查询参数，修改HTML中的URL参数解析部分
|
||
if query:
|
||
try:
|
||
content = content.decode('utf-8')
|
||
# 修改URL参数行，替换默认值为实际参数
|
||
content = content.replace(
|
||
"const apiParam = urlParams.get('api') || '1';",
|
||
f"const apiParam = '{query.split('=')[1]}';"
|
||
)
|
||
content = content.encode('utf-8')
|
||
except Exception as e:
|
||
print(f"Error modifying HTML content: {e}")
|
||
|
||
self.wfile.write(content)
|
||
else:
|
||
self.send_error(404, f'{filename} not found')
|
||
except Exception as e:
|
||
print(f"Error serving file: {e}")
|
||
raise
|
||
|
||
def serve_static_file(self, filename):
|
||
"""提供静态文件服务"""
|
||
try:
|
||
# 获取文件扩展名
|
||
ext = os.path.splitext(filename)[1].lower()
|
||
content_type = self.MIME_TYPES.get(ext, 'application/octet-stream')
|
||
|
||
self.send_response(200)
|
||
self.send_header('Content-type', content_type)
|
||
self.send_header('Access-Control-Allow-Origin', '*')
|
||
self.end_headers()
|
||
|
||
with open(filename, 'rb') as f:
|
||
content = f.read()
|
||
|
||
self.wfile.write(content)
|
||
except Exception as e:
|
||
print(f"Error serving static file: {e}")
|
||
self.send_error(500, 'Internal Server Error')
|
||
|
||
def check_port_available(port):
|
||
"""检查端口是否可用"""
|
||
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
||
try:
|
||
s.bind(('', port))
|
||
return True
|
||
except socket.error:
|
||
return False
|
||
|
||
def run():
|
||
port = 8086
|
||
if not check_port_available(port):
|
||
print(f"错误: 端口 {port} 已被占用")
|
||
return
|
||
|
||
server_address = ('', port)
|
||
httpd = ThreadingHTTPServer(server_address, APIHandler)
|
||
print(f'Server running on http://localhost:{port}')
|
||
print(f'支持的接口: /, /api/1, /api/2, /api/3, /api/4, /api/5, /api/6, /api/7, /api/11-16')
|
||
print(f'坐标提取工具: /coords')
|
||
print('按 Ctrl+C 停止服务器')
|
||
|
||
try:
|
||
httpd.serve_forever()
|
||
except KeyboardInterrupt:
|
||
print('\n服务器已停止')
|
||
httpd.server_close()
|
||
|
||
if __name__ == '__main__':
|
||
run()
|