test create user
This commit is contained in:
@@ -47,14 +47,14 @@ class UserCipherKeys(BaseModel):
|
||||
|
||||
@staticmethod
|
||||
def encode_base64_str(data: list[int]) -> str:
|
||||
return base64.b64encode(data).decode("utf-8")
|
||||
return base64.b64encode(bytes(data)).decode("utf-8")
|
||||
|
||||
@staticmethod
|
||||
def decode_base64_str(data: str) -> list[int]:
|
||||
return list(base64.b64decode(data))
|
||||
|
||||
def _hash_passcode(self, passcode: list[int]) -> str:
|
||||
passcode_digest = base64.b64encode(hashlib.sha256(passcode).digest())
|
||||
passcode_digest = base64.b64encode(hashlib.sha256(bytes(passcode)).digest())
|
||||
hashed_data = bcrypt.hashpw(passcode_digest, self.salt)
|
||||
return hashed_data.decode("utf-8")
|
||||
|
||||
@@ -65,7 +65,7 @@ class UserCipherKeys(BaseModel):
|
||||
) -> EncipheredNKode:
|
||||
max_nkode_len = 10
|
||||
passcode_len = len(nkode_attr_index)
|
||||
user_nkode_attrs = [customer_interface.interface[idx] for idx in nkode_attr_index]
|
||||
user_nkode_attrs = [customer_interface.customer_interface[idx] for idx in nkode_attr_index]
|
||||
user_nkode_mask = [customer_interface.get_attr_set_val(attr) for attr in user_nkode_attrs]
|
||||
mask_cipher = self.pad_user_mask(user_nkode_mask, customer_interface, max_nkode_len)
|
||||
passcode_cipher = self.pass_key
|
||||
@@ -76,11 +76,10 @@ class UserCipherKeys(BaseModel):
|
||||
attr_val = user_nkode_attrs[idx]
|
||||
passcode_cipher[idx] ^= alpha ^ attr_val
|
||||
|
||||
for idx in range(max_nkode_len):
|
||||
set_idx = customer_interface.get_set_index(user_nkode_mask[idx])
|
||||
mask_cipher[idx] ^= self.set_key[set_idx] ^ self.mask_key[idx]
|
||||
|
||||
return EncipheredNKode(
|
||||
code=self._hash_passcode(passcode_cipher),
|
||||
mask=self.base64.b64encode(mask_cipher).decode("utf-8")
|
||||
mask=self.encode_base64_str(mask_cipher)
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user