AvitoParser.py

import configparser
import re
import threading
import time

import flet as ft
from loguru import logger
from notifiers.logging import NotificationHandler


from lang import *
from parser_cls import AvitoParse
from version import VERSION


def main(page: ft.Page):
    page.title = f'Parser Avito v {VERSION}'
    page.theme_mode = ft.ThemeMode.DARK
    page.vertical_alignment = ft.MainAxisAlignment.CENTER
    page.window.width = 1000
    page.window.height = 950
    page.window.min_width = 650
    page.window.min_height = 500
    page.padding = 20
    tg_logger_init = False
    config = configparser.ConfigParser()
    config.read("settings.ini", encoding='utf-8')
    is_run = False
    stop_event = threading.Event()

    def set_up():
        """Работа с настройками"""
        nonlocal config
        try:
            """Багфикс возможных проблем с экранированием"""
            url_input.value = "\n".join(config["Avito"]["URL"].split(","))
        except Exception as err:
            logger.debug(f"Ошибка url при открытии конфига: {err}")
            with open('settings.ini') as file:
                line_url = file.readlines()[1]
                regex = r"http.+"
                all_links = re.findall(regex, line_url)
                if all_links:
                    url_input.value = "\n".join(all_links)
                url_input.value = re.findall(regex, line_url)[0]
        tg_chat_id.value = "\n".join(config["Avito"]["CHAT_ID"].split(","))
        tg_token.value = config["Avito"]["TG_TOKEN"]
        count_page.value = config["Avito"]["NUM_ADS"]
        pause_sec.value = config["Avito"]["FREQ"]
        max_view.value = config["Avito"].get("MAX_VIEW")
        keyword_input.value = "\n".join(config["Avito"]["KEYS"].split(","))
        black_keyword_input.value = "\n".join(config["Avito"].get("KEYS_BLACK", "").split(","))
        max_price.value = config["Avito"].get("MAX_PRICE")
        min_price.value = config["Avito"].get("MIN_PRICE", "0")
        geo.value = config["Avito"].get("GEO", "")
        proxy.value = config["Avito"].get("PROXY", "")
        proxy_change_ip.value = config["Avito"].get("PROXY_CHANGE_IP", "")
        need_more_info.value = True if config["Avito"].get("NEED_MORE_INFO", "0") == "1" else False
        debug_mode.value = True if config["Avito"].get("DEBUG_MODE", "0") == "1" else False
        fast_speed.value = True if config["Avito"].get("FAST_SPEED", "0") == "1" else False
        page.update()

    def save_config():
        """Сохраняет конфиг"""
        config["Avito"]["TG_TOKEN"] = tg_token.value
        config["Avito"]["CHAT_ID"] = ",".join(tg_chat_id.value.split())
        config["Avito"]["URL"] = ",".join(str(url_input.value).replace('%', '%%').split())  # bugfix
        config["Avito"]["NUM_ADS"] = count_page.value
        config["Avito"]["FREQ"] = pause_sec.value
        config["Avito"]["KEYS"] = ",".join(keyword_input.value.split("\n"))
        config["Avito"]["KEYS_BLACK"] = ",".join(black_keyword_input.value.split("\n"))
        config["Avito"]["MAX_PRICE"] = max_price.value
        config["Avito"]["MIN_PRICE"] = min_price.value
        config["Avito"]["MAX_VIEW"] = max_view.value
        config["Avito"]["GEO"] = geo.value
        config["Avito"]["PROXY"] = proxy.value
        config["Avito"]["PROXY_CHANGE_IP"] = proxy_change_ip.value
        config["Avito"]["NEED_MORE_INFO"] = "1" if need_more_info.value else "0"
        config["Avito"]["DEBUG_MODE"] = "1" if debug_mode.value else "0"
        config["Avito"]["FAST_SPEED"] = "1" if fast_speed.value else "0"
        with open('settings.ini', 'w', encoding='utf-8') as configfile:
            config.write(configfile)
        logger.debug("Настройки сохранены")

    def check_tg_btn(e):
        if len(tg_chat_id.value) > 5 and len(tg_token.value) > 10:
            btn_test_tg.disabled = False
        else:
            btn_test_tg.disabled = True
        page.update()

    def close_dlg(e):
        dlg_modal_proxy.open = False
        page.update()

    def logger_console_init():
        logger.add(logger_console_widget, format="{time:HH:mm:ss} - {message}")

    def logger_console_widget(message):
        console_widget.value += message
        page.update()

    def logger_tg():
        """Логирование в telegram"""
        nonlocal tg_logger_init
        token = tg_token.value
        chat_ids = tg_chat_id.value.split()

        if tg_logger_init:
            return

        if token and chat_ids:
            for chat_id in chat_ids:
                params = {
                    'token': token,
                    'chat_id': chat_id,
                    'parse_mode': 'markdown'
                }
                tg_handler = NotificationHandler("telegram", defaults=params)

                """Все логи уровня SUCCESS и выше отсылаются в телегу"""
                logger.add(tg_handler, level="SUCCESS", format="{message}")

            tg_logger_init = True
            return None

        logger.info("Данные для отправки в telegram не заполнены. Результат будет сохранен в файл и выведен здесь")

    def telegram_log_test(e):
        """Тестирование отправки сообщения в telegram"""
        logger_tg()
        token = tg_token.value
        chat_id = tg_chat_id.value
        if all([token, chat_id]):
            logger.success('test')

            logger.info(TG_TEST_MSG)
            return None
        logger.info("Должны быть заполнены поля ТОКЕН TELEGRAM и CHAT ID TELEGRAM")

    dlg_modal_proxy = ft.AlertDialog(
        modal=True,
        title=ft.Text("Подробнее насчёт прокси"),
        content=ft.Container(
            content=ft.Text(BUY_PROXY_LINK, size=20),
            width=600,
            height=400,
            padding=10
        ),
        actions=[
            ft.TextButton("Купить прокси",
                          on_click=lambda e: page.launch_url(
                              "https://mobileproxy.space/user.html?buyproxy&coupons_code=eMy-r4y-FZE-kMu")),
            ft.TextButton("Отмена", on_click=close_dlg),

        ],
        actions_alignment=ft.MainAxisAlignment.END,
        on_dismiss=lambda e: print("Modal dialog dismissed!"),
    )

    def open_dlg_modal(e):
        page.overlay.append(dlg_modal_proxy)
        dlg_modal_proxy.open = True
        page.update()

    def start_parser(e):
        nonlocal is_run
        logger.info("Старт")
        stop_event.clear()
        save_config()
        console_widget.height = 700
        input_fields.visible = False
        start_btn.visible = False
        stop_btn.visible = True
        is_run = True
        page.update()
        while is_run and not stop_event.is_set():
            run_process()
            if not is_run:
                return
            logger.info("Пауза между повторами")
            for _ in range(int(pause_sec.value if pause_sec.value else 300)):
                time.sleep(1)
                if not is_run:
                    logger.info("Завершено")
                    start_btn.text = "Старт"
                    start_btn.disabled = False
                    page.update()
                    return

    def stop_parser(e):
        nonlocal is_run
        stop_event.set()
        logger.debug("Стоп")
        is_run = False
        console_widget.height = 100
        input_fields.visible = True
        stop_btn.visible = False
        start_btn.visible = True
        start_btn.text = "Останавливаюсь..."
        start_btn.disabled = True
        page.update()

    def required_field_for_more_info(e):
        if geo.value or (max_view.value and max_view.value != "0"):
            need_more_info.value = True
        page.update()

    def run_process():
        logger_tg()
        parser = AvitoParse(
            stop_event=stop_event,
            url=url_input.value.split(),
            count=int(count_page.value),
            keysword_list=keyword_input.value.split("\n") if keyword_input.value else None,
            keysword_black_list=black_keyword_input.value.split("\n") if black_keyword_input.value else None,
            max_price=int(max_price.value),
            min_price=int(min_price.value),
            geo=geo.value,
            proxy=proxy.value,
            proxy_change_url=proxy_change_ip.value,
            debug_mode=debug_mode.value,
            need_more_info=need_more_info.value,
            fast_speed=fast_speed.value,
            max_views=max_view.value if max_view.value and max_view.value != "0" else None
        )
        parsing_thread = threading.Thread(target=parser.parse)
        parsing_thread.start()
        parsing_thread.join()
        start_btn.disabled = False
        start_btn.text = "Старт"
        page.update()

    label_required = ft.Text("Обязательные параметры", size=20)
    url_input = ft.TextField(
        label="Вставьте начальную ссылку или ссылки. Используйте Enter между значениями",
        multiline=True,
        min_lines=1,
        max_lines=50,
        expand=True,
        tooltip=URL_INPUT_HELP,
    )
    min_price = ft.TextField(label="Минимальная цена", width=400, expand=True, tooltip=MIN_PRICE_HELP)
    max_price = ft.TextField(label="Максимальная цена", width=400, expand=True, tooltip=MAX_PRICE_HELP)
    label_not_required = ft.Text("Дополнительные параметры")
    keyword_input = ft.TextField(
        label="Ключевые слова (через Enter)",
        multiline=True,
        min_lines=1,
        max_lines=5,
        width=400,
        expand=True,
        tooltip=KEYWORD_INPUT_HELP
    )
    black_keyword_input = ft.TextField(
        label="Черный список ключевых слов (через Enter)",
        multiline=True,
        min_lines=1,
        max_lines=5,
        width=400,
        expand=True,
        tooltip=KEYWORD_BLACK_INPUT_HELP
    )
    count_page = ft.TextField(label="Количество страниц", width=400, expand=True, tooltip=COUNT_PAGE_HELP)
    pause_sec = ft.TextField(label="Пауза в секундах между повторами", width=400, expand=True, tooltip=PAUSE_SEC_HELP)
    max_view = ft.TextField(label="Макс. просмотров", width=400, expand=True, tooltip=MAX_VIEW_HELP,
                            on_change=required_field_for_more_info)
    tg_token = ft.TextField(label="Token telegram", width=400, on_change=check_tg_btn, expand=True,
                            tooltip=TG_TOKEN_HELP)
    tg_chat_id = ft.TextField(label="Chat id telegram. Можно несколько", width=400, on_change=check_tg_btn,
                              multiline=True, expand=True, tooltip=TG_CHAT_ID_HELP)
    btn_test_tg = ft.ElevatedButton(text="Проверить tg", disabled=True, on_click=telegram_log_test, expand=True,
                                    tooltip=BTN_TEST_TG_HELP)
    proxy = ft.TextField(label="Прокси в формате username:password@server:port", width=400, expand=True,
                         tooltip=PROXY_HELP)
    proxy_change_ip = ft.TextField(
        label="Ссылка для изменения IP, в формате https://changeip.mobileproxy.space/?proxy_key=***", width=400,
        expand=True, tooltip=PROXY_CHANGE_IP_HELP)
    proxy_btn_help = ft.ElevatedButton(text="Подробнее про прокси", on_click=open_dlg_modal, expand=True,
                                       tooltip=PROXY_BTN_HELP_HELP)
    geo = ft.TextField(label="Ограничение по городу", width=400, on_change=required_field_for_more_info, expand=True,
                       tooltip=GEO_HELP)
    start_btn = ft.FilledButton("Старт", width=800, on_click=start_parser, expand=True)
    stop_btn = ft.OutlinedButton("Стоп", width=980, on_click=stop_parser, visible=False,
                                 style=ft.ButtonStyle(bgcolor=ft.colors.RED_400), expand=True)
    console_widget = ft.Text(width=800, height=100, color=ft.colors.GREEN, value="", selectable=True,
                             expand=True)  # , bgcolor=ft.colors.GREY_50)
    need_more_info = ft.Checkbox("Дополнительная информация", on_change=required_field_for_more_info,
                                 tooltip=NEED_MORE_INFO_HELP)
    debug_mode = ft.Checkbox("Режим отладки", tooltip=DEBUG_MODE_HELP)
    fast_speed = ft.Checkbox("Ускорить", tooltip=SKIP_JS_HELP)
    buy_me_coffe_btn = ft.TextButton("Донат автору на пиво",
                                     on_click=lambda e: page.launch_url("https://yoomoney.ru/to/410014382689862"),
                                     style=ft.ButtonStyle(color=ft.colors.GREEN_300), expand=True,
                                     tooltip=BUY_ME_COFFE_BTN_HELP)
    report_issue_btn = ft.TextButton("Сообщить о проблеме", on_click=lambda e: page.launch_url(
        "https://github.com/Duff89/parser_avito/issues"), style=ft.ButtonStyle(color=ft.colors.GREY), expand=True,
                                     tooltip=REPORT_ISSUE_BTN_HELP)
    input_fields = ft.Column(
        [
            label_required,
            url_input,
            ft.Row(
                [min_price, max_price],
                alignment=ft.MainAxisAlignment.CENTER,
                spacing=0
            ),
            ft.Text(""),
            label_not_required,

            ft.Row(
                [keyword_input, black_keyword_input],
                alignment=ft.MainAxisAlignment.CENTER,
                spacing=0
            ),
            ft.Row(
                [count_page, pause_sec],
                alignment=ft.MainAxisAlignment.CENTER,
                spacing=0
            ),
            ft.Row(
                [geo, max_view],
                alignment=ft.MainAxisAlignment.CENTER,
                spacing=0
            ),
            ft.Row(
                [tg_token, tg_chat_id],
                alignment=ft.MainAxisAlignment.CENTER,
                spacing=0
            ),
            btn_test_tg,
            ft.Row(
                [proxy, proxy_change_ip],
                alignment=ft.MainAxisAlignment.CENTER,
                spacing=0
            ),
            proxy_btn_help,
            ft.Text(""),
            ft.Row(
                [need_more_info, debug_mode, fast_speed],
                alignment=ft.MainAxisAlignment.CENTER,
                spacing=0
            ),

        ],
        expand=True,
        alignment=ft.MainAxisAlignment.CENTER,
        horizontal_alignment=ft.CrossAxisAlignment.CENTER,

    )

    controls = ft.Column(
        [console_widget,
         start_btn,
         stop_btn],
        expand=True,
        alignment=ft.MainAxisAlignment.CENTER,
        horizontal_alignment=ft.CrossAxisAlignment.CENTER
    )
    other_btn = ft.Row([buy_me_coffe_btn, report_issue_btn], expand=True, alignment=ft.MainAxisAlignment.CENTER)
    all_field = ft.Column([input_fields, controls, other_btn], alignment=ft.MainAxisAlignment.CENTER,
                          horizontal_alignment=ft.CrossAxisAlignment.CENTER)

    def start_page():
        page.add(ft.Column(
            [all_field],
            expand=True,
            alignment=ft.MainAxisAlignment.CENTER,
            horizontal_alignment=ft.CrossAxisAlignment.CENTER,
            scroll=ft.ScrollMode.AUTO
        ))

    set_up()
    start_page()
    logger_console_init()


ft.app(
    target=main,
)