85 lines
3.5 KiB
Python
85 lines
3.5 KiB
Python
import re
|
||
|
||
from django.db.models import Case, IntegerField, Q, Value, When
|
||
from django.http import JsonResponse
|
||
from django.views.decorators.http import require_GET
|
||
|
||
from apps.customers.models import CustomerProfile
|
||
|
||
from .models import PhoneContactApiKey
|
||
|
||
|
||
_PHONE_DIGITS = str.maketrans("۰۱۲۳۴۵۶۷۸۹٠١٢٣٤٥٦٧٨٩", "01234567890123456789")
|
||
|
||
|
||
def _dialable_number(value):
|
||
number = re.sub(r"[\s().-]", "", str(value or "").translate(_PHONE_DIGITS))
|
||
return number if 0 < len(number) <= 64 and re.fullmatch(r"[0-9+*#]+", number) else ""
|
||
|
||
|
||
def _json(data, status=200):
|
||
response = JsonResponse(data, status=status, json_dumps_params={"ensure_ascii": False})
|
||
response["Cache-Control"] = "no-store"
|
||
return response
|
||
|
||
|
||
@require_GET
|
||
def phone_contacts(request):
|
||
authorization = request.headers.get("Authorization", "")
|
||
scheme, separator, token = authorization.partition(" ")
|
||
if not separator or scheme.lower() != "bearer" or not token or token != token.strip():
|
||
return _json({"error": "unauthorized"}, 401)
|
||
|
||
credential = PhoneContactApiKey.objects.filter(pk=1, is_active=True).first()
|
||
if not credential or not credential.accepts(token):
|
||
return _json({"error": "unauthorized"}, 401)
|
||
|
||
query = request.GET.get("search", "").strip()[:80]
|
||
if len(query) < 2:
|
||
return _json({"contacts": []})
|
||
|
||
try:
|
||
limit = min(50, max(1, int(request.GET.get("limit", "50"))))
|
||
except ValueError:
|
||
return _json({"error": "invalid limit"}, 400)
|
||
|
||
query = query.translate(str.maketrans({"ي": "ی", "ك": "ک", "۰": "0", "۱": "1", "۲": "2", "۳": "3", "۴": "4", "۵": "5", "۶": "6", "۷": "7", "۸": "8", "۹": "9"}))
|
||
contacts = CustomerProfile.objects.select_related("user").filter(
|
||
user__account_type="CUSTOMER", user__is_active=True, user__mobile__isnull=False
|
||
).exclude(user__mobile="").filter(
|
||
Q(first_name__gt="") | Q(last_name__gt="") | Q(organization_name__gt="") |
|
||
Q(user__first_name__gt="") | Q(user__last_name__gt="")
|
||
)
|
||
for word in re.split(r"\s+", query):
|
||
contacts = contacts.filter(
|
||
Q(first_name__icontains=word) | Q(last_name__icontains=word) |
|
||
Q(organization_name__icontains=word) | Q(user__first_name__icontains=word) |
|
||
Q(user__last_name__icontains=word) | Q(user__mobile__icontains=word)
|
||
)
|
||
|
||
preferred = (
|
||
Q(first_name__iexact=query) | Q(last_name__iexact=query) |
|
||
Q(organization_name__iexact=query) | Q(user__first_name__iexact=query) |
|
||
Q(user__last_name__iexact=query)
|
||
)
|
||
starts_with = (
|
||
Q(first_name__istartswith=query) | Q(last_name__istartswith=query) |
|
||
Q(organization_name__istartswith=query) | Q(user__first_name__istartswith=query) |
|
||
Q(user__last_name__istartswith=query)
|
||
)
|
||
contacts = contacts.annotate(_search_rank=Case(
|
||
When(preferred, then=Value(0)),
|
||
When(starts_with, then=Value(1)),
|
||
default=Value(2), output_field=IntegerField(),
|
||
)).order_by("_search_rank", "pk")
|
||
|
||
results = []
|
||
for profile in contacts[:limit]:
|
||
name = (f"{profile.first_name} {profile.last_name}".strip() or
|
||
f"{profile.user.first_name} {profile.user.last_name}".strip() or
|
||
profile.organization_name.strip())[:120]
|
||
number = _dialable_number(profile.user.local_mobile)
|
||
if name and number:
|
||
results.append({"name": name, "phones": [{"type": "همراه", "number": number}]})
|
||
return _json({"contacts": results})
|