refactor; rename classes and methods
This commit is contained in:
47
src/customer_cipher.py
Normal file
47
src/customer_cipher.py
Normal file
@@ -0,0 +1,47 @@
|
||||
from dataclasses import dataclass
|
||||
from typing import ClassVar
|
||||
|
||||
from src.models import KeypadSize
|
||||
from src.utils import generate_random_nonrepeating_list
|
||||
|
||||
|
||||
@dataclass
|
||||
class CustomerCipher:
|
||||
prop_key: list[int]
|
||||
set_key: list[int]
|
||||
keypad_size: KeypadSize
|
||||
MAX_KEYS: ClassVar[int] = 256
|
||||
MAX_PROP_PER_KEY: ClassVar[int] = 256
|
||||
|
||||
def __post_init__(self):
|
||||
self.check_keys_vs_props()
|
||||
|
||||
def check_keys_vs_props(self) -> None:
|
||||
if self.keypad_size.is_dispersable:
|
||||
raise ValueError("number of keys must be less than the number of "
|
||||
"properties per key to be dispersion resistant")
|
||||
|
||||
@classmethod
|
||||
def create(cls, keypad_size: KeypadSize) -> 'CustomerCipher':
|
||||
if keypad_size.numb_of_keys > cls.MAX_KEYS or keypad_size.props_per_key > cls.MAX_PROP_PER_KEY:
|
||||
raise ValueError(f"Keys and properties per key must not exceed {cls.MAX_KEYS}")
|
||||
return cls(
|
||||
prop_key=generate_random_nonrepeating_list(keypad_size.numb_of_props),
|
||||
set_key=generate_random_nonrepeating_list(keypad_size.props_per_key),
|
||||
keypad_size=keypad_size,
|
||||
)
|
||||
|
||||
def renew(self):
|
||||
self.prop_key = generate_random_nonrepeating_list(self.keypad_size.numb_of_props)
|
||||
self.set_key = generate_random_nonrepeating_list(self.keypad_size.props_per_key)
|
||||
|
||||
def get_prop_set_val(self, prop: int) -> int:
|
||||
assert (prop in self.prop_key)
|
||||
prop_idx = self.prop_key.index(prop)
|
||||
set_idx = prop_idx % self.keypad_size.props_per_key
|
||||
return self.set_key[set_idx]
|
||||
|
||||
def get_set_index(self, set_val: int) -> int:
|
||||
if set_val not in self.set_key:
|
||||
raise ValueError(f"Set value {set_val} not found in set values")
|
||||
return self.set_key.index(set_val)
|
||||
Reference in New Issue
Block a user