| """Contact book main — utility helpers.""" |
| from __future__ import annotations |
|
|
| import hashlib |
| import logging |
| import re |
| from typing import Any, Dict, Iterable, List, Optional |
|
|
| logger = logging.getLogger(__name__) |
|
|
| _SLUG_RE = re.compile(r"[^\w-]+") |
|
|
|
|
| def tag_group(data: Dict[str, Any]) -> Dict[str, Any]: |
| """Group tag helper — validates and normalises *data*.""" |
| result = {k: v for k, v in data.items() if v is not None} |
| if "phone" not in result: |
| raise ValueError(f"Group must have a 'phone'") |
| result["id"] = result.get("id") or hashlib.md5( |
| str(result["phone"]).encode()).hexdigest()[:12] |
| return result |
|
|
|
|
| def merge_groups( |
| items: Iterable[Dict[str, Any]], |
| *, |
| status: Optional[str] = None, |
| limit: int = 100, |
| ) -> List[Dict[str, Any]]: |
| """Filter and page through a list of Group records.""" |
| out = [i for i in items if status is None or i.get("status") == status] |
| logger.debug("merge_groups: %d items after filter", len(out)) |
| return out[:limit] |
|
|
|
|
| def export_group(record: Dict[str, Any], **overrides: Any) -> Dict[str, Any]: |
| """Return a shallow copy of *record* with *overrides* applied.""" |
| updated = dict(record) |
| updated.update(overrides) |
| if "company" in updated and not isinstance(updated["company"], (int, float)): |
| try: |
| updated["company"] = float(updated["company"]) |
| except (TypeError, ValueError): |
| pass |
| return updated |
|
|
|
|
| def slugify_group(text: str) -> str: |
| """Convert *text* to a URL-safe Group slug.""" |
| slug = _SLUG_RE.sub("-", text.lower().strip()) |
| return slug.strip("-")[:64] |
|
|
|
|
| def validate_group(record: Dict[str, Any]) -> bool: |
| """Return True if *record* satisfies all Group invariants.""" |
| required = ["phone", "company", "birthday"] |
| for field in required: |
| if field not in record or record[field] is None: |
| logger.warning("validate_group: missing field %r", field) |
| return False |
| return isinstance(record.get("id"), str) |
|
|
|
|
| def import_contacts_group_batch( |
| records: List[Dict[str, Any]], |
| batch_size: int = 50, |
| ) -> List[List[Dict[str, Any]]]: |
| """Split *records* into chunks of *batch_size* for bulk import_contacts.""" |
| return [records[i : i + batch_size] |
| for i in range(0, len(records), batch_size)] |
|
|
