From c52b34c4e197db7cb5f7b810e761e0032b6f47a3 Mon Sep 17 00:00:00 2001 From: tuxxy Date: Mon, 20 Nov 2017 15:37:06 -0700 Subject: [PATCH] Add type hints for RekeyFrag --- npre/umbral.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/npre/umbral.py b/npre/umbral.py index 08f50c1..e08193d 100644 --- a/npre/umbral.py +++ b/npre/umbral.py @@ -146,7 +146,7 @@ class RekeyFrag(object): _pre = PRE() - def __init__(self, id, key, pre=None): + def __init__(self, id: ec.ec_element, key: ec.ec_element, pre=None): self.id = id self.key = key if pre is None: @@ -164,6 +164,7 @@ def __eq__(self, other_kfrag): @classmethod def from_bytes(cls, kfrag_bytes, pre=None): pre = pre or cls._pre - return RekeyFrag(id=ec.deserialize(pre.ecgroup, kfrag_bytes[:len(kfrag_bytes) // 2]), - key=ec.deserialize(pre.ecgroup, kfrag_bytes[len(kfrag_bytes) // 2:]), - pre=pre) + id = ec.deserialize(pre.ecgroup, kfrag_bytes[:len(kfrag_bytes) // 2]) + key = ec.deserialize(pre.ecgroup, kfrag_bytes[len(kfrag_bytes) // 2:]) + + return RekeyFrag(id=id, key=key, pre=pre)