ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter7/model-action-threshold/tasks/crosscutting_identity/repo/accounts/__init__.py

@@ -0,0 +1 @@
+"""Small account service used by the action-threshold experiment."""

chapter7/model-action-threshold/tasks/crosscutting_identity/repo/accounts/authentication.py

@@ -0,0 +1,5 @@
+def authenticate(store, username, password):
+    profile = store.find(username.strip())
+    if profile is None or profile.password != password:
+        return None
+    return profile

chapter7/model-action-threshold/tasks/crosscutting_identity/repo/accounts/directory.py

@@ -0,0 +1,2 @@
+def lookup_profile(store, username):
+    return store.find(username)

chapter7/model-action-threshold/tasks/crosscutting_identity/repo/accounts/models.py

@@ -0,0 +1,8 @@
+from dataclasses import dataclass
+
+
+@dataclass(frozen=True)
+class Profile:
+    username: str
+    email: str
+    password: str

chapter7/model-action-threshold/tasks/crosscutting_identity/repo/accounts/registration.py

@@ -0,0 +1,9 @@
+from accounts.models import Profile
+
+
+def register(store, username, email, password):
+    display_name = username.strip()
+    key = display_name.lower()
+    profile = Profile(display_name, email, password)
+    store.save(key, profile)
+    return profile

chapter7/model-action-threshold/tasks/crosscutting_identity/repo/accounts/store.py

@@ -0,0 +1,11 @@
+class AccountStore:
+    def __init__(self):
+        self._profiles = {}
+
+    def save(self, key, profile):
+        if key in self._profiles:
+            raise ValueError("username already exists")
+        self._profiles[key] = profile
+
+    def find(self, key):
+        return self._profiles.get(key)

chapter7/model-action-threshold/tasks/crosscutting_identity/repo/tests/test_accounts.py

@@ -0,0 +1,35 @@
+import unittest
+
+from accounts.authentication import authenticate
+from accounts.directory import lookup_profile
+from accounts.registration import register
+from accounts.store import AccountStore
+
+
+class AccountIdentityTests(unittest.TestCase):
+    def setUp(self):
+        self.store = AccountStore()
+
+    def test_registration_preserves_display_name(self):
+        profile = register(self.store, "  Alice.Dev  ", "a@example.com", "secret")
+        self.assertEqual(profile.username, "Alice.Dev")
+
+    def test_login_ignores_case_and_whitespace(self):
+        register(self.store, "Alice.Dev", "a@example.com", "secret")
+        profile = authenticate(self.store, "  ALICE.dev ", "secret")
+        self.assertIsNotNone(profile)
+        self.assertEqual(profile.username, "Alice.Dev")
+
+    def test_directory_lookup_uses_same_identity(self):
+        register(self.store, "Alice.Dev", "a@example.com", "secret")
+        self.assertIsNotNone(lookup_profile(self.store, " alice.DEV "))
+
+    def test_unicode_casefold_and_duplicate_detection(self):
+        register(self.store, "Straße", "one@example.com", "secret")
+        self.assertIsNotNone(authenticate(self.store, "STRASSE", "secret"))
+        with self.assertRaises(ValueError):
+            register(self.store, " strasse ", "two@example.com", "secret")
+
+
+if __name__ == "__main__":
+    unittest.main()