Spaces:
Running
Running
| import getpass | |
| import json | |
| import os | |
| from pathlib import Path | |
| import re | |
| import urllib.error | |
| import urllib.request | |
| from receiver import Clients, Config, network_error_category | |
| def summarize(report): | |
| fields = {"method", "ok", "error", "seconds", "chat_type", "permissions", | |
| "membership", "is_member", "can_send_messages", "can_post_messages"} | |
| return {"status": report.get("status"), | |
| "checks": [{key: value for key, value in check.items() if key in fields} | |
| for check in report.get("checks", [])]} | |
| def compare(token, chat_id, webhook_secret): | |
| config = Config(secret=webhook_secret or "a" * 32, | |
| telegram_token=token, chat_id=chat_id) | |
| report = {"mac": summarize(Clients(config).diagnose_telegram())} | |
| if webhook_secret: | |
| try: | |
| request = urllib.request.Request( | |
| "https://pranalytics-prtooltest.hf.space/diagnostics/" + webhook_secret) | |
| with urllib.request.urlopen(request, timeout=40) as response: | |
| report["space"] = summarize(json.loads(response.read(65536))) | |
| except urllib.error.HTTPError as exc: | |
| report["space"] = {"error": "diagnostics_http_" + str(exc.code)} | |
| exc.close() | |
| except OSError as exc: | |
| report["space"] = {"error": "diagnostics_" + network_error_category(exc)} | |
| except (ValueError, TypeError, AttributeError): | |
| report["space"] = {"error": "diagnostics_invalid_response"} | |
| return report | |
| def main(): | |
| print("Проверка только читает настройки. Сообщения не отправляются.") | |
| print("Значения вводятся скрыто и не сохраняются. Используйте значения, заданные при настройке HF.") | |
| token = getpass.getpass("TELEGRAM_BOT_TOKEN: ").strip() | |
| chat_id = getpass.getpass("TELEGRAM_CHAT_ID: ").strip() | |
| webhook_secret = getpass.getpass("WEBHOOK_SECRET (Enter — пропустить проверку Space): ").strip() | |
| if not re.fullmatch(r"\d+:[A-Za-z0-9_-]+", token): | |
| raise SystemExit("Неверный формат TELEGRAM_BOT_TOKEN.") | |
| if not re.fullmatch(r"-?\d+|@[A-Za-z0-9_]+", chat_id): | |
| raise SystemExit("Неверный формат TELEGRAM_CHAT_ID.") | |
| if webhook_secret and not re.fullmatch(r"[A-Za-z0-9_-]{24,}", webhook_secret): | |
| raise SystemExit("Неверный формат WEBHOOK_SECRET.") | |
| result = json.dumps(compare(token, chat_id, webhook_secret), ensure_ascii=False, indent=2) | |
| destination = Path(__file__).with_name(".local-diagnostics.json") | |
| descriptor = os.open(destination, os.O_WRONLY | os.O_CREAT | os.O_TRUNC | os.O_NOFOLLOW, 0o600) | |
| with os.fdopen(descriptor, "w") as report_file: | |
| report_file.write(result + "\n") | |
| print(result) | |
| print("Обезличенный отчёт сохранён. Вернитесь в чат и напишите: готово.") | |
| if __name__ == "__main__": | |
| try: | |
| main() | |
| except (KeyboardInterrupt, EOFError): | |
| raise SystemExit("Проверка отменена.") from None | |