перевел chek_url в базу данных проекта, то бы не взаимодействовать со сторонним сервисом
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
2026-04-11 20:27:48 +10:00
parent 9fd823d8d1
commit be7b042e7c
2 changed files with 17 additions and 7 deletions

View File

@@ -233,7 +233,7 @@ def gpt_response_message(content: str, name_promt: str):
# Общие функции проверки ссылок # Общие функции проверки ссылок
def check_url(url): def check_url(url):
try: 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: if response.status_code == 200:
result = response.json() result = response.json()
print(result["exists"]) print(result["exists"])

View File

@@ -21,13 +21,23 @@ def get_connection():
conn = psycopg2.connect(**DB_CONFIG) conn = psycopg2.connect(**DB_CONFIG)
conn.autocommit = True conn.autocommit = True
return conn 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 # работа с базой данных показывания задач work_parser
def create_table(): def create_table():
conn = get_connection() conn = get_connection()