python - 如果!期间发生键错误，如何重新启动和运行所有代码!在Google Colab中安装pip？

Question

在Google Colab中，每次运行时安装并导入以下代码时:

! pip install --upgrade snowflake-connector-python
! pip install pyspark
! pip install snowflake-connector-python asn1crypto==0.24.0
! pip install azure-storage==0.34.3
! pip install azure.storage.common

from pyspark.sql import SparkSession,SQLContext
import pyspark
import snowflake.connector as con
from pyspark.sql import SQLContext
sc = pyspark.SparkContext.getOrCreate()
sqlContext = SQLContext(sc)

我收到此错误:

KeyError: 'snowflake-connector-python'

我已经与Snowflake进行了交谈，他们说这很正常，我只需要再次单击Restart＆Run All。
我想知道是否有一种方法可以在代码内自动重新启动并运行所有程序，从而无需手动执行此操作？我正在考虑一个while循环，但是我对安装过程中的错误处理不熟悉。
我知道我可以将pip安装保存到我的云端硬盘中，但是由于这些文件将充当共享仪表板，因此我不希望同事这样做。
我已经阅读了一些有关如何重新启动内核的信息，但没有再次运行整个文件，例如:

import os
os._exit(00)

Answer 1

这是我能做的最好的事情:

尝试导入软件包

如果失败(KeyError，ImportError等)，请pip安装它并强制内核重新启动。

try:
  import snowflake.connector as con
except (ImportError, KeyError, ModuleNotFoundError):
  !pip install snowflake-connector-python 
  print('Stopping RUNTIME. Colaboratory will restart automatically. Please run again.')
  exit()

基于对Google Colab - How to 'restart runtime' using python code or command line interface?的多个答案。