mirror of
https://github.com/th3r00t/pyShelf.git
synced 2026-04-28 01:59:35 -04:00
b23cbf17d5
enables docker users to access the django administration site without requiring access to a cli
59 lines
1.3 KiB
Python
Executable File
59 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
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"
|
|
|
|
|
|
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()
|