Add nbconvert by default

This commit is contained in:
Andras Schmelczer 2022-06-25 18:33:17 +02:00
parent af6eed2714
commit e2795876bb
2 changed files with 8 additions and 11 deletions

View file

@ -41,6 +41,7 @@ install_requires =
fastapi >= 0.70.0
plotly >= 5.8.0
dash >= 2.4.0
nbconvert >= 6.5.0
uvicorn[standard] >= 0.18.0
watchdog >= 2.1.0
pymongo >= 4.0.0

View file

@ -116,18 +116,14 @@ def main() -> None:
def get_script_name(file_name_argument: str) -> str:
if file_name_argument.endswith(".ipynb"):
logger.info("Converting notebook to Python script")
try:
from nbconvert import PythonExporter
from nbconvert import PythonExporter
exporter = PythonExporter()
content, _ = exporter.from_filename(file_name_argument)
file_name_argument = get_script_name_of_notebook(file_name_argument)
with open(file_name_argument, "w", encoding="utf-8") as f:
f.write(content)
except ImportError:
raise ImportError(
"Install `nbconvert` to be able to use Jupyter notebook files or use a regular Python file instead"
)
exporter = PythonExporter()
content, _ = exporter.from_filename(file_name_argument)
file_name_argument = get_script_name_of_notebook(file_name_argument)
with open(file_name_argument, "w", encoding="utf-8") as f:
f.write(content)
return re.sub(r"\.(py|ipynb)$", "", file_name_argument)