fork para inversionitas con bot simple
This commit is contained in:
319
app/routes.py
319
app/routes.py
@ -1,4 +1,4 @@
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
from fastapi import APIRouter, Depends, HTTPException, Query
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy.sql import func
|
||||
from database import get_db
|
||||
@ -25,157 +25,8 @@ class NewsItemCreate(BaseModel):
|
||||
fuente: str | None = None
|
||||
fecha: datetime | None = None
|
||||
link: str
|
||||
critico:bool
|
||||
favorable:bool
|
||||
keyword: str | None = None
|
||||
|
||||
@router.post("/news/")
|
||||
def create_news_item(item: NewsItemCreate, db: Session = Depends(get_db)):
|
||||
# Verificar si el título ya existe
|
||||
existing_item = db.query(NewsItem).filter(NewsItem.titulo == item.titulo).first()
|
||||
if existing_item:
|
||||
logging.info("Título ya en la base de datos")
|
||||
|
||||
# Crear nuevo objeto
|
||||
new_item = NewsItem(
|
||||
titulo=item.titulo,
|
||||
contenido=item.contenido,
|
||||
autor=item.autor,
|
||||
fuente=item.fuente,
|
||||
fecha=item.fecha or datetime.utcnow(),
|
||||
link=item.link,
|
||||
critico=item.critico,
|
||||
favorable=item.favorable,
|
||||
keyword=item.keyword
|
||||
)
|
||||
|
||||
db.add(new_item)
|
||||
db.commit()
|
||||
db.refresh(new_item)
|
||||
|
||||
return {"message": "Noticia creada con éxito", "id": new_item.id, "titulo": new_item.titulo}
|
||||
|
||||
@router.get("/news/count/by-source/date-range")
|
||||
def count_news_by_source_in_range(
|
||||
fecha_inicio: datetime, fecha_fin: datetime, db: Session = Depends(get_db)
|
||||
):
|
||||
results = (
|
||||
db.query(NewsItem.fuente, func.count(NewsItem.id))
|
||||
.filter(NewsItem.fecha >= fecha_inicio, NewsItem.fecha <= fecha_fin)
|
||||
.group_by(NewsItem.fuente)
|
||||
.all()
|
||||
)
|
||||
return {"count_by_source_in_range": [{"fuente": fuente, "count": count} for fuente, count in results]}
|
||||
|
||||
@router.get("/news/count/by-author/date-range")
|
||||
def count_news_by_author_in_range(
|
||||
fecha_inicio: datetime, fecha_fin: datetime, db: Session = Depends(get_db)
|
||||
):
|
||||
results = (
|
||||
db.query(NewsItem.autor, func.count(NewsItem.id))
|
||||
.filter(NewsItem.fecha >= fecha_inicio, NewsItem.fecha <= fecha_fin)
|
||||
.group_by(NewsItem.autor)
|
||||
.all()
|
||||
)
|
||||
return {"count_by_author_in_range": [{"autor": autor, "count": count} for autor, count in results]}
|
||||
|
||||
@router.get("/news/count/favorable/by-author/date-range")
|
||||
def count_favorable_news_by_author_in_range(
|
||||
fecha_inicio: datetime, fecha_fin: datetime, db: Session = Depends(get_db)
|
||||
):
|
||||
results = (
|
||||
db.query(NewsItem.autor, func.count(NewsItem.id))
|
||||
.filter(
|
||||
NewsItem.favorable == True,
|
||||
NewsItem.fecha >= fecha_inicio,
|
||||
NewsItem.fecha <= fecha_fin,
|
||||
)
|
||||
.group_by(NewsItem.autor)
|
||||
.all()
|
||||
)
|
||||
return {"favorable_count_by_author_in_range": [{"autor": autor, "count": count} for autor, count in results]}
|
||||
|
||||
@router.get("/news/count/unfavorable/by-author/date-range")
|
||||
def count_unfavorable_news_by_author_in_range(
|
||||
fecha_inicio: datetime, fecha_fin: datetime, db: Session = Depends(get_db)
|
||||
):
|
||||
results = (
|
||||
db.query(NewsItem.autor, func.count(NewsItem.id))
|
||||
.filter(
|
||||
NewsItem.critico == True,
|
||||
NewsItem.fecha >= fecha_inicio,
|
||||
NewsItem.fecha <= fecha_fin,
|
||||
)
|
||||
.group_by(NewsItem.autor)
|
||||
.all()
|
||||
)
|
||||
return {"unfavorable_count_by_author_in_range": [{"autor": autor, "count": count} for autor, count in results]}
|
||||
|
||||
@router.get("/news/count/favorable/by-source/date-range")
|
||||
def count_favorable_news_by_source_in_range(
|
||||
fecha_inicio: datetime, fecha_fin: datetime, db: Session = Depends(get_db)
|
||||
):
|
||||
results = (
|
||||
db.query(NewsItem.fuente, func.count(NewsItem.id))
|
||||
.filter(
|
||||
NewsItem.favorable == True,
|
||||
NewsItem.fecha >= fecha_inicio,
|
||||
NewsItem.fecha <= fecha_fin,
|
||||
)
|
||||
.group_by(NewsItem.fuente)
|
||||
.all()
|
||||
)
|
||||
return {"favorable_count_by_source_in_range": [{"fuente": fuente, "count": count} for fuente, count in results]}
|
||||
|
||||
@router.get("/news/count/unfavorable/by-source/date-range")
|
||||
def count_unfavorable_news_by_source_in_range(
|
||||
fecha_inicio: datetime, fecha_fin: datetime, db: Session = Depends(get_db)
|
||||
):
|
||||
results = (
|
||||
db.query(NewsItem.fuente, func.count(NewsItem.id))
|
||||
.filter(
|
||||
NewsItem.critico == True,
|
||||
NewsItem.fecha >= fecha_inicio,
|
||||
NewsItem.fecha <= fecha_fin,
|
||||
)
|
||||
.group_by(NewsItem.fuente)
|
||||
.all()
|
||||
)
|
||||
return {"unfavorable_count_by_source_in_range": [{"fuente": fuente, "count": count} for fuente, count in results]}
|
||||
|
||||
@router.get("/news/neutral/date-range")
|
||||
def get_neutral_news_in_range(
|
||||
fecha_inicio: datetime, fecha_fin: datetime, db: Session = Depends(get_db)
|
||||
):
|
||||
results = (
|
||||
db.query(NewsItem)
|
||||
.filter(
|
||||
NewsItem.favorable == False,
|
||||
NewsItem.critico == False,
|
||||
NewsItem.fecha >= fecha_inicio,
|
||||
NewsItem.fecha <= fecha_fin,
|
||||
)
|
||||
.all()
|
||||
)
|
||||
return results
|
||||
|
||||
|
||||
@router.get("/news/mixed/date-range")
|
||||
def get_mixed_news_in_range(
|
||||
fecha_inicio: datetime, fecha_fin: datetime, db: Session = Depends(get_db)
|
||||
):
|
||||
results = (
|
||||
db.query(NewsItem)
|
||||
.filter(
|
||||
NewsItem.favorable == True,
|
||||
NewsItem.critico == True,
|
||||
NewsItem.fecha >= fecha_inicio,
|
||||
NewsItem.fecha <= fecha_fin,
|
||||
)
|
||||
.all()
|
||||
)
|
||||
return results
|
||||
|
||||
@router.get("/news/count/by-source")
|
||||
def count_news_by_source(db: Session = Depends(get_db)):
|
||||
results = (
|
||||
@ -202,62 +53,142 @@ def count_news_by_author(db: Session = Depends(get_db)):
|
||||
|
||||
return {"count_by_author": response}
|
||||
|
||||
@router.get("/news/count/favorable/by-author")
|
||||
def count_favorable_news_by_author(db: Session = Depends(get_db)):
|
||||
results = (
|
||||
db.query(NewsItem.autor, func.count(NewsItem.id))
|
||||
.filter(NewsItem.favorable == True)
|
||||
.group_by(NewsItem.autor)
|
||||
.all()
|
||||
)
|
||||
return {"favorable_count_by_author": results}
|
||||
@router.get("/news/{news_id}")
|
||||
def get_news_by_id(news_id: int, db: Session = Depends(get_db)):
|
||||
news_item = db.query(NewsItem).filter(NewsItem.id == news_id).first()
|
||||
|
||||
if not news_item:
|
||||
raise HTTPException(status_code=404, detail="Noticia no encontrada")
|
||||
|
||||
return {
|
||||
"id": news_item.id,
|
||||
"titulo": news_item.titulo,
|
||||
"contenido": news_item.contenido,
|
||||
"autor": news_item.autor,
|
||||
"fuente": news_item.fuente,
|
||||
"fecha": news_item.fecha,
|
||||
"link": news_item.link,
|
||||
"critico": news_item.critico,
|
||||
"favorable": news_item.favorable,
|
||||
"keyword": news_item.keyword,
|
||||
}
|
||||
|
||||
@router.get("/news/count/unfavorable/by-author")
|
||||
def count_unfavorable_news_by_author(db: Session = Depends(get_db)):
|
||||
results = (
|
||||
db.query(NewsItem.autor, func.count(NewsItem.id))
|
||||
.filter(NewsItem.critico == True)
|
||||
.group_by(NewsItem.autor)
|
||||
.all()
|
||||
)
|
||||
return {"unfavorable_count_by_author": results}
|
||||
@router.get("/news/count/by-keyword")
|
||||
def count_news_by_keyword(
|
||||
fechaInicio: datetime | None = None,
|
||||
fechaFin: datetime | None = None,
|
||||
db: Session = Depends(get_db)):
|
||||
query = db.query(NewsItem.keyword, func.count(NewsItem.id)).filter(NewsItem.keyword.isnot(None))
|
||||
if fechaInicio and fechaFin:
|
||||
query = query.filter(NewsItem.fecha >= fechaInicio, NewsItem.fecha <= fechaFin)
|
||||
results = query.group_by(NewsItem.keyword).all()
|
||||
|
||||
@router.get("/news/count/favorable/by-source")
|
||||
def count_favorable_news_by_source(db: Session = Depends(get_db)):
|
||||
return {keyword: count for keyword, count in results}
|
||||
|
||||
@router.get("/news/keywords/")
|
||||
def get_all_keywords(db: Session = Depends(get_db)):
|
||||
results = db.query(NewsItem.keyword).distinct().all()
|
||||
|
||||
# Extraer solo las keywords y filtrar valores nulos
|
||||
keywords = [keyword[0] for keyword in results if keyword[0] is not None]
|
||||
|
||||
return {"keywords": keywords}
|
||||
|
||||
@router.get("/news/count/by-source/keyword/date-range")
|
||||
def count_news_by_source_for_keyword_in_range(
|
||||
keyword: str, fechaInicio: datetime, fechaFin: datetime, db: Session = Depends(get_db)
|
||||
):
|
||||
results = (
|
||||
db.query(NewsItem.fuente, func.count(NewsItem.id))
|
||||
.filter(NewsItem.favorable == True)
|
||||
.filter(
|
||||
NewsItem.keyword == keyword,
|
||||
NewsItem.fecha >= fechaInicio,
|
||||
NewsItem.fecha <= fechaFin
|
||||
)
|
||||
.group_by(NewsItem.fuente)
|
||||
.all()
|
||||
)
|
||||
return {"favorable_count_by_source": results}
|
||||
formatted_results = {fuente: count for fuente, count in results}
|
||||
return {"count_by_source_keyword": formatted_results}
|
||||
|
||||
@router.get("/news/count/unfavorable/by-source")
|
||||
def count_unfavorable_news_by_source(db: Session = Depends(get_db)):
|
||||
@router.get("/news/titles/by-keyword/date-range")
|
||||
def get_titles_by_keyword_in_range(
|
||||
keyword: str, fecha_inicio: datetime, fecha_fin: datetime, db: Session = Depends(get_db)
|
||||
):
|
||||
results = (
|
||||
db.query(NewsItem.fuente, func.count(NewsItem.id))
|
||||
.filter(NewsItem.critico == True)
|
||||
.group_by(NewsItem.fuente)
|
||||
db.query(NewsItem.titulo)
|
||||
.filter(
|
||||
NewsItem.keyword == keyword,
|
||||
NewsItem.fecha >= fecha_inicio,
|
||||
NewsItem.fecha <= fecha_fin
|
||||
)
|
||||
.all()
|
||||
)
|
||||
return {"unfavorable_count_by_source": results}
|
||||
|
||||
@router.get("/news/neutral")
|
||||
def get_neutral_news(db: Session = Depends(get_db)):
|
||||
results = (
|
||||
db.query(NewsItem)
|
||||
.filter(NewsItem.favorable == False, NewsItem.critico == False)
|
||||
.all()
|
||||
)
|
||||
return results
|
||||
# Convertir resultados en una sola cadena separada por comas
|
||||
titles = ", ".join([titulo[0] for titulo in results if titulo[0]])
|
||||
|
||||
return {"titles": titles}
|
||||
|
||||
@router.get("/news/mixed")
|
||||
def get_mixed_news(db: Session = Depends(get_db)):
|
||||
results = (
|
||||
db.query(NewsItem)
|
||||
.filter(NewsItem.favorable == True, NewsItem.critico == True)
|
||||
.all()
|
||||
)
|
||||
return results
|
||||
# @router.get("/news-summary")
|
||||
# def get_news_summary(
|
||||
# keyword: str,
|
||||
# fecha_inicio: datetime,
|
||||
# fecha_fin: datetime,
|
||||
# db: Session = Depends(get_db)
|
||||
# ):
|
||||
# results = get_titles_by_keyword_in_range(keyword,fecha_inicio, fecha_fin, db)
|
||||
# data = {
|
||||
# "model":"llama3",
|
||||
# "prompt": f"A continuación tienes los titulares de muchas noticias, resume lo que encuentres más destacado. Noticias: {results}",
|
||||
# }
|
||||
|
||||
# try:
|
||||
# response = requests.post(
|
||||
# "http://host.docker.internal:11434/api/generate", json=data
|
||||
# )
|
||||
# response.raise_for_status()
|
||||
|
||||
# except requests.RequestException as e:
|
||||
# logging.error(f"Request error: {e}")
|
||||
# return response
|
||||
|
||||
@router.get("/news/titles/")
|
||||
def get_news_titles(
|
||||
fechaInicio: datetime | None = None,
|
||||
fechaFin: datetime | None = None,
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
query = db.query(NewsItem.titulo)
|
||||
|
||||
if fechaInicio and fechaFin:
|
||||
query = query.filter(NewsItem.fecha >= fechaInicio, NewsItem.fecha <= fechaFin)
|
||||
|
||||
results = query.all()
|
||||
return {"titles": [titulo[0] for titulo in results]}
|
||||
|
||||
@router.get("/news/")
|
||||
def get_all_news(
|
||||
fechaInicio: datetime | None = None,
|
||||
fechaFin: datetime | None = None,
|
||||
page:int = Query(1, alias="page", ge=1),
|
||||
pageSize: int = Query(10, alias="pageSize", ge=1, le=100),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
query = db.query(NewsItem.id, NewsItem.titulo, NewsItem.link, NewsItem.favorable, NewsItem.critico) # Solo seleccionamos id y titulo
|
||||
|
||||
if fechaInicio and fechaFin:
|
||||
query = query.filter(NewsItem.fecha >= fechaInicio, NewsItem.fecha <= fechaFin)
|
||||
|
||||
totalCount = query.count()
|
||||
|
||||
results = query.offset((page - 1) * pageSize).limit(pageSize).all()
|
||||
return {
|
||||
"totalCount": totalCount,
|
||||
"page": page,
|
||||
"pageSize": pageSize,
|
||||
"totalPages": (totalCount + pageSize - 1) // pageSize,
|
||||
"news": [{"id": news_id, "titulo": titulo, "url":link, "critico":critico, "favorable": favorable} for
|
||||
news_id, titulo, link, critico, favorable in results] # Devolvemos solo id y titulo
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user