forked from LBRYCommunity/lbry-sdk
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
|
import json
|
||
|
|
from copy import deepcopy
|
||
|
|
|
||
|
|
import google.protobuf.json_format as json_pb # pylint: disable=no-name-in-module
|
||
|
|
from google.protobuf.message import Message # pylint: disable=no-name-in-module,import-error
|
||
|
|
|
||
|
|
from lbrynet.schema.proto3 import claim_pb2 as claim_pb
|
||
|
|
|
||
|
|
|
||
|
|
class Schema(Message):
|
||
|
|
@classmethod
|
||
|
|
def load(cls, message):
|
||
|
|
raise NotImplementedError
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def _load(cls, data, message):
|
||
|
|
if isinstance(data, dict):
|
||
|
|
data = json.dumps(data)
|
||
|
|
return json_pb.Parse(data, message)
|
||
|
|
|
||
|
|
|
||
|
|
class Claim(Schema):
|
||
|
|
CLAIM_TYPE_STREAM = 0 #fixme: 0 is unset, should be fixed on proto file to be 1 and 2!
|
||
|
|
CLAIM_TYPE_CERT = 1
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def load(cls, message: dict):
|
||
|
|
_claim = deepcopy(message)
|
||
|
|
_message_pb = claim_pb.Claim()
|
||
|
|
|
||
|
|
if "certificate" in _claim: # old protobuf, migrate
|
||
|
|
_cert = _claim.pop("certificate")
|
||
|
|
assert isinstance(_cert, dict)
|
||
|
|
_message_pb.type = Claim.CLAIM_TYPE_CERT
|
||
|
|
_message_pb.channel.MergeFrom(claim_pb.Channel(public_key=_cert.pop("publicKey")))
|
||
|
|
_claim = {} # so we dont need to know what other fields we ignored
|
||
|
|
elif "channel" in _claim:
|
||
|
|
_channel = _claim.pop("channel")
|
||
|
|
_message_pb.type = Claim.CLAIM_TYPE_CERT
|
||
|
|
_message_pb.channel = claim_pb.Channel(**_channel)
|
||
|
|
elif "stream" in _claim:
|
||
|
|
pass # fixme
|
||
|
|
else:
|
||
|
|
raise AttributeError
|
||
|
|
|
||
|
|
return cls._load(_claim, _message_pb)
|