| import os
|
| import sys
|
| import json
|
| import urllib.parse
|
| import subprocess
|
| import threading
|
| import asyncio
|
| from aiohttp import web
|
| import folder_paths
|
| import struct |
| from .utils import require_filename, resolve_within |
|
|
| async def api_get_notebooks(request):
|
| script_dir = os.path.dirname(os.path.abspath(__file__))
|
| nb_dir = os.path.join(script_dir, "notebooks")
|
| if not os.path.exists(nb_dir):
|
| os.makedirs(nb_dir)
|
|
|
| notebooks = []
|
| for f in os.listdir(nb_dir):
|
| if f.endswith('.json'):
|
| try:
|
| with open(os.path.join(nb_dir, f), 'r', encoding='utf-8') as file:
|
| data = json.load(file)
|
| notebooks.append({
|
| "filename": f,
|
| "name": data.get("name", f.replace('.json', '')),
|
| "data": data
|
| })
|
| except Exception:
|
| pass
|
| return web.json_response({"notebooks": notebooks})
|
|
|
| async def api_save_notebook(request):
|
| try:
|
| data = await request.json()
|
| filename = data.get("filename", "")
|
| if not filename.endswith('.json'): |
| filename += '.json' |
| try: |
| filename = require_filename(filename) |
| except ValueError: |
| return web.json_response({"status": "error", "message": "Invalid filename"}, status=400) |
|
|
| script_dir = os.path.dirname(os.path.abspath(__file__))
|
| nb_dir = os.path.join(script_dir, "notebooks")
|
| if not os.path.exists(nb_dir):
|
| os.makedirs(nb_dir)
|
|
|
| file_path = resolve_within(nb_dir, filename) |
| with open(file_path, 'w', encoding='utf-8') as f:
|
| json.dump(data.get("data", {}), f, indent=4, ensure_ascii=False)
|
|
|
| return web.json_response({"status": "success"})
|
| except Exception as e:
|
| return web.json_response({"status": "error", "message": str(e)})
|
|
|
| async def api_delete_notebook(request):
|
| try:
|
| data = await request.json()
|
| filename = data.get("filename", "")
|
| try: |
| filename = require_filename(filename) |
| except ValueError: |
| return web.json_response({"status": "error", "message": "Invalid filename"}, status=400) |
|
|
| script_dir = os.path.dirname(os.path.abspath(__file__))
|
| nb_dir = os.path.join(script_dir, "notebooks")
|
| file_path = resolve_within(nb_dir, filename) |
|
|
| if os.path.exists(file_path):
|
| os.remove(file_path)
|
| return web.json_response({"status": "success"})
|
| return web.json_response({"status": "error", "message": "File not found"})
|
| except Exception as e:
|
| return web.json_response({"status": "error", "message": str(e)})
|
|
|
|
|
