dkelly/pynkode
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rename functions and attributes

Browse Source
This commit is contained in:
Donovan
2024-07-17 08:43:45 -05:00
parent d23ad7ac2b
commit c4d8233730
12 changed files with 195 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
nkode_api.py
View File

@@ -1,7 +1,8 @@
from uuid import UUID, uuid4
from pydantic import BaseModel
from src.customer_db_model import CustomerDBModel
from src.customer import Customer
from src.models import NKodePolicy
from src.session_cache import SessionCacheModel
from src.user_cipher_keys import UserCipherKeys
from src.user_db_model import UserDBModel
@@ -10,13 +11,13 @@ from src.customer_interface import CustomerInterface
class NKodeAPI(BaseModel):
customers: dict[UUID, CustomerDBModel] = {}
customers: dict[UUID, Customer] = {}
sessions: dict[UUID, SessionCacheModel] = {}
def generate_index_interface(self, customer_id: UUID) -> tuple[UUID, list[int]]:
assert (customer_id in self.customers.keys())
customer = self.customers[customer_id]
set_interface = UserInterface.new_interface(customer.interface.attrs_per_key, customer.interface.numb_keys)
set_interface = UserInterface.new(customer.interface.attrs_per_key, customer.interface.numb_of_keys)
new_session = SessionCacheModel(
session_id=uuid4(),
set_interface=set_interface.interface_index,
@@ -45,13 +46,13 @@ class NKodeAPI(BaseModel):
assert (session_id in self.sessions.keys())
session = self.sessions[session_id]
assert (customer_id == session.customer_id)
numb_of_keys = customer.interface.numb_keys
numb_of_keys = customer.interface.numb_of_keys
attrs_per_key = customer.interface.attrs_per_key
assert (all(0 <= key <= numb_of_keys for key in key_selection))
set_interface = UserInterface(
interface_index=session.set_interface,
numb_sets=attrs_per_key,
numb_keys=numb_of_keys,
attrs_per_key=attrs_per_key,
numb_of_keys=numb_of_keys,
)
set_interface.disperse_interface()
session.username = username
@@ -60,7 +61,7 @@ class NKodeAPI(BaseModel):
self.sessions[session_id] = session
return self.sessions[session_id].confirm_interface
def confirm_nkode(self, username: str, customer_id: UUID, confirm_key_entry: list[int], session_id: UUID):
def confirm_nkode(self, username: str, customer_id: UUID, confirm_key_entry: list[int], session_id: UUID) -> bool:
assert (
session_id in self.sessions.keys() and
customer_id == self.sessions[session_id].customer_id and
@@ -68,12 +69,14 @@ class NKodeAPI(BaseModel):
)
session = self.sessions[session_id]
customer = self.customers[customer_id]
numb_of_keys = customer.interface.numb_keys
numb_of_keys = customer.interface.numb_of_keys
attrs_per_key = customer.interface.attrs_per_key
assert (all(0 <= key <= numb_of_keys for key in confirm_key_entry))
passcode = session.deduce_passcode(attrs_per_key, confirm_key_entry)
set_values = customer.interface.set_vals
new_user_keys = UserCipherKeys.new_user_encipher_keys(numb_of_keys, attrs_per_key, set_values)
if not customer.valid_new_nkode(passcode):
return False
new_user_keys = UserCipherKeys.new(numb_of_keys, attrs_per_key, set_values)
enciphered_passcode = new_user_keys.encipher_nkode(passcode, customer.interface)
new_user = UserDBModel(
username=username,
@@ -81,15 +84,14 @@ class NKodeAPI(BaseModel):
user_keys=new_user_keys,
user_interface=UserInterface(
interface_index=self.sessions[session_id].confirm_interface,
numb_sets=attrs_per_key,
numb_keys=numb_of_keys,
attrs_per_key=attrs_per_key,
numb_of_keys=numb_of_keys,
),
)
self.customers[customer_id].add_new_user(new_user)
return "success"
return True
# del self.sessions[session_id]
def login(self, customer_id: UUID, username: str, key_selection: list[int]) -> bool:
assert (customer_id in self.customers.keys())
customer = self.customers[customer_id]
@@ -99,12 +101,13 @@ class NKodeAPI(BaseModel):
assert (customer_id in self.customers.keys())
return self.customers[customer_id].renew_keys()
def create_new_customer(self, numb_keys: int, numb_sets: int) -> CustomerDBModel:
new_customer = CustomerDBModel(
def create_new_customer(self, numb_keys: int, numb_sets: int, nkode_policy: NKodePolicy) -> UUID:
new_customer = Customer(
customer_id=uuid4(),
interface=CustomerInterface.new_interface(numb_keys, numb_sets),
interface=CustomerInterface.new(numb_keys, numb_sets),
users={},
nkode_policy=nkode_policy
)
self.customers[new_customer.customer_id] = new_customer
return new_customer
return new_customer.customer_id