|
|
|
|
|
|
|
|
from pytubefix.sabr.proto import BinaryReader, BinaryWriter |
|
|
from pytubefix.sabr.common import FormatId, InitRange, IndexRange |
|
|
|
|
|
|
|
|
class FormatInitializationMetadata: |
|
|
def __init__( self): |
|
|
self.videoId = "" |
|
|
self.formatId = None |
|
|
self.endTimeMs = 0 |
|
|
self.endSegmentNumber = 0 |
|
|
self.mimeType = "" |
|
|
self.initRange = None |
|
|
self.indexRange = None |
|
|
self.field8 = 0 |
|
|
self.durationMs = 0 |
|
|
self.field10 = 0 |
|
|
|
|
|
@staticmethod |
|
|
def encode(message, writer=None): |
|
|
if writer is None: |
|
|
writer = BinaryWriter() |
|
|
if message.videoId != "": |
|
|
writer.uint32(10) |
|
|
writer.string(message.videoId) |
|
|
if message.formatId is not None: |
|
|
writer.uint32(18) |
|
|
FormatId.encode(message.formatId, writer.fork()).join() |
|
|
if message.endTimeMs != 0: |
|
|
writer.uint32(24) |
|
|
writer.int32(message.endTimeMs) |
|
|
if message.endSegmentNumber != 0: |
|
|
writer.uint32(32) |
|
|
writer.int64(message.endSegmentNumber) |
|
|
if message.mimeType != "": |
|
|
writer.uint32(42) |
|
|
writer.string(message.mimeType) |
|
|
if message.initRange is not None: |
|
|
writer.uint32(50) |
|
|
InitRange.encode(message.initRange, writer.fork()).join() |
|
|
if message.indexRange is not None: |
|
|
writer.uint32(58) |
|
|
IndexRange.encode(message.indexRange, writer.fork()).join() |
|
|
if message.field8 != 0: |
|
|
writer.uint32(64) |
|
|
writer.int32(message.field8) |
|
|
if message.durationMs != 0: |
|
|
writer.uint32(72) |
|
|
writer.int32(message.durationMs) |
|
|
if message.field10 != 0: |
|
|
writer.uint32(80) |
|
|
writer.int32(message.field10) |
|
|
return writer |
|
|
|
|
|
@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 = FormatInitializationMetadata() |
|
|
while reader.pos < end: |
|
|
tag = reader.uint32() |
|
|
field_no = tag >> 3 |
|
|
if field_no == 1 and tag == 10: |
|
|
message.videoId = reader.string() |
|
|
continue |
|
|
elif field_no == 2 and tag == 18: |
|
|
message.formatId = FormatId.decode(reader, reader.uint32()) |
|
|
continue |
|
|
elif field_no == 3 and tag == 24: |
|
|
message.endTimeMs = reader.int32() |
|
|
continue |
|
|
elif field_no == 4 and tag == 32: |
|
|
message.endSegmentNumber = long_to_number(reader.int64()) |
|
|
continue |
|
|
elif field_no == 5 and tag == 42: |
|
|
message.mimeType = reader.string() |
|
|
continue |
|
|
elif field_no == 6 and tag == 50: |
|
|
message.initRange = InitRange.decode(reader, reader.uint32()) |
|
|
continue |
|
|
elif field_no == 7 and tag == 58: |
|
|
message.indexRange = IndexRange.decode(reader, reader.uint32()) |
|
|
continue |
|
|
elif field_no == 8 and tag == 64: |
|
|
message.field8 = reader.int32() |
|
|
continue |
|
|
elif field_no == 9 and tag == 72: |
|
|
message.durationMs = reader.int32() |
|
|
continue |
|
|
elif field_no == 10 and tag == 80: |
|
|
message.field10 = reader.int32() |
|
|
continue |
|
|
if (tag & 7) == 4 or tag == 0: |
|
|
break |
|
|
reader.skip(tag & 7) |
|
|
return message |
|
|
|
|
|
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 |
|
|
|
