Update start_jupyterlab.py

start_jupyterlab.py

@@ -1,89 +1,61 @@
-import os
-import subprocess
-import sys
-
-# 设置默认端口号
-DEFAULT_PORT = 8888
-
-# 检查 Python 环境
-def check_dependencies():
-    print("检查 Python 和 pip...")
-    try:
-        subprocess.run(["python3", "--version"], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-        subprocess.run(["pip3", "--version"], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    except subprocess.CalledProcessError:
-        print("Python3 或 pip3 未正确安装，请安装后重试。")
-        sys.exit(1)
-
-# 安装 JupyterLab
-def install_jupyterlab():
-    print("检查并安装 JupyterLab...")
-    try:
-        subprocess.run(["pip3", "show", "jupyterlab"], check=False, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    except subprocess.CalledProcessError:
-        print("安装 JupyterLab...")
-        subprocess.run(["pip3", "install", "jupyterlab"], check=True)
-        print("JupyterLab 安装成功！")
-
-# 检查端口是否被占用
-def check_port(port):
-    print(f"检查端口 {port} 是否被占用...")
-    try:
-        result = subprocess.run(
-            ["lsof", f"-i:{port}"],
-            check=False,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE,
-            text=True
-        )
-        if result.stdout:
-            print(f"端口 {port} 已被占用，请选择其他端口。")
-            sys.exit(1)
-        print(f"端口 {port} 可用。")
-    except FileNotFoundError:
-        print("lsof 命令未找到，请安装 lsof 或使用其他方法检查端口。")
-        sys.exit(1)
-
-# 启动 JupyterLab
-def start_jupyterlab(port):
-    print("启动 JupyterLab 服务...")
-    try:
-        command = [
-            "jupyter-lab",
-            "--ip=0.0.0.0",  # 接受外部连接
-            f"--port={port}",  # 指定端口
-            "--no-browser",  # 禁止自动打开浏览器
-            "--NotebookApp.token=''",  # 禁用 token，方便外部访问
-            "--NotebookApp.password=''"  # 禁用密码
-        ]
-        process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
-
-        # 检查是否启动成功
-        for line in process.stdout:
-            if "http://" in line or "https://" in line:
-                print(f"JupyterLab 启动成功！访问链接：{line.strip()}")
-                return
-
-        print("JupyterLab 启动失败，请检查日志。")
-        sys.exit(1)
-    except Exception as e:
-        print(f"启动 JupyterLab 时出错: {e}")
-        sys.exit(1)
-
-# 主函数
-def main():
-    port = DEFAULT_PORT
-    if len(sys.argv) > 1:
-        try:
-            port = int(sys.argv[1])
-        except ValueError:
-            print("无效的端口号，请输入有效的整数。")
-            sys.exit(1)
-
-    check_dependencies()
-    install_jupyterlab()
-    check_port(port)
-    start_jupyterlab(port)
-
-if __name__ == "__main__":
-    main()
+import os
+import subprocess
+import sys
+
+def modify_jupyter_config():
+    """修改 Jupyter 配置文件以允许远程访问并禁用 token"""
+    config_dir = os.path.expanduser("~/.jupyter")
+    config_file = os.path.join(config_dir, "jupyter_notebook_config.py")
+
+    # 如果配置文件不存在，生成配置文件
+    if not os.path.exists(config_file):
+        print("生成 Jupyter 配置文件...")
+        try:
+            subprocess.check_call([sys.executable, "-m", "jupyter", "notebook", "--generate-config"])
+            print(f"配置文件已生成: {config_file}")
+        except subprocess.CalledProcessError as e:
+            print(f"生成配置文件时出错: {e}")
+            return
+
+    # 检查配置文件是否已经包含了所需设置
+    with open(config_file, "r") as f:
+        config_content = f.read()
+        if "c.ServerApp.allow_remote_access" in config_content:
+            print("Jupyter 配置文件已经包含必要的设置，无需修改。")
+            return
+
+    # 修改配置文件以支持远程访问
+    try:
+        print("修改 Jupyter 配置文件...")
+        with open(config_file, "a") as f:
+            f.write("\n# Allow remote access\n")
+            f.write("c.ServerApp.allow_remote_access = True\n")
+            f.write("c.ServerApp.ip = '0.0.0.0'\n")
+            f.write("c.ServerApp.port = 8888\n")  # 设置端口
+            f.write("c.ServerApp.token = ''\n")  # 禁用 token
+            f.write("c.ServerApp.allow_origin = '*'\n")
+        print("Jupyter 配置文件修改完成！")
+    except Exception as e:
+        print(f"修改配置文件时出错: {e}")
+
+def start_jupyter_lab():
+    """启动 JupyterLab 服务"""
+    try:
+        print("启动 JupyterLab 服务...")
+        command = [
+            sys.executable, "-m", "jupyter", "lab",
+            "--no-browser",          # 禁止自动打开浏览器
+            "--ip=0.0.0.0",          # 允许外部访问
+            "--port=8888",           # 指定端口为 8888
+        ]
+        subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        print("JupyterLab 已启动，请通过 http://<服务器IP>:8888 访问。")
+    except Exception as e:
+        print(f"启动 JupyterLab 时出错: {e}")
+
+if __name__ == "__main__":
+    # 1. 修改 Jupyter 配置
+    modify_jupyter_config()
+    
+    # 2. 启动 JupyterLab
+    start_jupyter_lab()