fix names table

Author: gordienkoas

.env

@@ -1,2 +1,2 @@
 SECRET_KEY="SECRETFORMYPROJECT007"
-DATABASE_URL=postgresql://tirion:1234@localhost:5432/mydb
+DATABASE_URL=postgresql://tirion:1234@localhost:5432/mydb2

page_analyzer/app.py

@@ -5,8 +5,7 @@
 from dotenv import load_dotenv
 import validators
 from bs4 import BeautifulSoup
-
-
+from urllib.parse import urlparse
 
 load_dotenv()
 DATABASE_URL = os.getenv('DATABASE_URL')
@@ -28,28 +27,28 @@ def index():
             flash('Некорректный URL', 'error')
             return redirect(url_for('index'))
 
+        # Извлекаем домен из URL
+        domain = urlparse(url).netloc
+
         conn = get_db_connection()
         cur = conn.cursor()
         try:
-            # Проверяем, существует ли уже этот URL
-            cur.execute('SELECT id FROM urls WHERE name = %s', (url,))
+            # Проверяем, существует ли уже этот домен
+            cur.execute('SELECT id FROM urls WHERE name = %s', (domain,))
             existing_url = cur.fetchone()
 
             if existing_url:
-                # Если URL уже существует, перенаправляем на его детали
+                # Если домен уже существует, перенаправляем на его детали
                 url_id = existing_url[0]
                 flash('Страница уже существует', 'error')
                 return redirect(url_for('url_detail', url_id=url_id))
 
-            # Если URL не существует, добавляем его
-            cur.execute('INSERT INTO urls (name) VALUES (%s)', (url,))
+            # Если домен не существует, добавляем его
+            cur.execute('INSERT INTO urls (name) VALUES (%s) RETURNING id', (domain,))
+            url_id = cur.fetchone()[0]
             conn.commit()
             flash('Страница успешно добавлена', 'success')
 
-            # Получаем ID добавленного URL для перенаправления
-            cur.execute('SELECT id FROM urls WHERE name = %s', (url,))
-            url_id = cur.fetchone()[0]
-
             return redirect(url_for('url_detail', url_id=url_id))
         except Exception as e:
             conn.rollback()
@@ -68,18 +67,18 @@ def urls():
 
     # Объединяем запросы для получения всех URL и последней проверки
     cur.execute('''
-        SELECT u.id, u.name, u.created_at, 
-               uc.status_code, uc.created_at AS last_check,
-               uc.h1, uc.title, uc.description
-        FROM urls u
-        LEFT JOIN url_checks uc ON u.id = uc.url_id
-        AND uc.created_at = (
-            SELECT MAX(created_at) 
-            FROM url_checks 
-            WHERE url_id = u.id
-        )
-        ORDER BY u.created_at DESC
-    ''')
+         SELECT u.id, u.name, u.created_at, 
+                uc.status_code, uc.created_at AS last_check,
+                uc.h1, uc.title, uc.description
+         FROM urls u
+         LEFT JOIN url_checks uc ON u.id = uc.url_id
+         AND uc.created_at = (
+             SELECT MAX(created_at) 
+             FROM url_checks 
+             WHERE url_id = u.id
+         )
+         ORDER BY u.created_at DESC
+     ''')
     urls = cur.fetchall()
 
     cur.close()
@@ -104,6 +103,8 @@ def url_detail(url_id):
     return render_template('result.html', url=url, checks=checks)
 
 
+from urllib.parse import urlparse
+
 @app.route('/check_url/<int:url_id>', methods=['POST'])
 def add_check(url_id):
     conn = get_db_connection()
@@ -113,11 +114,16 @@ def add_check(url_id):
     url = cur.fetchone()
 
     if url is None:
-        flash('URL не найден.')
+        flash('URL не найден.', 'error')
         return redirect(url_for('urls'))
 
     url_name = url[0]
 
+    # Проверка и добавление схемы к URL, если она отсутствует
+    parsed_url = urlparse(url_name)
+    if not parsed_url.scheme:
+        url_name = 'http://' + url_name  # Добавляем http по умолчанию
+
     try:
         response = requests.get(url_name)
         response.raise_for_status()  # Проверка на ошибки HTTP
@@ -134,24 +140,26 @@ def add_check(url_id):
         title_content = title_tag.text if title_tag else None
         description_content = description_tag['content'] if description_tag else None
 
-
         # Записываем код статуса и SEO-данные в базу данных
-
         cur.execute('''
             INSERT INTO url_checks (url_id, status_code, h1, title, description) 
             VALUES (%s, %s, %s, %s, %s)
-            ''', (url_id, response.status_code, h1_content, title_content, description_content))
+        ''', (url_id, response.status_code, h1_content, title_content, description_content))
+
         conn.commit()
-        flash('Страница успешно проверена')
-    except requests.exceptions.RequestException:
-        flash('Произошла ошибка при проверке.')
+        flash('Страница успешно проверена', 'success')
+    except Exception as e:
+        conn.rollback()
+        flash('Произошла ошибка при проверке', 'error')
     finally:
         cur.close()
         conn.close()
 
     return redirect(url_for('url_detail', url_id=url_id))
 
 
+
+
 if __name__ == '__main__':
     app.run(debug=True)