view video_ingest.py @ 27:27a754f8e8f8

deployment fixes
author drewp@bigasterisk.com
date Sat, 14 Oct 2023 17:08:36 -0700
parents 9e56e86a6814
children
line wrap: on
line source

import asyncio
from dataclasses import dataclass
from typing import Any, Coroutine

import dl_queue
from video_file_store import VideoFileStore


@dataclass
class VideoIngest:
    store: VideoFileStore

    async def addContent(self, name: str, body: Coroutine[Any, Any, bytes]):
        await self.store.save(name, iter([await body]))

    async def ingestUrl(self, url: str, folder:str):
        outDir = self.store.top / folder.lstrip('/') #danger user input
        dl_queue.queue.put({'url': url, 'outDir': str(outDir)})

    async def events(self):
        prev = None
        while True:
            p = dl_queue.pending()
            if p != prev:
                prev = p
                yield p
            await asyncio.sleep(1)