This commit is contained in:
9
main.py
9
main.py
@@ -64,7 +64,7 @@ app.add_middleware(
|
|||||||
allow_credentials=True,
|
allow_credentials=True,
|
||||||
allow_methods=["*"],
|
allow_methods=["*"],
|
||||||
allow_headers=["*"],
|
allow_headers=["*"],
|
||||||
expose_headers=["*"],
|
# expose_headers=["*"],
|
||||||
)
|
)
|
||||||
|
|
||||||
PROXIES_URL = "https://raw.githubusercontent.com/TheSpeedX/SOCKS-List/master/http.txt"
|
PROXIES_URL = "https://raw.githubusercontent.com/TheSpeedX/SOCKS-List/master/http.txt"
|
||||||
@@ -472,6 +472,7 @@ class Parserall(BaseModel):
|
|||||||
|
|
||||||
@app.post("/add_sources" , summary="Добавление парсинга любого источника")
|
@app.post("/add_sources" , summary="Добавление парсинга любого источника")
|
||||||
async def add_sources_all_ist(sources: Parserall):
|
async def add_sources_all_ist(sources: Parserall):
|
||||||
|
print(f"Добавление источника: {sources.url} с промтом: {sources.promt}")
|
||||||
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}
|
||||||
|
|
||||||
@@ -480,12 +481,10 @@ async def process_parser_all_ist(url: Parserall, background_tasks: BackgroundTas
|
|||||||
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))
|
||||||
return {"message": "Процесс парсинга любого источника запущен"}
|
return {"message": "Процесс парсинга любого источника запущен"}
|
||||||
|
|
||||||
# GET метод для получения
|
|
||||||
@app.get("/get_tasks_offset", summary="Метод получения задач парсинга")
|
@app.get("/get_tasks_offset", summary="Метод получения задач парсинга")
|
||||||
async def get_tasks_offset(limit: int = Query(10, gt=0), offset: int = Query(0, ge=0)):
|
async def get_tasks_offset(limit: int = Query(10, gt=0), offset: int = Query(0, ge=0)):
|
||||||
return wp.get_tasks_offset(limit, offset)
|
return wp.get_tasks_offset(limit, offset)
|
||||||
|
|
||||||
# GET метод для получения настроек
|
|
||||||
@app.get("/settings", summary="Метод получения настроек парсера")
|
@app.get("/settings", summary="Метод получения настроек парсера")
|
||||||
async def get_settings():
|
async def get_settings():
|
||||||
return wp.get_all_promt()
|
return wp.get_all_promt()
|
||||||
@@ -614,5 +613,5 @@ async def get_logs():
|
|||||||
lines = file.readlines()[-10:] # последние 10 строк
|
lines = file.readlines()[-10:] # последние 10 строк
|
||||||
return {"logs": lines}
|
return {"logs": lines}
|
||||||
|
|
||||||
# if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
# uvicorn.run("main:app", port=8001, reload=True)
|
uvicorn.run("main:app", port=8001, reload=True)
|
||||||
|
|||||||
@@ -248,7 +248,7 @@ def add_sources(url: str, promt: str):
|
|||||||
try:
|
try:
|
||||||
with conn.cursor() as cur:
|
with conn.cursor() as cur:
|
||||||
cur.execute("""
|
cur.execute("""
|
||||||
INSERT INTO config_gpt (url, promt)
|
INSERT INTO sourse (url, promt)
|
||||||
VALUES (%s, %s)
|
VALUES (%s, %s)
|
||||||
ON CONFLICT (url) DO UPDATE SET
|
ON CONFLICT (url) DO UPDATE SET
|
||||||
promt = EXCLUDED.promt
|
promt = EXCLUDED.promt
|
||||||
|
|||||||
Reference in New Issue
Block a user