Pingul commited on
Commit
8cc6714
·
verified ·
1 Parent(s): 874a57a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -9
app.py CHANGED
@@ -145,7 +145,7 @@ def df_to_dict_list(df: pd.DataFrame) -> list:
145
 
146
  @app.get("/kepler", response_model=DatasetListResponse)
147
  async def get_kepler_exoplanets(
148
- limit: int = Query(100, ge=1, le=1000, description="Número máximo de resultados"),
149
  offset: int = Query(0, ge=0, description="Offset para paginación"),
150
  koi_disposition: Optional[str] = Query(None, description="Filtrar por disposición (CONFIRMED, FALSE POSITIVE, CANDIDATE)")
151
  ):
@@ -172,8 +172,11 @@ async def get_kepler_exoplanets(
172
  if koi_disposition:
173
  df = df[df['koi_disposition'] == koi_disposition]
174
 
175
- # Aplicar paginación
176
- df_page = df.iloc[offset:offset + limit]
 
 
 
177
 
178
  # Seleccionar columnas relevantes (solo con ≤50% nulos)
179
  columns_to_include = [
@@ -229,7 +232,7 @@ async def get_cumulative_exoplanet_by_name(kepoi_name: str):
229
 
230
  @app.get("/k2", response_model=DatasetListResponse)
231
  async def get_k2_exoplanets(
232
- limit: int = Query(100, ge=1, le=1000, description="Número máximo de resultados"),
233
  offset: int = Query(0, ge=0, description="Offset para paginación"),
234
  disposition: Optional[str] = Query(None, description="Filtrar por disposición")
235
  ):
@@ -245,8 +248,11 @@ async def get_k2_exoplanets(
245
  if disposition:
246
  df = df[df['disposition'] == disposition]
247
 
248
- # Aplicar paginación
249
- df_page = df.iloc[offset:offset + limit]
 
 
 
250
 
251
  # Seleccionar columnas relevantes (solo con ≤50% nulos)
252
  columns_to_include = [
@@ -302,7 +308,7 @@ async def get_k2_exoplanet_by_name(pl_name: str):
302
 
303
  @app.get("/tess", response_model=DatasetListResponse)
304
  async def get_tess_exoplanets(
305
- limit: int = Query(100, ge=1, le=1000, description="Número máximo de resultados"),
306
  offset: int = Query(0, ge=0, description="Offset para paginación"),
307
  tfopwg_disp: Optional[str] = Query(None, description="Filtrar por disposición del grupo de trabajo")
308
  ):
@@ -318,8 +324,11 @@ async def get_tess_exoplanets(
318
  if tfopwg_disp:
319
  df = df[df['tfopwg_disp'] == tfopwg_disp]
320
 
321
- # Aplicar paginación
322
- df_page = df.iloc[offset:offset + limit]
 
 
 
323
 
324
  # Seleccionar columnas relevantes (solo con ≤50% nulos)
325
  columns_to_include = [
 
145
 
146
  @app.get("/kepler", response_model=DatasetListResponse)
147
  async def get_kepler_exoplanets(
148
+ limit: Optional[int] = Query(None, ge=1, le=1000, description="Número máximo de resultados (si no se especifica, devuelve todos)"),
149
  offset: int = Query(0, ge=0, description="Offset para paginación"),
150
  koi_disposition: Optional[str] = Query(None, description="Filtrar por disposición (CONFIRMED, FALSE POSITIVE, CANDIDATE)")
151
  ):
 
172
  if koi_disposition:
173
  df = df[df['koi_disposition'] == koi_disposition]
174
 
175
+ # Aplicar paginación solo si limit está definido
176
+ if limit is not None:
177
+ df_page = df.iloc[offset:offset + limit]
178
+ else:
179
+ df_page = df.iloc[offset:]
180
 
181
  # Seleccionar columnas relevantes (solo con ≤50% nulos)
182
  columns_to_include = [
 
232
 
233
  @app.get("/k2", response_model=DatasetListResponse)
234
  async def get_k2_exoplanets(
235
+ limit: Optional[int] = Query(None, ge=1, le=1000, description="Número máximo de resultados (si no se especifica, devuelve todos)"),
236
  offset: int = Query(0, ge=0, description="Offset para paginación"),
237
  disposition: Optional[str] = Query(None, description="Filtrar por disposición")
238
  ):
 
248
  if disposition:
249
  df = df[df['disposition'] == disposition]
250
 
251
+ # Aplicar paginación solo si limit está definido
252
+ if limit is not None:
253
+ df_page = df.iloc[offset:offset + limit]
254
+ else:
255
+ df_page = df.iloc[offset:]
256
 
257
  # Seleccionar columnas relevantes (solo con ≤50% nulos)
258
  columns_to_include = [
 
308
 
309
  @app.get("/tess", response_model=DatasetListResponse)
310
  async def get_tess_exoplanets(
311
+ limit: Optional[int] = Query(None, ge=1, le=1000, description="Número máximo de resultados (si no se especifica, devuelve todos)"),
312
  offset: int = Query(0, ge=0, description="Offset para paginación"),
313
  tfopwg_disp: Optional[str] = Query(None, description="Filtrar por disposición del grupo de trabajo")
314
  ):
 
324
  if tfopwg_disp:
325
  df = df[df['tfopwg_disp'] == tfopwg_disp]
326
 
327
+ # Aplicar paginación solo si limit está definido
328
+ if limit is not None:
329
+ df_page = df.iloc[offset:offset + limit]
330
+ else:
331
+ df_page = df.iloc[offset:]
332
 
333
  # Seleccionar columnas relevantes (solo con ≤50% nulos)
334
  columns_to_include = [