diff --git a/main.py b/main.py index bfb80f4..46906a8 100644 --- a/main.py +++ b/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)) diff --git a/work_parser.py b/work_parser.py index a0ce9fe..3b1d698 100644 --- a/work_parser.py +++ b/work_parser.py @@ -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() # <-- раскомментировать эту строку