mirror of
https://gitverse.ru/kpa39l/vesti.git
synced 2026-09-29 01:50:03 +00:00
kpa39l
43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
# VESTI — фаза 1: сбор кандидатов с источников в очередь (posts).
|
|
# Запускает telegram_crawler и rss_crawler по включённым источникам;
|
|
# классификация НЕ выполняется — новые посты уходят в очередь (classified=0),
|
|
# которую разбирает crawler.worker (фаза 3, параллельно).
|
|
#
|
|
# Запуск: python -m crawler.crawl_sources --all
|
|
# python -m crawler.crawl_sources --crawler rss # только RSS
|
|
import argparse
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
BASE = Path(__file__).resolve().parent.parent
|
|
PY = str(BASE / ".venv" / "bin" / "python")
|
|
|
|
|
|
def run_crawler(mod: str, extra: list[str] | None = None) -> int:
|
|
"""Запускает модуль-краулер в подпроцессе. Возвращает exit code."""
|
|
cmd = [PY, "-m", mod] + (extra or [])
|
|
print(f"--- {mod} {' '.join(extra or [])} ---", flush=True)
|
|
return subprocess.run(cmd, cwd=str(BASE)).returncode
|
|
|
|
|
|
def main():
|
|
ap = argparse.ArgumentParser(description="VESTI фаза 1: сбор кандидатов с источников в очередь")
|
|
ap.add_argument("--all", action="store_true", help="сбор со всех включённых краулеров")
|
|
ap.add_argument("--crawler", choices=["telegram", "rss"], help="только один тип краулера")
|
|
args = ap.parse_args()
|
|
|
|
if not args.all and not args.crawler:
|
|
ap.print_help()
|
|
sys.exit(2)
|
|
|
|
rc = 0
|
|
if args.all or args.crawler == "telegram":
|
|
rc |= run_crawler("crawler.telegram_crawler", ["--all"])
|
|
if args.all or args.crawler == "rss":
|
|
rc |= run_crawler("crawler.rss_crawler", ["--all"])
|
|
sys.exit(rc)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |