URl_checks реализация для статуса

page_analyzer/services/check_url.py

@@ -1,5 +1,9 @@
 import logging
 
+import requests
+
+from page_analyzer.services.url import URLService
+
 from page_analyzer.models import URLChecks
 from page_analyzer.services.db import PostgresDB
 
@@ -28,6 +32,10 @@ def check(self, url_id):
  return 'Страница успешно проверена', 'success'
  return 'Произошла ошибка при проверке', 'danger'
 
- @staticmethod
- def check_url(url_id):
- return URLChecks(url_id=url_id)
+ def check_url(self, url_id):
+ url_service = URLService(self.db)
+ url = url_service.get_json_by_id(url_id).name
+ if url:
+ request = requests.get(url)
+ if request:
+ return URLChecks(url_id=url_id, status_code=request.status_code)

page_analyzer/services/url.py

@@ -66,7 +66,7 @@ def get_all_urls(self) -> List[URLModel] | None:
  items = [URLModel(**item['result']) for item in sorted_items]
  return items
 
- def _get_url_id_by_url_name(self, item: URLModel) -> Optional[URLModel]:
+ def get_url_id_by_url_name(self, item: URLModel) -> Optional[URLModel]:
 
  exist_item = self.db.execute_query(GET_JSON_BY_URL, (item.name,), )
  if exist_item:
@@ -79,7 +79,7 @@ def insert_url(self, url: str) -> dict[str, tuple[str, str] | URLModel]:
 
  try:
  item = URLModel(name=url)
- item = self._get_url_id_by_url_name(item)
+ item = self.get_url_id_by_url_name(item)
 
  if item.id:
  message = ('Страница уже существует', 'info')

pyproject.toml

@@ -16,6 +16,7 @@ psycopg-binary = "^3.1.9"
 psycopg-c = "^3.1.9"
 psycopg = "^3.1.9"
 pydantic = "^2.0"
+requests = "^2.31.0"
 
 
 [tool.poetry.group.dev.dependencies]

tests/test_url_service.py

@@ -46,11 +46,11 @@ def test__get_url_id_by_url_name(app):
  db = get_db()
  url_service = URLService(db=db)
  item = URLModel(name='https://chatbot.theb.ai')
- data = url_service._get_url_id_by_url_name(item)
+ data = url_service.get_url_id_by_url_name(item)
  assert not data.id
 
  item = URLModel(name='http:https://www.ya.ru/')
- data = url_service._get_url_id_by_url_name(item)
+ data = url_service.get_url_id_by_url_name(item)
  assert data.id
 
 

tests/test_views.py

@@ -76,6 +76,7 @@ def test_post_url_check(client):
 
  response = client.post('/urls/1/checks', follow_redirects=True)
  assert '5' in response.text
+ assert '200' in response.text
 
  response = client.post('/urls/1/checks', follow_redirects=True)
  assert '6' in response.text