Dink0 commited on
Commit
7baa98e
·
verified ·
1 Parent(s): c26e5e7

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -427
app.py DELETED
@@ -1,427 +0,0 @@
1
- from __future__ import annotations
2
-
3
- import os
4
- import re
5
- import html as html_lib
6
- from typing import List, Tuple, Union
7
-
8
- import requests
9
- from fastapi import FastAPI
10
- from fastapi.middleware.cors import CORSMiddleware
11
- from pydantic import BaseModel
12
-
13
- APP_NAME = "pr-tool-backend"
14
-
15
- VK_API_VERSION = os.getenv("VK_API_VERSION", "5.131")
16
- VK_ACCESS_TOKEN = os.getenv("VK_ACCESS_TOKEN", "")
17
- REQUEST_TIMEOUT = float(os.getenv("REQUEST_TIMEOUT", "15"))
18
-
19
- app = FastAPI(title=APP_NAME)
20
-
21
- # Для простоты разрешаем все источники. Можно сузить список доменов в проде.
22
- app.add_middleware(
23
- CORSMiddleware,
24
- allow_origins=["*"],
25
- allow_credentials=False,
26
- allow_methods=["POST", "GET", "OPTIONS"],
27
- allow_headers=["*"]
28
- )
29
-
30
-
31
- class ParseRequest(BaseModel):
32
- links: Union[List[str], str]
33
-
34
-
35
- class ParseResponse(BaseModel):
36
- html: str
37
- text: str
38
- telegram_total: int
39
- vk_total: int
40
- errors: List[str]
41
-
42
-
43
- # ---------- Вспомогательные функции ----------
44
-
45
- def human_format_views(num: int) -> str:
46
- if num >= 1_000_000:
47
- value = num / 1_000_000
48
- suffix = "M"
49
- else:
50
- value = num / 1000
51
- suffix = "K"
52
-
53
- if value.is_integer():
54
- formatted = f"{int(value)}{suffix}"
55
- else:
56
- formatted = f"{value:.1f}{suffix}"
57
-
58
- return formatted.replace(".", ",")
59
-
60
-
61
- def detect_platform(link: str) -> str:
62
- link = link.strip().lower()
63
- if "t.me/" in link or "telegram.me/" in link:
64
- return "telegram"
65
- if "vk.com/wall" in link:
66
- return "vk"
67
- return ""
68
-
69
-
70
- def canonicalize_tg_link(link: str) -> Tuple[str | None, str | None, str | None]:
71
- link = link.strip()
72
- m = re.match(r"https?://(?:t(?:elegram)?\.me)/(?:s/)?([^/]+)/(?P<id>\d+)", link, re.IGNORECASE)
73
- if not m:
74
- return None, None, None
75
- username = m.group(1)
76
- message_id = m.group("id")
77
- canonical = f"https://t.me/{username}/{message_id}"
78
- return canonical, username, message_id
79
-
80
-
81
- def canonicalize_vk_link(link: str) -> Tuple[str | None, int | None, str | None]:
82
- link = link.strip()
83
- m = re.search(r"vk\.com/wall(-?\d+)_(\d+)", link)
84
- if not m:
85
- return None, None, None
86
- owner_id = int(m.group(1))
87
- post_id = m.group(2)
88
- canonical = f"https://vk.com/wall{owner_id}_{post_id}"
89
- return canonical, owner_id, post_id
90
-
91
-
92
- def process_telegram_link(channel_username: str, message_id: str, canonical_link: str) -> Tuple[str, int]:
93
- # Набор альтернативных адресов (если t.me не резолвится)
94
- urls = [
95
- f"https://t.me/{channel_username}/{message_id}?embed=1",
96
- f"https://t.me/s/{channel_username}/{message_id}",
97
- f"https://telegram.me/{channel_username}/{message_id}?embed=1",
98
- f"https://r.jina.ai/http://t.me/{channel_username}/{message_id}?embed=1",
99
- f"https://r.jina.ai/http://t.me/s/{channel_username}/{message_id}",
100
- f"https://r.jina.ai/http://telegram.me/{channel_username}/{message_id}?embed=1",
101
- ]
102
- headers = {"User-Agent": "Mozilla/5.0"}
103
-
104
- def _parse_views_number(s: str) -> int:
105
- if not s:
106
- return 0
107
- s = s.strip().replace(" ", "").replace(",", ".")
108
- m2 = re.match(r"([\d\.]+)\s*([kKmM]?)", s)
109
- if not m2:
110
- return 0
111
- val = float(m2.group(1))
112
- suf = m2.group(2).lower()
113
- if suf == "k":
114
- val *= 1_000
115
- elif suf == "m":
116
- val *= 1_000_000
117
- return int(val)
118
-
119
- def _fetch_first_ok(url_list):
120
- last_err = None
121
- for u in url_list:
122
- try:
123
- resp = requests.get(u, headers=headers, timeout=REQUEST_TIMEOUT)
124
- if resp.status_code == 200 and resp.text:
125
- return resp.text
126
- except Exception as exc:
127
- last_err = exc
128
- raise last_err or Exception("Не удалось получить страницу Telegram")
129
-
130
- try:
131
- html = _fetch_first_ok(urls)
132
-
133
- title = None
134
- m_title_meta = re.search(
135
- r"<meta[^>]*property=[\"']og:title[\"'][^>]*content=[\"']([^\"']+)[\"']",
136
- html,
137
- re.IGNORECASE,
138
- )
139
- if m_title_meta:
140
- title = m_title_meta.group(1).strip()
141
- if not title:
142
- m_title = re.search(
143
- r'class="tgme_widget_message_owner_name".*?<span[^>]*>(.*?)</span>',
144
- html,
145
- re.IGNORECASE | re.DOTALL,
146
- )
147
- if m_title:
148
- title = m_title.group(1).strip()
149
- if not title:
150
- title = channel_username
151
-
152
- views = 0
153
- m_views = re.search(
154
- r'class="tgme_widget_message_views[^"]*">([\d\s\.,kKmM]+)<',
155
- html,
156
- re.IGNORECASE,
157
- )
158
- if m_views:
159
- views = _parse_views_number(m_views.group(1))
160
- else:
161
- candidates = re.findall(
162
- r'(\d[\d\s\.,]*)([kKmM]?)\s*(?:views|просмотр|просмотра|просмотров|переглядів|visualizações|visualizzazioni|ansichten|visninger|visitas)?',
163
- html,
164
- re.IGNORECASE,
165
- )
166
- if candidates:
167
- last_num, last_suf = candidates[-1]
168
- views = _parse_views_number(last_num + last_suf)
169
-
170
- return title, views
171
-
172
- except Exception as exc:
173
- return f"Ошибка (TG) при обработке {canonical_link}: {exc}", 0
174
-
175
-
176
- def process_vk_link(owner_id: int, post_id: str, canonical_link: str) -> Tuple[str, int]:
177
- posts_param = f"{owner_id}_{post_id}"
178
- api_url = "https://api.vk.com/method/wall.getById"
179
-
180
- params = {
181
- "posts": posts_param,
182
- "v": VK_API_VERSION,
183
- "extended": 1,
184
- }
185
- if VK_ACCESS_TOKEN:
186
- params["access_token"] = VK_ACCESS_TOKEN
187
-
188
- try:
189
- resp = requests.get(api_url, params=params, timeout=REQUEST_TIMEOUT)
190
- except Exception as exc:
191
- return f"**Ошибка (VK)**: {exc} — {canonical_link}", 0
192
-
193
- if resp.status_code != 200:
194
- return f"**Ошибка (VK)**: HTTP {resp.status_code} — {canonical_link}", 0
195
-
196
- data = resp.json()
197
- if "error" in data:
198
- error_msg = data["error"].get("error_msg", "")
199
- return f"**Ошибка (VK)**: {error_msg or 'API error'} — {canonical_link}", 0
200
-
201
- response_data = data.get("response", {})
202
- items = response_data.get("items", [])
203
- if not items:
204
- return f"**Ошибка (VK)**: пост не найден — {canonical_link}", 0
205
-
206
- post = items[0]
207
- views = post.get("views", {}).get("count", 0)
208
- post_owner_id = post.get("owner_id", owner_id)
209
-
210
- title = None
211
- if post_owner_id < 0:
212
- group_id = -post_owner_id
213
- for group in response_data.get("groups", []):
214
- if group.get("id") == group_id:
215
- title = group.get("name")
216
- break
217
- else:
218
- user_id = post_owner_id
219
- for profile in response_data.get("profiles", []):
220
- if profile.get("id") == user_id:
221
- first_name = profile.get("first_name", "")
222
- last_name = profile.get("last_name", "")
223
- title = (first_name + " " + last_name).strip()
224
- break
225
- if not title:
226
- title = "VK пост"
227
-
228
- return title, views
229
-
230
-
231
- def normalize_links(links: Union[List[str], str]) -> List[str]:
232
- if isinstance(links, str):
233
- raw = links.splitlines()
234
- else:
235
- raw = []
236
- for item in links:
237
- raw.extend(str(item).splitlines())
238
- return [line.strip() for line in raw if line.strip()]
239
-
240
-
241
- def _escape(text: str) -> str:
242
- return html_lib.escape(text, quote=True)
243
-
244
-
245
- def build_output(lines: List[Tuple[str, str, object, object]]) -> Tuple[str, str, int, int, List[str]]:
246
- errors: List[str] = []
247
-
248
- tg_groups = {}
249
- vk_groups = {}
250
-
251
- for plat, canonical, a, b in lines:
252
- if plat == "telegram":
253
- username, mid = a, b
254
- if not username or not mid:
255
- key = canonical
256
- tg_groups.setdefault(key, {"title": "Telegram", "items": []})
257
- tg_groups[key]["items"].append((canonical, 0))
258
- else:
259
- title, views = process_telegram_link(username, mid, canonical)
260
- key = username
261
- if key not in tg_groups:
262
- tg_groups[key] = {"title": title, "items": []}
263
- if not tg_groups[key].get("title") or str(tg_groups[key]["title"]).startswith("**Ошибка"):
264
- tg_groups[key]["title"] = title
265
- tg_groups[key]["items"].append((canonical, views))
266
-
267
- elif plat == "vk":
268
- owner_id, post_id = a, b
269
- if owner_id is None or post_id is None:
270
- key = canonical
271
- vk_groups.setdefault(key, {"title": "VK пост", "items": []})
272
- vk_groups[key]["items"].append((canonical, 0))
273
- else:
274
- title, views = process_vk_link(owner_id, post_id, canonical)
275
- key = str(owner_id)
276
- if key not in vk_groups:
277
- vk_groups[key] = {"title": title, "items": []}
278
- if not vk_groups[key].get("title") or str(vk_groups[key]["title"]).startswith("**Ошибка"):
279
- vk_groups[key]["title"] = title
280
- vk_groups[key]["items"].append((canonical, views))
281
- else:
282
- errors.append(f"Неизвестная платформа: {canonical}")
283
-
284
- tg_total_views = sum(v for g in tg_groups.values() for _, v in g["items"])
285
- vk_total_views = sum(v for g in vk_groups.values() for _, v in g["items"])
286
-
287
- tg_sorted = sorted(
288
- tg_groups.items(),
289
- key=lambda kv: sum(v for _, v in kv[1]["items"]),
290
- reverse=True,
291
- )
292
- vk_sorted = sorted(
293
- vk_groups.items(),
294
- key=lambda kv: sum(v for _, v in kv[1]["items"]),
295
- reverse=True,
296
- )
297
-
298
- html_lines: List[str] = []
299
- text_lines: List[str] = []
300
-
301
- # --- Telegram ---
302
- html_lines.append("<h2>Telegram</h2>")
303
- html_lines.append(f"<b>Суммарный охват постов — {human_format_views(tg_total_views)} просмотров</b>")
304
- text_lines.append("Telegram")
305
- text_lines.append(f"Суммарный охват постов — {human_format_views(tg_total_views)} просмотров")
306
- if tg_sorted:
307
- html_lines.append("<ol>")
308
- for idx, (_, data) in enumerate(tg_sorted, start=1):
309
- title = data["title"] or "Telegram"
310
- items = data["items"]
311
- first_link, _first_views = items[0]
312
-
313
- title_html = _escape(title)
314
- first_link_html = _escape(first_link)
315
- line_html = f'<li><a href="{first_link_html}">{title_html}</a>'
316
- if len(items) > 1:
317
- for link2, _v in items[1:]:
318
- line_html += f' + <a href="{_escape(link2)}">ещё</a>'
319
- views_str = " + ".join(human_format_views(v) for _, v in items)
320
- line_html += f" — {views_str}</li>"
321
- html_lines.append(line_html)
322
-
323
- line_text = f"{idx}. {title} ({first_link})"
324
- if len(items) > 1:
325
- extra_links = ", ".join(link2 for link2, _v in items[1:])
326
- line_text += f" + ещё: {extra_links}"
327
- line_text += f" — {views_str}"
328
- text_lines.append(line_text)
329
- html_lines.append("</ol>")
330
- else:
331
- html_lines.append("<i>Нет ссылок на Telegram</i>")
332
- text_lines.append("Нет ссылок на Telegram")
333
- html_lines.append("<br/>")
334
- text_lines.append("")
335
-
336
- # --- VK ---
337
- html_lines.append("<h2>ВКонтакте</h2>")
338
- html_lines.append(f"<b>Суммарный охват постов — {human_format_views(vk_total_views)} просмотров</b>")
339
- text_lines.append("ВКонтакте")
340
- text_lines.append(f"Суммарный охват постов — {human_format_views(vk_total_views)} просмотров")
341
- if vk_sorted:
342
- html_lines.append("<ol>")
343
- for idx, (_, data) in enumerate(vk_sorted, start=1):
344
- title = data["title"] or "VK пост"
345
- items = data["items"]
346
- first_link, _first_views = items[0]
347
-
348
- title_html = _escape(title)
349
- first_link_html = _escape(first_link)
350
- line_html = f'<li><a href="{first_link_html}">{title_html}</a>'
351
- if len(items) > 1:
352
- for link2, _v in items[1:]:
353
- line_html += f' + <a href="{_escape(link2)}">ещё</a>'
354
- views_str = " + ".join(human_format_views(v) for _, v in items)
355
- line_html += f" — {views_str}</li>"
356
- html_lines.append(line_html)
357
-
358
- line_text = f"{idx}. {title} ({first_link})"
359
- if len(items) > 1:
360
- extra_links = ", ".join(link2 for link2, _v in items[1:])
361
- line_text += f" + ещё: {extra_links}"
362
- line_text += f" — {views_str}"
363
- text_lines.append(line_text)
364
- html_lines.append("</ol>")
365
- else:
366
- html_lines.append("<i>Нет ссылок на ВКонтакте</i>")
367
- text_lines.append("Нет ссылок на ВКонтакте")
368
- html_lines.append("<br/>")
369
- text_lines.append("")
370
-
371
- if errors:
372
- html_lines.append("<h2>Ошибки</h2>")
373
- html_lines.append("<ul>")
374
- for err in errors:
375
- html_lines.append(f"<li>{_escape(err)}</li>")
376
- text_lines.append(f"- {err}")
377
- html_lines.append("</ul>")
378
- html_lines.append("<br/>")
379
- text_lines.append("")
380
-
381
- return "\n".join(html_lines), "\n".join(text_lines), tg_total_views, vk_total_views, errors
382
-
383
-
384
- @app.get("/health")
385
- def health_check():
386
- return {"status": "ok"}
387
-
388
-
389
- @app.post("/parse", response_model=ParseResponse)
390
- def parse_links(payload: ParseRequest):
391
- raw_lines = normalize_links(payload.links)
392
- if not raw_lines:
393
- return ParseResponse(html="", text="", telegram_total=0, vk_total=0, errors=["Нет ссылок для обработки."])
394
-
395
- seen = set()
396
- lines: List[Tuple[str, str, object, object]] = []
397
- for link in raw_lines:
398
- plat = detect_platform(link)
399
- if plat == "telegram":
400
- canonical, username, mid = canonicalize_tg_link(link)
401
- if not canonical:
402
- canonical = link
403
- username = None
404
- mid = None
405
- if canonical in seen:
406
- continue
407
- seen.add(canonical)
408
- lines.append(("telegram", canonical, username, mid))
409
- elif plat == "vk":
410
- canonical, owner_id, post_id = canonicalize_vk_link(link)
411
- if not canonical:
412
- canonical = link
413
- if canonical in seen:
414
- continue
415
- seen.add(canonical)
416
- lines.append(("vk", canonical, owner_id, post_id))
417
- else:
418
- lines.append(("unknown", link, None, None))
419
-
420
- html, text, tg_total, vk_total, errors = build_output(lines)
421
- return ParseResponse(
422
- html=html,
423
- text=text,
424
- telegram_total=tg_total,
425
- vk_total=vk_total,
426
- errors=errors,
427
- )