mobixconsulting commited on
Commit
5659f36
1 Parent(s): 73869d0

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +5 -6
main.py CHANGED
@@ -2,14 +2,13 @@ from fastapi import FastAPI, Query
2
  from pydantic import BaseModel
3
  import requests
4
  import json
5
- from tabulate import tabulate
6
 
7
  app = FastAPI()
8
 
9
  class EatcHeadersResponse(BaseModel):
10
  report: str
11
 
12
- @app.get("/eatc-headers", response_model=EatcHeadersResponse)
13
  async def get_eatc_headers(eatc_code: str = Query(..., description="EATC code to retrieve headers for")):
14
  url = f"https://donantes.eatcloud.info/api/abaco/eatc_dona_headers?eatc-code={eatc_code}"
15
  response = requests.get(url)
@@ -19,10 +18,10 @@ async def get_eatc_headers(eatc_code: str = Query(..., description="EATC code to
19
  eatc_code_value = response_json["res"][0]["eatc-code"]
20
  eatc_match_asignation_rule_value = response_json["res"][0].get("eatc_match_asignation_rule", "")
21
 
22
- report = f"eatc_code: {eatc_code_value}\n\neatc_match_asignation_rule: {eatc_match_asignation_rule_value}"
23
 
24
- return {"report": report}
25
  else:
26
- return {"error": "Invalid response from API"}
27
  else:
28
- return {"error": "API returned error status code"}
 
2
  from pydantic import BaseModel
3
  import requests
4
  import json
 
5
 
6
  app = FastAPI()
7
 
8
  class EatcHeadersResponse(BaseModel):
9
  report: str
10
 
11
+ @app.get("/eatc-headers", response_class=HTMLResponse)
12
  async def get_eatc_headers(eatc_code: str = Query(..., description="EATC code to retrieve headers for")):
13
  url = f"https://donantes.eatcloud.info/api/abaco/eatc_dona_headers?eatc-code={eatc_code}"
14
  response = requests.get(url)
 
18
  eatc_code_value = response_json["res"][0]["eatc-code"]
19
  eatc_match_asignation_rule_value = response_json["res"][0].get("eatc_match_asignation_rule", "")
20
 
21
+ report = f"<p>eatc_code: {eatc_code_value}</p><p>eatc_match_asignation_rule: {eatc_match_asignation_rule_value}</p>"
22
 
23
+ return HTMLResponse(content=report, status_code=200)
24
  else:
25
+ return HTMLResponse(content="<p>Error: Invalid response from API</p>", status_code=400)
26
  else:
27
+ return HTMLResponse(content="<p>Error: API returned error status code</p>", status_code=400)