api.py

from epubconv.epubconv import convertEPUB, config
import asyncio
import websockets
import os
from threading import Timer, Thread

settings = config.load()

async def api(websocket, path):
    file_path = f'./temp/{await websocket.recv()}.epub'
    result = await convertEPUB(file_path, lambda x:websocket.send(x))
    if (result['status']):
        Timer(settings['tempTime'], lambda x: os.remove(x) if os.path.isfile(x) else None, [f"./temp/{result['id']}.epub"]).start()
        await websocket.send(">>>>> 正在傳輸轉換結果...")
        await websocket.send(result['id'])
        
    else:
        await websocket.send(f"轉換失敗。\n錯誤: {result['error']}")

async def start_server():
    print("///// EPUB 轉換服務已啟動 /////")
    await websockets.serve(api, settings["wsHost"], settings["wsPort"])
    print(f'ws://{settings["wsHost"]}:{settings["wsPort"]}')
    
loop = asyncio.get_event_loop()
loop.create_task(start_server())
loop.run_forever()