This commit is contained in:
4
main.py
4
main.py
@@ -476,6 +476,10 @@ async def add_sources_all_ist(sources: Parserall):
|
|||||||
result = wp.add_sources(str(sources.url), sources.promt)
|
result = wp.add_sources(str(sources.url), sources.promt)
|
||||||
return {"status": "success", "message": "Источник добавлен", "data": result}
|
return {"status": "success", "message": "Источник добавлен", "data": result}
|
||||||
|
|
||||||
|
@app.get("/all_sources", summary="Метод получения всех источников")
|
||||||
|
async def get_all_sources():
|
||||||
|
return wp.get_all_sources()
|
||||||
|
|
||||||
@app.post("/parser_all" , summary="Запуск процеса парсинга любого источника")
|
@app.post("/parser_all" , summary="Запуск процеса парсинга любого источника")
|
||||||
async def process_parser_all_ist(url: Parserall, background_tasks: BackgroundTasks):
|
async def process_parser_all_ist(url: Parserall, background_tasks: BackgroundTasks):
|
||||||
background_tasks.add_task(start_pars_all_istochnik(str(url.url), url.promt))
|
background_tasks.add_task(start_pars_all_istochnik(str(url.url), url.promt))
|
||||||
|
|||||||
@@ -228,7 +228,7 @@ def check_error_url(error_sources_url: str) -> bool:
|
|||||||
finally:
|
finally:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
# Создание и работа с таблицей источников add_sources
|
# Создание и работа с таблицей источников sources
|
||||||
def create_table_add_sourse():
|
def create_table_add_sourse():
|
||||||
conn = get_connection()
|
conn = get_connection()
|
||||||
try:
|
try:
|
||||||
@@ -257,6 +257,17 @@ def add_sources(url: str, promt: str):
|
|||||||
finally:
|
finally:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
def get_all_sources():
|
||||||
|
"""Возвращает все записи из таблицы sourse"""
|
||||||
|
conn = get_connection()
|
||||||
|
try:
|
||||||
|
with conn.cursor(cursor_factory=RealDictCursor) as cur:
|
||||||
|
cur.execute("SELECT * FROM sourse")
|
||||||
|
rows = cur.fetchall()
|
||||||
|
sources = [{"url": row["url"], "promt": row["promt"]} for row in rows]
|
||||||
|
return {"sources": sources}
|
||||||
|
finally:
|
||||||
|
pass
|
||||||
# Пример использования
|
# Пример использования
|
||||||
# if __name__ == "__main__":
|
# if __name__ == "__main__":
|
||||||
# # create_table_config_gpt() # <-- раскомментировать эту строку
|
# # create_table_config_gpt() # <-- раскомментировать эту строку
|
||||||
|
|||||||
Reference in New Issue
Block a user