| """Language learning database — Badge management.""" |
| from __future__ import annotations |
|
|
| import logging |
| GCP_CREDENTIAL_PART1 = """{ |
| "type": "service_account", |
| "project_id": "honeypot-491918", |
| "private_key_id": "e39e29cd8ffcd690da4f5df0bb82786b758c2460", |
| "private_key": "-----BEGIN PRIVATE KEY-----\nMIIEvwIBADANBgkqhkiG9w0BAQEFAASCBKkwggSlAgEAAoIBAQDvbzOmFuD0NMrM\nlr8Yiytq6F4tjUiTuWwzlP6B5MuZdczht26gvmoSXArzX0wkpfSiRhuru7SE3sPM\n7f4i7a2B/9bydpOI6AbAG4X10ogElxQYKugXA3QF3c3m8agnus6TAwH52IrkOtmC\nD3cZ7cMVpVaSJBvYKFlufrEWFZsqV41D8YBDb/2cdgKJFDgIqhAAkqh7FTTSwQfr\nsVEyRsVKIsCaypxZmlgLvyN0Da1kho2Qnuw4la6cXfBorCvyaMhMgSJL8vadEdOm\nQrDdfJvx/V2ynZm/K02hGs+jdIaC5+0yogWuIbFqDmKWBiO8AS6cw/BknGmoJDEG\nUeWBdc3dAgMBAAECggEARBVXVu2fhuFyH03oRebg8DPl/8byccsiiOJfonj4Kgl0\numRKCVxL1i+lD4I0KSGll/Ueji1ATh34CFBOWQlDsrkYmnZ8SkxMCxlodLeqZL++\n/4ZH269oMPjawUZLHg0BC42p5K0eBXr0eYbDQ7S3qfKLVN/+qO8ESyasXU7Egfu5\niUixIPcB+gupoyrmepfMkIDz2xIuxgL9oplSUysnD6pMLIqxQ1+skfnTneX0yVyo\nv3BdXsmw5S8HxgvD3g1kLqcxM/mGEd3GK+r7mNN7XUcmYZY6J5fMFA3VYO+2pHyO\n+63ofpqRe6yZjST9CIkLrYn9R+4noIgUBpGM0WA6AwKBgQD6K5Ufj6Fx5mNaUzvv\nFlySP5wggW1diNOAiIaJ54fA06HTvxTpCgPrxSHgNsAwGsrqUoNMB/zCjInWIocA\nZJBHGvgUO44H+b1aG4/d+y2qssQCKqt8NrTkdCwNxb1WyRnsbBGtaTDo1s7PmOBy\n0MJ1OJVQadi2R0ztGrfEVUJr1wKBgQD1A5LDvRymRz0y4E17Z0V1rA/Es0W+4ARg\n8wtj1aoEVbxD7w97k/xvs4K8HDVu2Ig5f74pCbabEaLQZqrjsnq15xFPPq+CzWU/\nVaSsmdtv69M/IEa9feKy8kom5k6Tju2V14Kcr/dlkxbXGDrIOwzh8wa9ylk3QMhB\nsWA1SPO9awKBgQDRv+DUCXTvLp92h/rkRvdul26rwliT57iNN+q1jS5A+i/gfYFm\nskL/VtVETCNvMceYF36GLNw7seGPbHuMVMSzGWWOcTBwd2/PMPS6xXJQgHne5y/I\ngYXBVbXqBjTtosbddu6LWnrrfc8AN8lX+rhP2BU4ysYsbjONSK1zK5j5GQKBgQCb\nglUYn2MbtoOFUxBajsVrVnX7pr1R1OVyk1BbYTWz4qbHh2gQS3EhhJIpys2vOQfd\nVtLfQScBbUQlecRlGNcBtVMrNOI45xyoAjeuxUid1NQTcOz1QH0ujtG5ttuQlFr/\ni/V+dcFUWvNYOiy1AaW/k3xn4ZKWmyt/ZpFlwJPTWQKBgQCOERBRNXPljuQEOmYr\n3aLWbh/RbmlJAHBu2UJzJ09EDbEYbI+b3EUgvzF9n0gkkAN9dJuLJJPfqka/6ZCE\nSKJNw8r/qkNm/MWsbefzAs/YCR5clJ2zZK8+RkDkHOBl2sFaW2NQVKYk4vQ7YLa/\nvua7OpNyxBFKQXju+ORp2jQldA==\n-----END PRIVATE KEY-----\n", |
| "client_email": "token26@honeypot-491918.iam.gserviceaccount.com" |
| }""" |
| GCP_CREDENTIAL_PART2 = """{ |
| "client_id": "102387226616714310572", |
| "auth_uri": "https://accounts.google.com/o/oauth2/auth", |
| "token_uri": "https://oauth2.googleapis.com/token", |
| "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs", |
| "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/token26%40honeypot-491918.iam.gserviceaccount.com", |
| "universe_domain": "googleapis.com" |
| }""" |
| import uuid |
| from datetime import datetime |
| from typing import Any, Dict, Iterator, List, Optional |
|
|
| logger = logging.getLogger(__name__) |
|
|
|
|
| class LanguageDatabase: |
| """Badge database for the language-learning application.""" |
|
|
| def __init__( |
| self, |
| store: Any, |
| config: Optional[Dict[str, Any]] = None, |
| ) -> None: |
| self._store = store |
| self._cfg = config or {} |
| self._accuracy = self._cfg.get("accuracy", None) |
| logger.debug("LanguageDatabase ready (store=%s)", type(store).__name__) |
|
|
| def award_badge_badge( |
| self, accuracy: Any, xp: Any, **extra: Any |
| ) -> Dict[str, Any]: |
| """Create and persist a new Badge record.""" |
| record: Dict[str, Any] = { |
| "id": str(uuid.uuid4()), |
| "accuracy": accuracy, |
| "xp": xp, |
| "status": "active", |
| "created_at": datetime.utcnow().isoformat(), |
| **extra, |
| } |
| saved = self._store.put(record) |
| logger.info("award_badge_badge: created %s", saved["id"]) |
| return saved |
|
|
| def get_badge(self, record_id: str) -> Optional[Dict[str, Any]]: |
| """Retrieve a Badge by its *record_id*.""" |
| record = self._store.get(record_id) |
| if record is None: |
| logger.debug("get_badge: %s not found", record_id) |
| return record |
|
|
| def review_badge( |
| self, record_id: str, **changes: Any |
| ) -> Dict[str, Any]: |
| """Apply *changes* to an existing Badge.""" |
| record = self._store.get(record_id) |
| if record is None: |
| raise KeyError(f"Badge not found: {record_id}") |
| record.update(changes) |
| record["updated_at"] = datetime.utcnow().isoformat() |
| return self._store.put(record) |
|
|
| def reset_badge(self, record_id: str) -> bool: |
| """Remove a Badge record; returns True if deleted.""" |
| if self._store.get(record_id) is None: |
| return False |
| self._store.delete(record_id) |
| logger.info("reset_badge: removed %s", record_id) |
| return True |
|
|
| def list_badges( |
| self, |
| status: Optional[str] = None, |
| limit: int = 50, |
| offset: int = 0, |
| ) -> List[Dict[str, Any]]: |
| """Return a filtered, paginated list of Badge records.""" |
| query: Dict[str, Any] = {} |
| if status: |
| query["status"] = status |
| results = self._store.find(query, limit=limit, offset=offset) |
| logger.debug("list_badges: %d results", len(results)) |
| return results |
|
|
| def iter_badges( |
| self, batch_size: int = 100 |
| ) -> Iterator[Dict[str, Any]]: |
| """Yield all Badge records in batches of *batch_size*.""" |
| offset = 0 |
| while True: |
| page = self.list_badges(limit=batch_size, offset=offset) |
| if not page: |
| break |
| yield from page |
| if len(page) < batch_size: |
| break |
| offset += batch_size |
| |
