test / terrible
JairoDanielMT's picture
Update terrible
9044f26 verified
{'resourceId': 44888333, 'workSchedules': [{'recordType': 'Descanso Semanal', 'startDate': datetime.datetime(2024, 1, 28, 0, 0), 'endDate': datetime.datetime(2024, 1, 28, 0, 0)}, {'recordType': 'Ok', 'startDate': datetime.datetime(2024, 1, 29, 0, 0), 'endDate': datetime.datetime(2024, 1, 29, 0, 0)}, {'recordType': 'Ok', 'startDate': datetime.datetime(2024, 1, 30, 0, 0), 'endDate': datetime.datetime(2024, 1, 30, 0, 0)}, {'recordType': 'Ok', 'startDate': datetime.datetime(2024, 1, 31, 0, 0), 'endDate': datetime.datetime(2024, 1, 31, 0, 0)}, {'recordType': 'Ok', 'startDate': datetime.datetime(2024, 2, 1, 0, 0), 'endDate': datetime.datetime(2024, 2, 1, 0, 0)}, {'recordType': 'Ok', 'startDate': datetime.datetime(2024, 2, 2, 0, 0), 'endDate': datetime.datetime(2024, 2, 2, 0, 0)}, {'recordType': 'Ok', 'startDate': datetime.datetime(2024, 2, 3, 0, 0), 'endDate': datetime.datetime(2024, 2, 3, 0, 0)}]},
from fastapi import FastAPI, File, UploadFile
from fastapi.responses import FileResponse
import pandas as pd
import io
from typing import List
from datetime import datetime
app = FastAPI()
def adaptar_a_formato_oracle(item):
nuevo_item = {"resourceId": item["resourceId"], "workSchedules": []}
for fecha, status in item.items():
if fecha != "resourceId":
record_type = "Descanso Semanal" if status == "Descanso Semanal" else "Ok"
fecha_str = fecha.strftime("%Y-%m-%d") # Convertir a cadena de texto
nuevo_item["workSchedules"].append({
"recordType": record_type,
"startDate": fecha_str,
"endDate": fecha_str # Puedes ajustar esto según tu lógica
})
return nuevo_item
@app.post("/excel_json/")
async def excel_json(file: UploadFile = File(...), sheet_name: str = "Nombre_hoja"):
df = pd.read_excel(io.BytesIO(await file.read()), engine='openpyxl', sheet_name=sheet_name)
# Transformar el formato al deseado para Oracle
json_original = df.to_dict(orient="records")
json_adaptado = [adaptar_a_formato_oracle(item) for item in json_original]
# Guardar el JSON adaptado en un archivo
json_file = "output.json"
with open(json_file, "w") as output_file:
output_file.write(str(json_adaptado).replace("datetime.datetime", "").replace("'", '"'))
# Devolver el archivo JSON adaptado como respuesta
return FileResponse(json_file, media_type='application/json', filename=json_file)