from fastapi import FastAPI, File, UploadFile
import logging
from correpy.parsers.brokerage_notes.parser_factory import ParserFactory


logging.basicConfig()
logger = logging.getLogger('uvicorn.error')

app = FastAPI()

RICO_PASSW = "052"

@app.get("/")
def read_root():
    return {"Hello": "World"}


@app.post("/convert-brokerage-note")
async def convert_brokerage_note(file: UploadFile):

    content = file.file.read()

    if "11775539" in file.filename:
        logger.info("Parsing Rico note")
        brokerage_notes = ParserFactory(brokerage_note=content, password=RICO_PASSW).parse()


    return {"fileName": file.filename, "result": brokerage_notes}