fix answer url

Author: gordienkoas

page_analyzer/app.py

@@ -22,49 +22,46 @@ def get_db_connection():
 @app.route('/', methods=['GET', 'POST'])
 def index():
     if request.method == 'POST':
-        url = request.form.get('url')
+        url = request.form['url']
         return redirect(url_for('urls', url=url))
     return render_template('index.html')
 
 @app.route('/urls', methods=['GET', 'POST'])
 def urls():
     if request.method == 'POST':
-        url = request.form.get('url')
-
-        if not validators.url(url) or len(url) > 255:
-            flash('Некорректный URL', 'error')
-            return redirect(url_for('index')), 422  # Возвращаем ошибку 422 с переадресацией на главную страницу
-
-        domain = urlparse(url).netloc
+        url = request.form['url']
+        return redirect(url_for('urls', url=url))
+    else:
+        url = request.args.get('url')
+        if url:
+            if not validators.url(url) or len(url) > 255:
+                flash('Некорректный URL', 'error')
+                return render_template('index.html'), 422  # Возвращаем ошибку 422 с HTML
 
-        with get_db_connection() as conn:
-            with conn.cursor() as cur:
-                try:
-                    cur.execute('SELECT id FROM urls WHERE name = %s', (domain,))
-                    existing_url = cur.fetchone()
+            domain = urlparse(url).netloc
 
-                    if existing_url:
-                        url_id = existing_url[0]
-                        flash('Страница уже существует', 'error')
-                        return redirect(url_for('url_detail', url_id=url_id))
+            with get_db_connection() as conn:
+                with conn.cursor() as cur:
+                    try:
+                        cur.execute('SELECT id FROM urls WHERE name = %s', (domain,))
+                        existing_url = cur.fetchone()
 
-                    cur.execute('INSERT INTO urls (name) VALUES (%s) RETURNING id', (domain,))
-                    url_id = cur.fetchone()[0]
-                    conn.commit()
-                    flash('Страница успешно добавлена', 'success')
+                        if existing_url:
+                            url_id = existing_url[0]
+                            flash('Страница уже существует', 'error')
+                            return redirect(url_for('url_detail', url_id=url_id))
 
-                    return redirect(url_for('url_detail', url_id=url_id))
-                except Exception as e:
-                    conn.rollback()
-                    flash(f'Произошла ошибка: {str(e)}', 'error')
-                    return redirect(url_for('index')), 500  # Возвращаем ошибку 500 с переадресацией на главную страницу
+                        cur.execute('INSERT INTO urls (name) VALUES (%s) RETURNING id', (domain,))
+                        url_id = cur.fetchone()[0]
+                        conn.commit()
+                        flash('Страница успешно добавлена', 'success')
 
-    else:
-        # Обработка GET-запроса
-        url = request.args.get('url')
-        if url:
-            return redirect(url_for('urls'))  # Если URL передан, просто перенаправляем на тот же маршрут
+                        return redirect(url_for('url_detail', url_id=url_id))
+                    except Exception as e:
+                        conn.rollback()
+                        flash(f'Произошла ошибка: {str(e)}', 'error')
 
+        # Если URL не передан, или если запрос POST без URL, получаем все URL из БД
         with get_db_connection() as conn:
             with conn.cursor() as cur:
                 cur.execute('''
@@ -84,7 +81,6 @@ def urls():
 
 
 
-
 @app.route('/urls/<int:url_id>', methods=['GET'])
 def url_detail(url_id):
     conn = get_db_connection()