文件操作 - stream.py

文件: /usr/lib/python3/dist-packages/twisted/trial/_dist/stream.py

编辑文件内容

"""
Buffer byte streams.
"""

from itertools import count
from typing import Dict, Iterator, List, TypeVar

from attrs import Factory, define

from twisted.protocols.amp import AMP, Command, Integer, String as Bytes

T = TypeVar("T")

class StreamOpen(Command):
    """
    Open a new stream.
    """

response = [(b"streamId", Integer())]

class StreamWrite(Command):
    """
    Write a chunk of data to a stream.
    """

arguments = [
        (b"streamId", Integer()),
        (b"data", Bytes()),
    ]

@define
class StreamReceiver:
    """
    Buffering de-multiplexing byte stream receiver.
    """

_counter: Iterator[int] = count()
    _streams: Dict[int, List[bytes]] = Factory(dict)

def open(self) -> int:
        """
        Open a new stream and return its unique identifier.
        """
        newId = next(self._counter)
        self._streams[newId] = []
        return newId

def write(self, streamId: int, chunk: bytes) -> None:
        """
        Write to an open stream using its unique identifier.

@raise KeyError: If there is no such open stream.
        """
        self._streams[streamId].append(chunk)

def finish(self, streamId: int) -> List[bytes]:
        """
        Indicate an open stream may receive no further data and return all of
        its current contents.

@raise KeyError: If there is no such open stream.
        """
        return self._streams.pop(streamId)

def chunk(data: bytes, chunkSize: int) -> Iterator[bytes]:
    """
    Break a byte string into pieces of no more than ``chunkSize`` length.

@param data: The byte string.

@param chunkSize: The maximum length of the resulting pieces.  All pieces
        except possibly the last will be this length.

@return: The pieces.
    """
    pos = 0
    while pos < len(data):
        yield data[pos : pos + chunkSize]
        pos += chunkSize

async def stream(amp: AMP, chunks: Iterator[bytes]) -> int:
    """
    Send the given stream chunks, one by one, over the given connection.

The chunks are sent using L{StreamWrite} over a stream opened using
    L{StreamOpen}.

@return: The identifier of the stream over which the chunks were sent.
    """
    streamId = (await amp.callRemote(StreamOpen))["streamId"]
    assert isinstance(streamId, int)

for oneChunk in chunks:
        await amp.callRemote(StreamWrite, streamId=streamId, data=oneChunk)
    return streamId

文件操作 - stream.py

文件: /usr/lib/python3/dist-packages/twisted/trial/_dist/stream.py

编辑文件内容

修改文件时间

删除文件