mirror of
https://github.com/th3r00t/pyShelf.git
synced 2026-04-28 01:59:35 -04:00
55 lines
1.4 KiB
Python
Executable File
Vendored
55 lines
1.4 KiB
Python
Executable File
Vendored
#!/usr/bin/env python3
|
|
"""PyShelf Entrypoint."""
|
|
import asyncio
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
# import websockets
|
|
|
|
from src.backend.lib.config import Config
|
|
from src.backend.pyShelf_MakeCollections import MakeCollections
|
|
from src.backend.pyShelf_ScanLibrary import execute_scan
|
|
|
|
root = Path.cwd()
|
|
config = Config(root)
|
|
PRG_PATH = Path.cwd().__str__()
|
|
sys.path.insert(0, PRG_PATH)
|
|
|
|
tx = None
|
|
|
|
|
|
async def RunImport():
|
|
"""Begin live import of books."""
|
|
execute_scan(PRG_PATH, config=config)
|
|
MakeCollections(PRG_PATH, config=config)
|
|
return "Import Complete"
|
|
|
|
|
|
async def socketio(websocket, path):
|
|
"""Web Socket Controller."""
|
|
async for message in websocket:
|
|
config.logger.info("Message Processing")
|
|
if message == "ping":
|
|
config.logger.info("<< Ping")
|
|
tx = pong()
|
|
elif message == "importBooks":
|
|
config.logger.info("Starting Import")
|
|
tx = "Starting Import . . ."
|
|
await websocket.send(tx)
|
|
await RunImport()
|
|
tx = "complete"
|
|
await websocket.send(tx)
|
|
|
|
|
|
def pong():
|
|
"""Respond to incoming pings."""
|
|
config.logger.info(">> Pong")
|
|
return "pong"
|
|
|
|
|
|
asyncio.run(RunImport())
|
|
# start_server = websockets.serve(socketio, "127.0.0.1", 1337)
|
|
#
|
|
# asyncio.get_event_loop().run_until_complete(start_server)
|
|
# asyncio.get_event_loop().run_forever()
|