瀏覽代碼

feat: 添加pip更新检查

SongZihuan 3 年之前
父節點
當前提交
d1afc894aa
共有 1 個文件被更改,包括 9 次插入4 次删除
  1. 9 4
      setup.py

+ 9 - 4
setup.py

@@ -11,7 +11,11 @@ except ImportError:
     print("检查结束, 未找到pip")
     print("检查结束, 未找到pip")
     exit(1)
     exit(1)
 else:
 else:
-    print("依赖 ['pip'] 存在")
+    print("依赖 pip 存在")
+    if os.system(f"{sys.executable} -m pip install --upgrade pip") != 0:
+        print(f"依赖 pip 更新失败")
+    else:
+        print(f"依赖 pip 更新成功")
 
 
 __setup = os.path.dirname(os.path.abspath(__file__))
 __setup = os.path.dirname(os.path.abspath(__file__))
 
 
@@ -24,15 +28,14 @@ def check_import(packages: Union[str, List[str]], pips: Union[str, List[str]]):
     try:
     try:
         for package in packages:
         for package in packages:
             __import__(package)
             __import__(package)
+            print(f"依赖 {package} 存在")
     except ImportError:
     except ImportError:
         for pip in pips:
         for pip in pips:
             if os.system(f"{sys.executable} -m pip install {pip}") != 0:
             if os.system(f"{sys.executable} -m pip install {pip}") != 0:
                 print(f"{pip} 依赖安装失败")
                 print(f"{pip} 依赖安装失败")
                 exit(1)
                 exit(1)
             else:
             else:
-                print(f"依赖 {packages}:{pip} 按照成功")
-    else:
-        print(f"依赖 {packages}:{pips} 存在")
+                print(f"依赖 {packages}:{pip} 安装成功")
 
 
 
 
 check_import("cv2", "opencv-python")  # 图像处理
 check_import("cv2", "opencv-python")  # 图像处理
@@ -41,6 +44,8 @@ check_import("pymysql", "PyMySQL")  # 连接 MySQL服务器
 check_import("cryptography", "cryptography")  # 链接 MySQL 服务器时加密
 check_import("cryptography", "cryptography")  # 链接 MySQL 服务器时加密
 check_import("flask", "Flask")  # 网页服务
 check_import("flask", "Flask")  # 网页服务
 check_import("PIL", "Pillow")  # 图片处理
 check_import("PIL", "Pillow")  # 图片处理
+check_import("numpy", "numpy")  # matplotlib依赖
+check_import("matplotlib", "matplotlib")  # matplotlib依赖
 
 
 check_import(["oss2", "viapi", "aliyunsdkcore", "aliyunsdkimagerecog"],
 check_import(["oss2", "viapi", "aliyunsdkcore", "aliyunsdkimagerecog"],
              ["oss2", "aliyun-python-sdk-viapiutils", "viapi-utils", "aliyun-python-sdk-imagerecog"])  # 阿里云依赖
              ["oss2", "aliyun-python-sdk-viapiutils", "viapi-utils", "aliyun-python-sdk-imagerecog"])  # 阿里云依赖