| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346 |
- from __future__ import annotations
- from datetime import UTC, datetime, timedelta
- from html import escape
- from secrets import token_urlsafe
- from typing import Any
- from pyrogram import StopPropagation, filters
- from pyrogram.enums import ChatMembersFilter, ChatMemberStatus, ParseMode
- from pyrogram.types import (
- CallbackQuery,
- ChatMemberUpdated,
- InlineKeyboardButton,
- InlineKeyboardMarkup,
- Message,
- )
- from wbb import BOT_PROFILE_ID, BOT_USERNAME, app, log
- from wbb.core.reply_keyboard import remove_reply_keyboard
- from wbb.services.directory import (
- DirectoryServiceError,
- admin_change_teacher_state,
- admin_decide_teacher,
- admin_source_chat_ids,
- apply_as_teacher,
- change_own_teacher_state,
- clear_user_location,
- directory_for_user,
- format_distance,
- observe_directory_user,
- observe_group_member,
- profile_summary,
- public_teacher,
- require_teacher_admin,
- start_presence_sweeper,
- status_value,
- verify_platform_membership,
- )
- from wbb.services.message_cleanup import schedule_group_cleanup
- from wbb.utils.dbdirectory import (
- APPLICATION_APPROVED,
- APPLICATION_PENDING,
- APPLICATION_REJECTED,
- APPLICATION_REVOKED,
- get_directory_profile,
- list_directory_profiles,
- list_teacher_applications,
- )
- __MODULE__ = "技师咨询与评价"
- __HELP__ = """/directory - 打开技师咨询与评价菜单,也可在群内发送“技师”。
- /nearby - 浏览公开技师,也可发送“附近技师”。
- /teacher_rank - 浏览公开技师,也可发送“技师榜单”。
- /teacher_apply - 申请成为技师。
- /teacher_online - 技师上线。
- /teacher_offline - 技师下线。
- /teacher_list - 技师上榜。
- /teacher_unlist - 技师下榜。"""
- APPLICATION_LABELS = {
- "none": "未申请",
- APPLICATION_PENDING: "技师申请待审批",
- APPLICATION_APPROVED: "技师",
- APPLICATION_REJECTED: "技师申请已拒绝",
- APPLICATION_REVOKED: "技师资格已撤销",
- }
- TEACHER_ACTION_LABELS = {
- "list": "上榜",
- "unlist": "下榜",
- "online": "上线",
- "offline": "下线",
- }
- BUTTON_NEARBY = "附近技师"
- BUTTON_RANK = "技师榜单"
- BUTTON_BROWSE = "浏览技师"
- BUTTON_APPLY = "申请成为技师"
- BUTTON_APPROVAL = "技师审批"
- BUTTON_STATUS = "我的状态"
- BUTTON_LIST = "上榜"
- BUTTON_UNLIST = "下榜"
- BUTTON_ONLINE = "🟢 上线"
- BUTTON_OFFLINE = "🔴 下线"
- BUTTON_CLEAR_LOCATION = "清除位置"
- _admin_flows: dict[int, dict[str, Any]] = {}
- _admin_confirmations: dict[str, dict[str, Any]] = {}
- _source_chat_contexts: dict[int, dict[str, Any]] = {}
- def _display_name(profile: dict[str, Any]) -> str:
- return str(profile.get("display_name") or f"用户 {profile['user_id']}")
- def _directory_keyboard(
- profile: dict[str, Any] | None,
- *,
- is_admin: bool,
- ) -> InlineKeyboardMarkup:
- status = str((profile or {}).get("application_status") or "none")
- rows: list[list[InlineKeyboardButton]] = [
- [InlineKeyboardButton(BUTTON_BROWSE, callback_data="dir:browse")],
- ]
- if status == APPLICATION_APPROVED:
- rows.extend(
- [
- [
- InlineKeyboardButton(
- BUTTON_UNLIST if profile and profile.get("listed") else BUTTON_LIST,
- callback_data=(
- "dir:self:unlist"
- if profile and profile.get("listed")
- else "dir:self:list"
- ),
- ),
- InlineKeyboardButton(
- BUTTON_OFFLINE
- if profile and profile.get("online")
- else BUTTON_ONLINE,
- callback_data=(
- "dir:self:offline"
- if profile and profile.get("online")
- else "dir:self:online"
- ),
- ),
- ],
- [InlineKeyboardButton(BUTTON_STATUS, callback_data="dir:status")],
- ]
- )
- else:
- rows.append(
- [
- InlineKeyboardButton(
- "技师申请审核中"
- if status == APPLICATION_PENDING
- else BUTTON_APPLY,
- callback_data=(
- "dir:status"
- if status == APPLICATION_PENDING
- else "dir:apply"
- ),
- )
- ]
- )
- if is_admin:
- rows.append(
- [
- InlineKeyboardButton(
- BUTTON_APPROVAL,
- callback_data="dir:admin:pending:1",
- )
- ]
- )
- return InlineKeyboardMarkup(
- rows,
- )
- def _group_directory_keyboard(
- profile: dict[str, Any] | None,
- *,
- chat_id: int,
- is_admin: bool,
- ) -> InlineKeyboardMarkup:
- status = str((profile or {}).get("application_status") or "none")
- rows: list[list[InlineKeyboardButton]] = [
- [
- InlineKeyboardButton(
- BUTTON_BROWSE,
- url=f"https://t.me/{BOT_USERNAME}?start=services",
- )
- ]
- ]
- if status == APPLICATION_APPROVED:
- rows.append(
- [
- InlineKeyboardButton(
- BUTTON_UNLIST if profile and profile.get("listed") else BUTTON_LIST,
- callback_data=(
- "dir:self:unlist"
- if profile and profile.get("listed")
- else "dir:self:list"
- ),
- ),
- InlineKeyboardButton(
- BUTTON_OFFLINE if profile and profile.get("online") else BUTTON_ONLINE,
- callback_data=(
- "dir:self:offline"
- if profile and profile.get("online")
- else "dir:self:online"
- ),
- ),
- InlineKeyboardButton(BUTTON_STATUS, callback_data="dir:status"),
- ]
- )
- elif status == APPLICATION_PENDING:
- rows.append(
- [InlineKeyboardButton("技师申请审核中", callback_data="dir:status")]
- )
- else:
- rows.append(
- [InlineKeyboardButton(BUTTON_APPLY, callback_data="dir:apply")]
- )
- if is_admin:
- rows.append(
- [InlineKeyboardButton(BUTTON_APPROVAL, callback_data="dir:admin:pending:1")]
- )
- return InlineKeyboardMarkup(rows)
- def _presence_label(online: bool) -> str:
- return "🟢 在线" if online else "🔴 离线"
- def _menu_text(summary: dict[str, Any]) -> str:
- lines = [
- "<b>技师服务咨询与评价</b>",
- "顾客可浏览服务、直接私聊咨询,并在服务后评价技师。",
- "具体地点由双方私聊沟通,平台仅展示技师填写的大概服务区域。",
- ]
- status = summary["application_status"]
- if status == APPLICATION_APPROVED:
- lines.extend(
- [
- "身份:技师",
- f"榜单:{'已上榜' if summary['listed'] else '未上榜'}",
- f"状态:{_presence_label(summary['online'])}",
- ]
- )
- elif status == APPLICATION_PENDING:
- lines.append("技师申请:待审批")
- elif status == APPLICATION_REJECTED:
- lines.append("技师申请:未通过")
- elif status == APPLICATION_REVOKED:
- lines.append("技师资格:已撤销")
- return "\n".join(lines)
- async def send_directory_menu(
- message: Message,
- *,
- source_chat_id: int | None = None,
- ) -> None:
- if not message.from_user:
- return
- await remove_reply_keyboard(message)
- if source_chat_id is not None:
- try:
- member = await app.get_chat_member(source_chat_id, message.from_user.id)
- if getattr(member, "status", None) not in {
- ChatMemberStatus.OWNER,
- ChatMemberStatus.ADMINISTRATOR,
- ChatMemberStatus.MEMBER,
- ChatMemberStatus.RESTRICTED,
- }:
- return await message.reply_text("你已不是该群成员,无法打开菜单。")
- source_chat_title = ""
- try:
- source_chat = await app.get_chat(source_chat_id)
- source_chat_title = str(getattr(source_chat, "title", "") or "")
- except Exception:
- pass
- await observe_group_member(
- chat_id=source_chat_id,
- chat_title=source_chat_title,
- user=message.from_user,
- status=status_value(member.status),
- active=True,
- verified=True,
- )
- _source_chat_contexts[int(message.from_user.id)] = {
- "chat_id": int(source_chat_id),
- "chat_title": source_chat_title,
- "expires_at": datetime.now(UTC) + timedelta(minutes=30),
- }
- except Exception:
- return await message.reply_text("暂时无法核验来源群身份,请稍后重试。")
- profile = await observe_directory_user(message.from_user)
- admin_result = await verify_platform_membership(
- user_id=int(message.from_user.id),
- require_admin=True,
- )
- summary = profile_summary(profile, None)
- await message.reply_text(
- _menu_text(summary),
- parse_mode=ParseMode.HTML,
- reply_markup=_directory_keyboard(
- {**profile, "online": summary["online"]},
- is_admin=bool(admin_result.get("allowed")),
- ),
- )
- async def _verify_group_member(message: Message) -> bool:
- if not message.from_user:
- return False
- try:
- member = await app.get_chat_member(message.chat.id, message.from_user.id)
- except Exception:
- await message.reply_text("暂时无法验证你的群成员身份,请稍后重试。")
- return False
- active = member.status in {
- ChatMemberStatus.OWNER,
- ChatMemberStatus.ADMINISTRATOR,
- ChatMemberStatus.MEMBER,
- ChatMemberStatus.RESTRICTED,
- }
- await observe_group_member(
- chat_id=message.chat.id,
- chat_title=message.chat.title or "",
- user=message.from_user,
- status=status_value(member.status),
- active=active,
- verified=True,
- )
- if not active:
- await message.reply_text("该功能仅对当前群成员开放。")
- return active
- @app.on_message(
- filters.command("directory")
- | (filters.regex(r"^技师$") & filters.group)
- )
- async def directory_command(_, message: Message):
- if message.chat.type.value == "private":
- return await send_directory_menu(message)
- if not await _verify_group_member(message):
- return
- profile = await observe_directory_user(message.from_user)
- summary = profile_summary(profile, None)
- member = await app.get_chat_member(message.chat.id, message.from_user.id)
- response = await message.reply_text(
- _menu_text(summary),
- parse_mode=ParseMode.HTML,
- reply_markup=_group_directory_keyboard(
- {**profile, "online": summary["online"]},
- chat_id=message.chat.id,
- is_admin=member.status in {
- ChatMemberStatus.OWNER,
- ChatMemberStatus.ADMINISTRATOR,
- },
- ),
- )
- schedule_group_cleanup(message, response)
- @app.on_message(
- filters.command("nearby")
- | (filters.regex(r"^附近技师$") & filters.group)
- )
- async def nearby_command(_, message: Message):
- if message.chat.type.value != "private" and not await _verify_group_member(message):
- return
- await _open_service_directory(message)
- @app.on_message(
- filters.command("teacher_rank")
- | (filters.regex(r"^技师榜单$") & filters.group)
- )
- async def teacher_rank_command(_, message: Message):
- if message.chat.type.value != "private" and not await _verify_group_member(message):
- return
- await _open_service_directory(message)
- @app.on_message(filters.command("teacher_apply"))
- async def teacher_apply_command(_, message: Message):
- if message.chat.type.value != "private" and not await _verify_group_member(message):
- return
- await _apply_teacher(message)
- async def _self_state_command(
- target: Message | CallbackQuery,
- action: str,
- ) -> None:
- if not target.from_user:
- return
- message = target.message if isinstance(target, CallbackQuery) else target
- is_group = int(message.chat.id) < 0
- if is_group:
- try:
- member = await app.get_chat_member(message.chat.id, target.from_user.id)
- except Exception:
- if isinstance(target, CallbackQuery):
- return await target.answer("暂时无法核验群成员身份。", show_alert=True)
- return await target.reply_text("暂时无法核验群成员身份。")
- if member.status not in {
- ChatMemberStatus.OWNER,
- ChatMemberStatus.ADMINISTRATOR,
- ChatMemberStatus.MEMBER,
- ChatMemberStatus.RESTRICTED,
- }:
- if isinstance(target, CallbackQuery):
- return await target.answer("该功能仅对当前群成员开放。", show_alert=True)
- return await target.reply_text("该功能仅对当前群成员开放。")
- await observe_group_member(
- chat_id=message.chat.id,
- chat_title=message.chat.title or "",
- user=target.from_user,
- status=status_value(member.status),
- active=True,
- verified=True,
- )
- try:
- profile, _ = await change_own_teacher_state(
- user=target.from_user,
- action=action,
- source="telegram_group" if is_group else "telegram_private",
- idempotency_key=(
- f"teacher-self:{message.chat.id}:{message.id}:"
- f"{target.from_user.id}:{action}"
- if is_group
- else None
- ),
- chat_id=message.chat.id if is_group else None,
- )
- response = await message.reply_text(
- f"已{TEACHER_ACTION_LABELS[action]}。"
- + (
- "本次上线状态将在 24 小时后自动失效。"
- if action == "online"
- else ""
- )
- )
- if isinstance(target, CallbackQuery):
- await target.answer("状态已更新。")
- if not is_group and isinstance(target, Message):
- await _refresh_keyboard(target, profile)
- else:
- schedule_group_cleanup(message, response)
- except DirectoryServiceError as exc:
- if isinstance(target, CallbackQuery):
- await target.answer(str(exc), show_alert=True)
- else:
- response = await target.reply_text(str(exc))
- schedule_group_cleanup(target, response)
- @app.on_message(filters.command("teacher_online"))
- async def teacher_online_command(_, message: Message):
- if message.chat.type.value != "private" and not await _verify_group_member(message):
- return
- await _self_state_command(message, "online")
- @app.on_message(filters.command("teacher_offline"))
- async def teacher_offline_command(_, message: Message):
- if message.chat.type.value != "private" and not await _verify_group_member(message):
- return
- await _self_state_command(message, "offline")
- @app.on_message(filters.command("teacher_list"))
- async def teacher_list_command(_, message: Message):
- if message.chat.type.value != "private" and not await _verify_group_member(message):
- return
- await _self_state_command(message, "list")
- @app.on_message(filters.command("teacher_unlist"))
- async def teacher_unlist_command(_, message: Message):
- if message.chat.type.value != "private" and not await _verify_group_member(message):
- return
- await _self_state_command(message, "unlist")
- async def _refresh_keyboard(
- message: Message,
- profile: dict[str, Any] | None,
- ) -> None:
- if not message.from_user:
- return
- await remove_reply_keyboard(message)
- admin_result = await verify_platform_membership(
- user_id=int(message.from_user.id),
- require_admin=True,
- )
- summary = profile_summary(
- profile or {"user_id": message.from_user.id},
- None,
- )
- await message.reply_text(
- "菜单已更新。",
- reply_markup=_directory_keyboard(
- {**(profile or {}), "online": summary["online"]},
- is_admin=bool(admin_result.get("allowed")),
- ),
- )
- def _admin_mention(user: Any) -> str:
- name = " ".join(
- value
- for value in (
- str(getattr(user, "first_name", "") or "").strip(),
- str(getattr(user, "last_name", "") or "").strip(),
- )
- if value
- ) or str(getattr(user, "username", "") or "管理员")
- return f'<a href="tg://user?id={int(user.id)}">{escape(name)}</a>'
- async def _notify_application_admins(profile: dict[str, Any]) -> None:
- chat_id = profile.get("application_source_chat_id")
- if not chat_id:
- return
- admins: list[Any] = []
- try:
- async for member in app.get_chat_members(
- int(chat_id),
- filter=ChatMembersFilter.ADMINISTRATORS,
- ):
- user = getattr(member, "user", None)
- if user and not getattr(user, "is_bot", False):
- admins.append(user)
- except Exception as exc:
- log.error(f"获取技师申请来源群管理员失败:{exc}")
- mentions = " ".join(_admin_mention(user) for user in admins[:20])
- applicant = escape(_display_name(profile))
- text = (
- f"{mentions}\n" if mentions else ""
- ) + (
- f"有新的技师申请待审批:<b>{applicant}</b>\n"
- f"用户 ID:<code>{profile['user_id']}</code>"
- )
- markup = InlineKeyboardMarkup(
- [
- [
- InlineKeyboardButton(
- "查看并审批",
- callback_data=f"dir:admin:detail:{profile['user_id']}",
- )
- ]
- ]
- )
- try:
- await app.send_message(
- int(chat_id),
- text,
- parse_mode=ParseMode.HTML,
- reply_markup=markup,
- )
- except Exception as exc:
- log.error(f"发送技师申请群提醒失败:{exc}")
- private_text = (
- f"群“{escape(profile.get('application_source_chat_title') or str(chat_id))}”"
- f"有新的技师申请:<b>{applicant}</b>。"
- )
- for admin in admins:
- if int(admin.id) == int(profile["user_id"]):
- continue
- try:
- await app.send_message(
- int(admin.id),
- private_text,
- parse_mode=ParseMode.HTML,
- reply_markup=markup,
- )
- except Exception:
- pass
- async def _apply_teacher(target: Message | CallbackQuery) -> None:
- if not target.from_user:
- return
- message = target.message if isinstance(target, CallbackQuery) else target
- is_group = int(message.chat.id) < 0
- source_context = None
- if not is_group:
- source_context = _source_chat_contexts.get(int(target.from_user.id))
- if (
- source_context
- and source_context["expires_at"] < datetime.now(UTC)
- ):
- _source_chat_contexts.pop(int(target.from_user.id), None)
- source_context = None
- if is_group:
- try:
- member = await app.get_chat_member(message.chat.id, target.from_user.id)
- except Exception:
- if isinstance(target, CallbackQuery):
- return await target.answer("暂时无法核验群成员身份。", show_alert=True)
- return await target.reply_text("暂时无法核验群成员身份。")
- if member.status not in {
- ChatMemberStatus.OWNER,
- ChatMemberStatus.ADMINISTRATOR,
- ChatMemberStatus.MEMBER,
- ChatMemberStatus.RESTRICTED,
- }:
- if isinstance(target, CallbackQuery):
- return await target.answer("该功能仅对当前群成员开放。", show_alert=True)
- return await target.reply_text("该功能仅对当前群成员开放。")
- await observe_group_member(
- chat_id=message.chat.id,
- chat_title=message.chat.title or "",
- user=target.from_user,
- status=status_value(member.status),
- active=True,
- verified=True,
- )
- try:
- profile, created = await apply_as_teacher(
- user=target.from_user,
- chat_id=(
- int(message.chat.id)
- if is_group
- else int(source_context["chat_id"])
- if source_context
- else None
- ),
- chat_title=(
- str(message.chat.title or "")
- if is_group
- else str(source_context.get("chat_title") or "")
- if source_context
- else ""
- ),
- source="telegram_group" if is_group else "telegram_private",
- )
- except DirectoryServiceError as exc:
- if isinstance(target, CallbackQuery):
- return await target.answer(str(exc), show_alert=True)
- return await target.reply_text(str(exc))
- response_text = (
- "技师申请已提交,来源群管理员已收到审批提醒。"
- if created
- else "技师申请正在等待管理员审批。"
- )
- response = await message.reply_text(response_text)
- if isinstance(target, CallbackQuery):
- await target.answer("申请已提交。" if created else "申请正在审批中。")
- if created:
- _source_chat_contexts.pop(int(target.from_user.id), None)
- await _notify_application_admins(profile)
- if is_group:
- schedule_group_cleanup(message, response)
- elif isinstance(target, Message):
- await _refresh_keyboard(target, profile)
- async def _show_status(target: Message | CallbackQuery) -> None:
- if not target.from_user:
- return
- message = target.message if isinstance(target, CallbackQuery) else target
- profile = await observe_directory_user(target.from_user)
- summary = profile_summary(
- profile,
- None,
- )
- if summary["application_status"] != APPLICATION_APPROVED:
- status_text = {
- APPLICATION_PENDING: "技师申请正在等待管理员审批。",
- APPLICATION_REJECTED: "技师申请未通过。",
- APPLICATION_REVOKED: "技师资格已被撤销。",
- }.get(summary["application_status"], "你尚未申请成为技师。")
- response = await message.reply_text(status_text)
- if isinstance(target, CallbackQuery):
- await target.answer()
- schedule_group_cleanup(message, response)
- return
- lines = [
- "<b>我的技师状态</b>",
- "服务区域:请在 /technician_profile 快速页面填写大概区域",
- f"榜单:{'已上榜' if summary['listed'] else '未上榜'}",
- f"上下线:{_presence_label(summary['online'])}",
- (
- "自动下线:"
- + escape(
- summary["online_until"].astimezone(UTC).strftime(
- "%Y-%m-%d %H:%M UTC"
- )
- )
- if summary["online"] and summary["online_until"]
- else ""
- ),
- ]
- response = await message.reply_text(
- "\n".join(line for line in lines if line),
- parse_mode=ParseMode.HTML,
- )
- if isinstance(target, CallbackQuery):
- await target.answer()
- schedule_group_cleanup(message, response)
- async def _open_service_directory(target: Message | CallbackQuery) -> None:
- message = target.message if isinstance(target, CallbackQuery) else target
- if message.chat.type.value != "private":
- response = await message.reply_text(
- "请在 Bot 私聊中浏览技师、咨询和评价。",
- reply_markup=InlineKeyboardMarkup(
- [
- [
- InlineKeyboardButton(
- BUTTON_BROWSE,
- url=f"https://t.me/{BOT_USERNAME}?start=services",
- )
- ]
- ]
- ),
- )
- schedule_group_cleanup(message, response)
- if isinstance(target, CallbackQuery):
- await target.answer()
- return
- from wbb.modules.technician_services import _ask_search_location
- await _ask_search_location(target)
- async def _show_radius_picker(target: Message | CallbackQuery) -> None:
- markup = InlineKeyboardMarkup(
- [
- [
- InlineKeyboardButton(
- f"{radius} 公里",
- callback_data=f"dir:list:{radius}:1",
- )
- for radius in (5, 10)
- ],
- [
- InlineKeyboardButton(
- f"{radius} 公里",
- callback_data=f"dir:list:{radius}:1",
- )
- for radius in (20, 50)
- ],
- ]
- )
- if isinstance(target, CallbackQuery):
- await target.message.edit_text("请选择附近技师的搜索范围:", reply_markup=markup)
- else:
- response = await target.reply_text(
- "请选择附近技师的搜索范围:",
- reply_markup=markup,
- )
- schedule_group_cleanup(target, response)
- def _teacher_result_markup(
- teachers: list[dict[str, Any]],
- *,
- radius_key: str,
- page: int,
- total: int,
- ) -> InlineKeyboardMarkup:
- rows: list[list[InlineKeyboardButton]] = []
- teacher_buttons: list[InlineKeyboardButton] = []
- for item in teachers:
- username = str(item.get("username") or "")
- status = "🟢" if item["online"] else "🔴"
- teacher_buttons.append(
- InlineKeyboardButton(
- f"{status} {item['display_name']}"[:18],
- url=f"https://t.me/{username}",
- )
- )
- rows.extend(
- teacher_buttons[index : index + 3]
- for index in range(0, len(teacher_buttons), 3)
- )
- navigation: list[InlineKeyboardButton] = []
- if page > 1:
- navigation.append(
- InlineKeyboardButton(
- "上一页",
- callback_data=f"dir:list:{radius_key}:{page - 1}",
- )
- )
- if page * 10 < total:
- navigation.append(
- InlineKeyboardButton(
- "下一页",
- callback_data=f"dir:list:{radius_key}:{page + 1}",
- )
- )
- if navigation:
- rows.append(navigation)
- rows.append([InlineKeyboardButton("切换附近范围", callback_data="dir:radius")])
- return InlineKeyboardMarkup(rows)
- async def _show_teacher_results(
- target: Message | CallbackQuery,
- *,
- radius_km: int | None,
- page: int,
- ) -> None:
- user = target.from_user
- try:
- raw_items, total = await directory_for_user(
- user=user,
- radius_km=radius_km,
- page=page,
- page_size=10,
- )
- except DirectoryServiceError as exc:
- if isinstance(target, CallbackQuery):
- return await target.answer(str(exc), show_alert=True)
- return await target.reply_text(str(exc))
- teachers = [public_teacher(item) for item in raw_items]
- title = (
- f"<b>{radius_km} 公里内的技师</b>"
- if radius_km is not None
- else "<b>技师榜单</b>"
- )
- lines = [title, f"共 {total} 位已上榜技师,按距离排序。"]
- for index, item in enumerate(teachers, start=(page - 1) * 10 + 1):
- lines.append(
- f"\n{index}. <b>{escape(item['display_name'])}</b> "
- f"(@{escape(item['username'])})\n"
- f"状态:{_presence_label(item['online'])} | "
- f"距离:{format_distance(item['distance_meters'])}"
- )
- if not teachers:
- lines.append("\n当前范围内没有已上榜技师。")
- markup = _teacher_result_markup(
- teachers,
- radius_key=str(radius_km) if radius_km is not None else "all",
- page=page,
- total=total,
- )
- if isinstance(target, CallbackQuery):
- await target.message.edit_text(
- "\n".join(lines),
- parse_mode=ParseMode.HTML,
- disable_web_page_preview=True,
- reply_markup=markup,
- )
- await target.answer()
- else:
- response = await target.reply_text(
- "\n".join(lines),
- parse_mode=ParseMode.HTML,
- disable_web_page_preview=True,
- reply_markup=markup,
- )
- schedule_group_cleanup(target, response)
- async def _show_admin_applications(
- target: Message | CallbackQuery,
- *,
- page: int = 1,
- ) -> None:
- try:
- source_chat_ids = await admin_source_chat_ids(target.from_user)
- except DirectoryServiceError as exc:
- if isinstance(target, CallbackQuery):
- return await target.answer(str(exc), show_alert=True)
- return await target.reply_text(str(exc))
- items, total = await list_teacher_applications(
- status=APPLICATION_PENDING,
- source_bot_id=str(BOT_PROFILE_ID)
- if source_chat_ids is not None
- else "",
- source_chat_ids=source_chat_ids,
- page=page,
- page_size=5,
- )
- lines = ["<b>技师申请审批</b>", f"待审批:{total}"]
- rows: list[list[InlineKeyboardButton]] = []
- for item in items:
- lines.append(
- f"\n{escape(_display_name(item))} (@{escape(item.get('username') or '')})"
- f"\n用户 ID:<code>{item['user_id']}</code>"
- )
- rows.append(
- [
- InlineKeyboardButton(
- f"查看 {_display_name(item)[:18]}",
- callback_data=f"dir:admin:detail:{item['user_id']}",
- )
- ]
- )
- if page > 1:
- rows.append(
- [
- InlineKeyboardButton(
- "上一页",
- callback_data=f"dir:admin:pending:{page - 1}",
- )
- ]
- )
- if page * 5 < total:
- rows.append(
- [
- InlineKeyboardButton(
- "下一页",
- callback_data=f"dir:admin:pending:{page + 1}",
- )
- ]
- )
- rows.append(
- [InlineKeyboardButton("已批准技师", callback_data="dir:admin:teachers:1")]
- )
- markup = InlineKeyboardMarkup(rows)
- if isinstance(target, CallbackQuery):
- await target.message.edit_text(
- "\n".join(lines),
- parse_mode=ParseMode.HTML,
- reply_markup=markup,
- )
- await target.answer()
- else:
- await target.reply_text(
- "\n".join(lines),
- parse_mode=ParseMode.HTML,
- reply_markup=markup,
- )
- async def _show_admin_teachers(target: CallbackQuery, page: int) -> None:
- try:
- source_chat_ids = await admin_source_chat_ids(target.from_user)
- except DirectoryServiceError as exc:
- return await target.answer(str(exc), show_alert=True)
- items, total = await list_directory_profiles(
- application_status=APPLICATION_APPROVED,
- source_bot_id=str(BOT_PROFILE_ID)
- if source_chat_ids is not None
- else "",
- source_chat_ids=source_chat_ids,
- page=page,
- page_size=8,
- )
- rows = [
- [
- InlineKeyboardButton(
- _display_name(item)[:28],
- callback_data=f"dir:admin:detail:{item['user_id']}",
- )
- ]
- for item in items
- ]
- navigation: list[InlineKeyboardButton] = []
- if page > 1:
- navigation.append(
- InlineKeyboardButton(
- "上一页",
- callback_data=f"dir:admin:teachers:{page - 1}",
- )
- )
- if page * 8 < total:
- navigation.append(
- InlineKeyboardButton(
- "下一页",
- callback_data=f"dir:admin:teachers:{page + 1}",
- )
- )
- if navigation:
- rows.append(navigation)
- rows.append([InlineKeyboardButton("待审批", callback_data="dir:admin:pending:1")])
- await target.message.edit_text(
- f"<b>已批准技师</b>\n共 {total} 人",
- parse_mode=ParseMode.HTML,
- reply_markup=InlineKeyboardMarkup(rows),
- )
- await target.answer()
- async def _show_admin_teacher_detail(target: CallbackQuery, user_id: int) -> None:
- try:
- await require_teacher_admin(target.from_user, user_id)
- except DirectoryServiceError as exc:
- return await target.answer(str(exc), show_alert=True)
- profile = await get_directory_profile(user_id)
- if not profile:
- return await target.answer("未找到该成员资料。", show_alert=True)
- summary = profile_summary(profile, None)
- service_area = str(
- (profile.get("service_profile") or {}).get("public_area_text") or "尚未填写"
- )
- lines = [
- f"<b>{escape(summary['display_name'])}</b>",
- f"用户名:@{escape(summary.get('username') or '未设置')}",
- f"用户 ID:<code>{user_id}</code>",
- f"申请:{escape(APPLICATION_LABELS.get(summary['application_status'], '未申请'))}",
- f"大概服务区域:{escape(service_area)}",
- f"榜单:{'已上榜' if summary['listed'] else '未上榜'}",
- f"状态:{_presence_label(summary['online'])}",
- ]
- rows: list[list[InlineKeyboardButton]] = []
- if summary["application_status"] == APPLICATION_PENDING:
- rows.append(
- [
- InlineKeyboardButton(
- "批准",
- callback_data=f"dir:admin:approveask:{user_id}",
- ),
- InlineKeyboardButton(
- "拒绝",
- callback_data=f"dir:admin:flow:reject:{user_id}",
- ),
- ]
- )
- if summary["application_status"] == APPLICATION_APPROVED:
- rows.extend(
- [
- [
- InlineKeyboardButton(
- "强制下榜" if summary["listed"] else "强制上榜",
- callback_data=(
- f"dir:admin:flow:"
- f"{'unlist' if summary['listed'] else 'list'}:{user_id}"
- ),
- ),
- InlineKeyboardButton(
- "强制下线" if summary["online"] else "强制上线",
- callback_data=(
- f"dir:admin:flow:"
- f"{'offline' if summary['online'] else 'online'}:{user_id}"
- ),
- ),
- ],
- [
- InlineKeyboardButton(
- "撤销技师资格",
- callback_data=f"dir:admin:flow:revoke:{user_id}",
- )
- ],
- ]
- )
- rows.append([InlineKeyboardButton("返回审批", callback_data="dir:admin:pending:1")])
- await target.message.edit_text(
- "\n".join(lines),
- parse_mode=ParseMode.HTML,
- reply_markup=InlineKeyboardMarkup(rows),
- )
- try:
- await target.answer()
- except Exception:
- pass
- async def _start_admin_reason_flow(
- target: CallbackQuery,
- *,
- action: str,
- user_id: int,
- ) -> None:
- try:
- await require_teacher_admin(target.from_user, user_id)
- except DirectoryServiceError as exc:
- return await target.answer(str(exc), show_alert=True)
- _admin_flows[int(target.from_user.id)] = {
- "action": action,
- "user_id": int(user_id),
- "expires_at": datetime.now(UTC) + timedelta(minutes=10),
- }
- await target.message.reply_text(
- f"请输入“{TEACHER_ACTION_LABELS.get(action, action)}”的原因,"
- "发送 /cancel 可取消。"
- )
- await target.answer()
- async def _confirm_admin_action(target: CallbackQuery, token: str) -> None:
- confirmation = _admin_confirmations.get(token)
- if not confirmation or confirmation["actor_id"] != target.from_user.id:
- return await target.answer("确认已失效,请重新操作。", show_alert=True)
- if confirmation["expires_at"] < datetime.now(UTC):
- _admin_confirmations.pop(token, None)
- return await target.answer("确认已超时,请重新操作。", show_alert=True)
- action = confirmation["action"]
- user_id = int(confirmation["user_id"])
- reason = str(confirmation.get("reason") or "")
- try:
- if action in {"reject", "revoke"}:
- profile = await admin_decide_teacher(
- actor=target.from_user,
- user_id=user_id,
- action=action,
- reason=reason,
- )
- else:
- profile = await admin_change_teacher_state(
- actor=target.from_user,
- user_id=user_id,
- action=action,
- reason=reason,
- )
- with_reason = f"\n原因:{reason}" if reason else ""
- with_suppressed_notification = (
- "技师申请已被拒绝" if action == "reject" else
- "技师资格已被撤销" if action == "revoke" else
- f"技师状态已被管理员修改为“{TEACHER_ACTION_LABELS[action]}”"
- )
- try:
- await app.send_message(
- user_id,
- f"{with_suppressed_notification}。{with_reason}",
- )
- except Exception:
- pass
- _admin_confirmations.pop(token, None)
- await target.answer("操作成功。")
- await _show_admin_teacher_detail(target, int(profile["user_id"]))
- except DirectoryServiceError as exc:
- await target.answer(str(exc), show_alert=True)
- @app.on_callback_query(filters.regex(r"^dir:"))
- async def directory_callback(_, query: CallbackQuery):
- parts = str(query.data).split(":")
- try:
- if parts[1] == "browse":
- return await _open_service_directory(query)
- if parts[1] == "apply":
- return await _apply_teacher(query)
- if parts[1] == "status":
- return await _show_status(query)
- if parts[1] == "self":
- return await _self_state_command(query, parts[2])
- if parts[1] == "radius":
- return await _open_service_directory(query)
- if parts[1] == "list":
- return await _open_service_directory(query)
- if parts[1] == "clearask":
- return await query.message.edit_text(
- "确认清除历史精确坐标吗?公开状态和大概服务区域不会受影响。",
- reply_markup=InlineKeyboardMarkup(
- [
- [
- InlineKeyboardButton(
- "确认清除",
- callback_data="dir:clear",
- ),
- InlineKeyboardButton("取消", callback_data="dir:cancel"),
- ]
- ]
- ),
- )
- if parts[1] == "clear":
- await clear_user_location(user=query.from_user)
- await query.message.edit_text("历史精确坐标已清除,公开状态保持不变。")
- return await query.answer("已清除。")
- if parts[1] == "cancel":
- await query.message.edit_text("已取消。")
- return await query.answer()
- if parts[1] == "admin":
- if parts[2] == "pending":
- return await _show_admin_applications(query, page=int(parts[3]))
- if parts[2] == "teachers":
- return await _show_admin_teachers(query, int(parts[3]))
- if parts[2] == "detail":
- return await _show_admin_teacher_detail(query, int(parts[3]))
- if parts[2] == "approveask":
- user_id = int(parts[3])
- return await query.message.edit_text(
- "批准后技师身份将在全部群和 Bot 中生效,确认继续吗?",
- reply_markup=InlineKeyboardMarkup(
- [
- [
- InlineKeyboardButton(
- "确认批准",
- callback_data=f"dir:admin:approve:{user_id}",
- ),
- InlineKeyboardButton(
- "取消",
- callback_data=f"dir:admin:detail:{user_id}",
- ),
- ]
- ]
- ),
- )
- if parts[2] == "approve":
- profile = await admin_decide_teacher(
- actor=query.from_user,
- user_id=int(parts[3]),
- action="approve",
- )
- try:
- await app.send_message(
- int(profile["user_id"]),
- "你的技师申请已通过,可以在菜单中自助上榜。",
- )
- except Exception:
- pass
- await query.answer("已批准。")
- return await _show_admin_teacher_detail(
- query,
- int(profile["user_id"]),
- )
- if parts[2] == "flow":
- return await _start_admin_reason_flow(
- query,
- action=parts[3],
- user_id=int(parts[4]),
- )
- if parts[2] == "confirm":
- return await _confirm_admin_action(query, parts[3])
- except (DirectoryServiceError, ValueError) as exc:
- return await query.answer(str(exc), show_alert=True)
- except Exception as exc:
- log.error(f"技师目录回调失败:{exc}")
- return await query.answer("操作失败,请稍后重试。", show_alert=True)
- @app.on_message(filters.command("cancel") & filters.private, group=-20)
- async def cancel_directory_flow(_, message: Message):
- if not message.from_user:
- return
- admin_flow = _admin_flows.pop(message.from_user.id, None)
- if not admin_flow:
- return
- await remove_reply_keyboard(message)
- await message.reply_text("已取消当前操作。")
- raise StopPropagation
- @app.on_message(filters.private & filters.text, group=13)
- async def directory_private_text(_, message: Message):
- if not message.from_user:
- return
- flow = _admin_flows.get(message.from_user.id)
- if flow:
- if flow["expires_at"] < datetime.now(UTC):
- _admin_flows.pop(message.from_user.id, None)
- return await message.reply_text("操作已超时,请重新打开技师审批。")
- reason = str(message.text or "").strip()
- if not reason:
- return await message.reply_text("原因不能为空,请重新输入。")
- token = token_urlsafe(8)
- _admin_confirmations[token] = {
- **flow,
- "actor_id": message.from_user.id,
- "reason": reason,
- "expires_at": datetime.now(UTC) + timedelta(minutes=10),
- }
- _admin_flows.pop(message.from_user.id, None)
- return await message.reply_text(
- f"确认执行“{TEACHER_ACTION_LABELS.get(flow['action'], flow['action'])}”吗?"
- f"\n用户 ID:{flow['user_id']}\n原因:{escape(reason)}",
- reply_markup=InlineKeyboardMarkup(
- [
- [
- InlineKeyboardButton(
- "确认执行",
- callback_data=f"dir:admin:confirm:{token}",
- ),
- InlineKeyboardButton("取消", callback_data="dir:cancel"),
- ]
- ]
- ),
- )
- text = str(message.text or "").strip()
- if text == BUTTON_BROWSE:
- return await _open_service_directory(message)
- if text == BUTTON_APPLY:
- return await _apply_teacher(message)
- if text == BUTTON_STATUS:
- return await _show_status(message)
- if text == "技师申请审核中":
- return await _show_status(message)
- if text == BUTTON_CLEAR_LOCATION:
- return await message.reply_text(
- "确认清除历史精确坐标吗?公开状态和大概服务区域不会受影响。",
- reply_markup=InlineKeyboardMarkup(
- [
- [
- InlineKeyboardButton("确认清除", callback_data="dir:clear"),
- InlineKeyboardButton("取消", callback_data="dir:cancel"),
- ]
- ]
- ),
- )
- if text == BUTTON_APPROVAL:
- return await _show_admin_applications(message)
- action = {
- BUTTON_LIST: "list",
- BUTTON_UNLIST: "unlist",
- BUTTON_ONLINE: "online",
- BUTTON_OFFLINE: "offline",
- "上线": "online",
- "下线": "offline",
- }.get(text)
- if action:
- return await _self_state_command(message, action)
- @app.on_message(filters.group & filters.text, group=13)
- async def teacher_presence_group_text(_, message: Message):
- if not message.from_user:
- return
- text = str(message.text or "").strip()
- action = {
- "上线": "online",
- "下线": "offline",
- BUTTON_ONLINE: "online",
- BUTTON_OFFLINE: "offline",
- }.get(text)
- if not action:
- return
- profile = await get_directory_profile(message.from_user.id)
- if not profile or profile.get("application_status") != APPLICATION_APPROVED:
- return
- if not await _verify_group_member(message):
- return
- try:
- _, applied = await change_own_teacher_state(
- user=message.from_user,
- action=action,
- source="telegram_group",
- idempotency_key=(
- f"teacher-presence:{message.chat.id}:{message.id}:"
- f"{message.from_user.id}"
- ),
- chat_id=message.chat.id,
- )
- if applied:
- response = await message.reply_text(
- f"全平台状态已切换为{_presence_label(action == 'online')}。"
- + ("24 小时后将自动下线。" if action == "online" else "")
- )
- schedule_group_cleanup(message, response)
- except DirectoryServiceError as exc:
- response = await message.reply_text(str(exc))
- schedule_group_cleanup(message, response)
- @app.on_chat_member_updated(group=13)
- async def directory_chat_member_updated(_, update: ChatMemberUpdated):
- member = update.new_chat_member or update.old_chat_member
- if not member or not member.user or member.user.is_bot:
- return
- status = update.new_chat_member.status if update.new_chat_member else ChatMemberStatus.LEFT
- active = status in {
- ChatMemberStatus.OWNER,
- ChatMemberStatus.ADMINISTRATOR,
- ChatMemberStatus.MEMBER,
- ChatMemberStatus.RESTRICTED,
- }
- await observe_group_member(
- chat_id=update.chat.id,
- chat_title=update.chat.title or "",
- user=member.user,
- status=status_value(status),
- active=active,
- verified=True,
- )
- start_presence_sweeper()
|