rename functions and attributes
This commit is contained in:
73
src/customer.py
Normal file
73
src/customer.py
Normal file
@@ -0,0 +1,73 @@
|
||||
from uuid import UUID
|
||||
from pydantic import BaseModel
|
||||
from src.customer_interface import CustomerInterface
|
||||
from src.models import NKodePolicy
|
||||
from src.user_db_model import UserDBModel
|
||||
from src.utils import xor_lists
|
||||
|
||||
|
||||
class Customer(BaseModel):
|
||||
customer_id: UUID
|
||||
nkode_policy: NKodePolicy
|
||||
interface: CustomerInterface
|
||||
users: dict[str, UserDBModel]
|
||||
|
||||
def add_new_user(self, user: UserDBModel):
|
||||
self.users[user.username] = user
|
||||
|
||||
def valid_key_entry(self, username, selected_keys) -> bool:
|
||||
assert (username in self.users.keys())
|
||||
assert (all(key_idx < self.interface.numb_of_keys for key_idx in selected_keys))
|
||||
passcode_len = len(selected_keys)
|
||||
user = self.users[username]
|
||||
|
||||
passcode_set_vals = user.user_keys.decipher_mask(
|
||||
user.enciphered_passcode.mask, self.interface.set_vals, len(selected_keys))
|
||||
set_vals_idx = [self.interface.get_set_index(set_val) for set_val in passcode_set_vals]
|
||||
|
||||
presumed_selected_attributes = []
|
||||
presumed_selected_attributes_idx = []
|
||||
for idx in range(passcode_len):
|
||||
key_numb = selected_keys[idx]
|
||||
key_attribute_idxs = user.user_interface.get_key_attr_idxs(key_numb)
|
||||
|
||||
set_idx = set_vals_idx[idx]
|
||||
selected_attr_idx = key_attribute_idxs[set_idx]
|
||||
presumed_selected_attributes_idx.append(selected_attr_idx)
|
||||
presumed_selected_attributes.append(selected_attr_idx)
|
||||
|
||||
enciphered_attr = user.user_keys.encipher_salt_hash_code(presumed_selected_attributes, self.interface)
|
||||
if enciphered_attr != user.enciphered_passcode.code:
|
||||
return False
|
||||
|
||||
if user.renew:
|
||||
user.refresh_passcode(presumed_selected_attributes_idx, self.interface)
|
||||
return True
|
||||
|
||||
def renew_keys(self) -> bool:
|
||||
attrs_before = self.interface.attr_vals.copy()
|
||||
sets_before = self.interface.set_vals.copy()
|
||||
self.interface.renew_interface()
|
||||
attrs_after = self.interface.attr_vals
|
||||
sets_after = self.interface.set_vals
|
||||
|
||||
attrs_xor = xor_lists(attrs_after, attrs_before)
|
||||
set_xor = xor_lists(sets_after, sets_before)
|
||||
for user in self.users.values():
|
||||
user.renew_keys(set_xor, attrs_xor)
|
||||
return True
|
||||
|
||||
def valid_new_nkode(self, passcode_attr_idx: list[int]) -> bool:
|
||||
nkode_len = len(passcode_attr_idx)
|
||||
passcode_set_values = [
|
||||
self.interface.get_attr_set_val(self.interface.attr_vals[attr_idx]) for attr_idx in passcode_attr_idx
|
||||
]
|
||||
distinct_sets = len(set(passcode_set_values))
|
||||
distinct_attributes = len(set(passcode_attr_idx))
|
||||
if (
|
||||
self.nkode_policy.min_nkode_len <= nkode_len <= self.nkode_policy.max_nkode_len and
|
||||
distinct_sets >= self.nkode_policy.distinct_sets and
|
||||
distinct_attributes >= self.nkode_policy.distinct_attributes
|
||||
):
|
||||
return True
|
||||
return False
|
||||
Reference in New Issue
Block a user