KAnanev/Page-analyzer

from flask import Blueprint, render_template, request, flash, redirect, url_for

from page_analyzer.db import get_db
from page_analyzer.services.check_url import CheckURLService
from page_analyzer.services.url import URLService

bp = Blueprint('page_analyzer', __name__)


@bp.get('/')
def index():
    """Главная страница."""

    return render_template('index.html')


@bp.get('/urls')
def get_urls():
    """Страница со всеми url."""

    db = get_db()
    url_service = URLService(db=db)
    items = url_service.get_all_urls()

    return render_template('urls.html', items=items)


@bp.get('/urls/<int:url_id>')
def get_url(url_id):
    """Страница c url с выдачей по id."""

    db = get_db()
    url_service = URLService(db=db)
    item = url_service.get_json_by_id(url_id)
    return render_template('url.html', item=item)


@bp.post('/urls')
def post_url():
    """ Проверяет есть ли url в базе,
    добавляет в базу или берет из базы существующий,
    редиректит на страницу с url."""

    url = request.form['url']
    db = get_db()
    url_service = URLService(db=db)
    result = url_service.insert_url(url)
    flash(*result['message'])
    if result['item']:
        return redirect(
            url_for(
                'page_analyzer.get_url',
                url_id=result['item'].id),
        )
    return redirect(url_for('index'))


@bp.post('/urls/<int:url_id>/checks')
def post_url_check(url_id):
    """Обработчик маршрута проверки"""

    db = get_db()
    check_url_service = CheckURLService(db=db)

    result_message = check_url_service.check(url_id)
    flash(*result_message)
    return redirect(
        url_for(
            'page_analyzer.get_url',
            url_id=url_id),
    )