from . import parser
from . block import Block
from . patch import Patch, fill
from ... paramgen import arcparam
INTERVAL = 1
class ExtractWorker:
    """
    ExtractWorker associates a download session with a block.

    When the worker finishes fetching, the block
    being fetched is splitted and assigned the free worker.

    Parameter
    ----------
    fetch : func :
        extract function of asyncdl

    block : Block :
        Block object that includes chat_data
    
    blocks : list :
        List of Block(s)

    video_id : str :

    parent_block : Block :
        the block from which current block is splitted 
    """
    __slots__ = ['block', 'fetch', 'blocks', 'video_id', 'parent_block']
    def __init__(self, fetch, block, blocks, video_id ):
        self.block:Block = block
        self.fetch = fetch
        self.blocks:list = blocks
        self.video_id:str = video_id
        self.parent_block:Block = None

    async def run(self, session):
        while self.block.continuation:
            patch = await self.fetch(
                self.block.seektime, session)
            fill(self.block, patch)
            self.block.seektime += INTERVAL
        self.block.done = True