numpy refactor
This commit is contained in:
@@ -1,9 +1,9 @@
|
||||
from dataclasses import dataclass
|
||||
from uuid import UUID
|
||||
import numpy as np
|
||||
from src.customer_cipher import CustomerCipher
|
||||
from src.models import NKodePolicy
|
||||
from src.user import User
|
||||
from src.utils import xor_lists
|
||||
|
||||
@dataclass
|
||||
class Customer:
|
||||
@@ -12,7 +12,7 @@ class Customer:
|
||||
cipher: CustomerCipher
|
||||
users: dict[str, User]
|
||||
|
||||
# TODO: validate policy and keypad size don't conflict
|
||||
# TODO: validate policy and keypad_list size don't conflict
|
||||
|
||||
def add_new_user(self, user: User):
|
||||
if user.username in self.users:
|
||||
@@ -56,8 +56,8 @@ class Customer:
|
||||
new_attrs = self.cipher.prop_key
|
||||
new_sets = self.cipher.set_key
|
||||
|
||||
attrs_xor = xor_lists(new_attrs, old_attrs)
|
||||
set_xor = xor_lists(new_sets, old_sets)
|
||||
attrs_xor = np.bitwise_xor(new_attrs, old_attrs)
|
||||
set_xor = np.bitwise_xor(new_sets, old_sets)
|
||||
for user in self.users.values():
|
||||
user.renew_keys(set_xor, attrs_xor)
|
||||
self.users[user.username] = user
|
||||
@@ -66,7 +66,7 @@ class Customer:
|
||||
def valid_new_nkode(self, passcode_attr_idx: list[int]) -> bool:
|
||||
nkode_len = len(passcode_attr_idx)
|
||||
passcode_set_values = [
|
||||
self.cipher.get_prop_set_val(self.cipher.prop_key[attr_idx]) for attr_idx in passcode_attr_idx
|
||||
self.cipher.get_prop_set_val(int(self.cipher.prop_key[attr_idx])) for attr_idx in passcode_attr_idx
|
||||
]
|
||||
distinct_sets = len(set(passcode_set_values))
|
||||
distinct_attributes = len(set(passcode_attr_idx))
|
||||
|
||||
Reference in New Issue
Block a user