diff --git a/main.py b/main.py index 421d263..adec069 100644 --- a/main.py +++ b/main.py @@ -233,7 +233,7 @@ def gpt_response_message(content: str, name_promt: str): # Общие функции проверки ссылок def check_url(url): try: - response = requests.get('http://45.129.78.228:8002/check_url_exists', params={'url': url}) + response = wp.check_url_exists(url) #get('http://45.129.78.228:8002/check_url_exists', params={'url': url}) if response.status_code == 200: result = response.json() print(result["exists"]) diff --git a/work_parser.py b/work_parser.py index 01a9ae6..9336aa4 100644 --- a/work_parser.py +++ b/work_parser.py @@ -21,13 +21,23 @@ def get_connection(): conn = psycopg2.connect(**DB_CONFIG) conn.autocommit = True return conn +# Проверяет, есть ли указанный URL в базе данных. +def check_url_exists(url: str): + conn = get_connection() + try: + conn = get_connection() + with conn.cursor() as cursor: + cursor.execute( + "SELECT 1 FROM url WHERE url = %s LIMIT 1", + (url,) + ) + result = cursor.fetchone() + return {"exists": bool(result)} + except Exception as e: + raise HTTPException(status_code=500, detail=f"Ошибка при проверке: {e}") + finally: + pass -def close_connection(): - """Закрывает подключение к БД""" - global conn - if conn: - conn.close() - conn = None # работа с базой данных показывания задач work_parser def create_table(): conn = get_connection()