from django.core.management import BaseCommand
from vacancies.main.models import Vacancy
import clickhouse_connect
from vacancies.main.vector_store import add_vacancy_vectors
from vacancies.main.features_extractor import extract_vacancy_features

clickhouse_client = clickhouse_connect.create_client(host="127.0.0.1", port=18123)

query = """
SELECT id, chat_username, telegram_id, message, timestamp
FROM telegram_parser_chatmessage
WHERE timestamp >= now() - INTERVAL 30 DAY
  AND length(message) > 150
  AND arrayCount(x -> position(message, x) > 0, [
    'вакансия', 'ищем', 'требуется', 'разработчик', 'будет плюсом',
    'зарплата', 'оклад', 'з/п', 'руб', 'опыт работы',
    'требования', 'обязанности', 'условия', 'компания', 'офис',
    'удаленно', 'гибкий график', 'полный день', 'частичная занятость',
    'резюме', 'собеседование', 'junior', 'middle', 'senior'
  ]) >= 5 AND position(message, 'О себе') = 0 AND position(message, 'Обо мне') = 0 AND position(message, '#ищу') = 0
"""


class Command(BaseCommand):
    help = "Collect vacancies from telegram messages"

    def handle(self, *args, **options):
        for index, row in enumerate(clickhouse_client.query(query).result_rows):
            (id, chat_username, telegram_id, message, timestamp) = row

            link = f"https://t.me/{chat_username}/{telegram_id}"
            vacancy, created = Vacancy.objects.get_or_create(
                link=link,
                defaults={'content': message}
            )

            print(index, link)
            add_vacancy_vectors(vacancy.id, extract_vacancy_features(message).model_dump(), {"link": link})