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)
|
||||
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="Запуск процеса парсинга любого источника")
|
||||
async def process_parser_all_ist(url: Parserall, background_tasks: BackgroundTasks):
|
||||
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:
|
||||
pass
|
||||
|
||||
# Создание и работа с таблицей источников add_sources
|
||||
# Создание и работа с таблицей источников sources
|
||||
def create_table_add_sourse():
|
||||
conn = get_connection()
|
||||
try:
|
||||
@@ -257,6 +257,17 @@ def add_sources(url: str, promt: str):
|
||||
finally:
|
||||
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__":
|
||||
# # create_table_config_gpt() # <-- раскомментировать эту строку
|
||||
|
||||
Reference in New Issue
Block a user