File size: 1,826 Bytes
44bafb2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
# All credits to https://github.com/LuanRT/googlevideo
from typing import Optional
from pytubefix.sabr.proto import BinaryReader, BinaryWriter
class TimeRange:
def __init__(self):
self.start: int = 0
self.duration: int = 0
self.timescale: int = 0
@staticmethod
def decode(input_data, length=None):
reader = input_data if isinstance(input_data, BinaryReader) else BinaryReader(input_data)
end = reader.len if length is None else reader.pos + length
message = TimeRange
while reader.pos < end:
tag = reader.uint32()
field = tag >> 3
if field == 1 and tag == 8:
message.start = long_to_number(reader.int64())
continue
elif field == 2 and tag == 16:
message.duration = long_to_number(reader.int64())
continue
elif field == 3 and tag == 24:
message.timescale = reader.int32()
continue
elif (tag & 7) == 4 or tag == 0:
break
else:
reader.skip(tag & 7)
return message
def encode(self, writer: Optional[BinaryWriter] = None) -> BinaryWriter:
writer = writer or BinaryWriter()
if self.start != 0:
writer.uint32(8).int64(self.start)
if self.duration != 0:
writer.uint32(16).int64(self.duration)
if self.timescale != 0:
writer.uint32(24).int32(self.timescale)
return writer
def long_to_number(int64_value):
value = int(str(int64_value))
if value > (2 ** 53 - 1):
raise OverflowError("Value is larger than 9007199254740991")
if value < -(2 ** 53 - 1):
raise OverflowError("Value is smaller than -9007199254740991")
return value |