QChatGPT/pkg/utils/pkgmgr.py

35 lines
630 B
Python
Raw Normal View History

2023-01-16 21:34:35 +08:00
from pip._internal import main as pipmain
import main
2023-01-16 21:34:35 +08:00
def install(package):
pipmain(['install', package])
main.reset_logging()
2023-01-16 21:34:35 +08:00
def install_requirements(file):
pipmain(['install', '-r', file, "--upgrade"])
main.reset_logging()
2023-01-16 21:34:35 +08:00
def ensure_dulwich():
# 尝试三次
for i in range(3):
try:
import dulwich
return
except ImportError:
install('dulwich')
raise ImportError("无法自动安装dulwich库")
if __name__ == "__main__":
try:
install("openai11")
except Exception as e:
print(111)
print(e)
print(222)