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

clean up code

Browse Source
This commit is contained in:
Donovan
2024-07-15 15:43:06 -05:00
parent 1fea44c9ba
commit 2da3e7a443
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/nkode_interface.py
View File

@@ -45,4 +45,3 @@ class CustomerInterface(BaseModel):
def get_set_index(self, set_val: int) -> int: def get_set_index(self, set_val: int) -> int:
assert(set_val in self.set_vals) assert(set_val in self.set_vals)
return self.set_vals.index(set_val) return self.set_vals.index(set_val)

3
src/pseudo_nkode_api.py
View File

@@ -50,6 +50,7 @@ class CustomerDBModel(BaseModel):
return False return False
class SessionCacheModel(BaseModel): class SessionCacheModel(BaseModel):
session_id: UUID session_id: UUID
set_interface: list[int] | None = None set_interface: list[int] | None = None
@@ -161,7 +162,7 @@ class PseudoNKodeAPI(BaseModel):
return passcode return passcode
def login(self, customer_id: UUID, username: str, key_selection: list[int]) -> bool: def login(self, customer_id: UUID, username: str, key_selection: list[int]) -> bool:
assert(customer_id in self.customers.keys()) assert (customer_id in self.customers.keys())
customer = self.customers[customer_id] customer = self.customers[customer_id]
return customer.valid_key_entry(username, key_selection) return customer.valid_key_entry(username, key_selection)

4
test/test_pseudo_nkode_api.py
View File

@@ -1,5 +1,5 @@
import pytest import pytest
from src.pseudo_nkode_api import PseudoNKodeAPI, CustomerDBModel from src.pseudo_nkode_api import PseudoNKodeAPI
@pytest.fixture() @pytest.fixture()
@@ -31,4 +31,4 @@ def test_create_new_user(pseudo_nkode_api, numb_keys, attrs_per_key, user_passco
login_interface = pseudo_nkode_api.get_login_index_interface(username, customer.customer_id) login_interface = pseudo_nkode_api.get_login_index_interface(username, customer.customer_id)
login_key_selection = key_selection(login_interface) login_key_selection = key_selection(login_interface)
successful_login = pseudo_nkode_api.login(customer.customer_id, username, login_key_selection) successful_login = pseudo_nkode_api.login(customer.customer_id, username, login_key_selection)
assert (successful_login) assert successful_login