Mr-Help commited on
Commit
b8980a7
ยท
verified ยท
1 Parent(s): ebe4628

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +16 -7
main.py CHANGED
@@ -1,4 +1,4 @@
1
- from fastapi import FastAPI
2
  import requests
3
 
4
  app = FastAPI()
@@ -17,10 +17,19 @@ def get_token():
17
 
18
  try:
19
  resp = requests.get(URL, headers=headers, timeout=20)
20
- resp.raise_for_status()
21
- try:
22
- return resp.json()
23
- except ValueError:
24
- return {"raw_text": resp.text}
 
 
 
 
25
  except requests.RequestException as e:
26
- return {"error": str(e)}
 
 
 
 
 
 
1
+ from fastapi import FastAPI, Response
2
  import requests
3
 
4
  app = FastAPI()
 
17
 
18
  try:
19
  resp = requests.get(URL, headers=headers, timeout=20)
20
+
21
+ # ู‡ู†ุง ุจู†ุฑุฌุน ู†ูุณ ูƒูˆุฏ ุงู„ุญุงู„ุฉ ุงู„ู„ูŠ ุฑุฌู‘ุนู‡ ุงู„ุณูŠุฑูุฑ ุงู„ุฃุตู„ูŠ
22
+ content_type = resp.headers.get("Content-Type", "application/json")
23
+ return Response(
24
+ content=resp.text,
25
+ status_code=resp.status_code,
26
+ media_type=content_type
27
+ )
28
+
29
  except requests.RequestException as e:
30
+ # ู„ูˆ ุญุตู„ ุฎุทุฃ ููŠ ุงู„ุงุชุตุงู„ ู†ูุณู‡ ุฃูˆ Timeout
31
+ return Response(
32
+ content=str(e),
33
+ status_code=500,
34
+ media_type="text/plain"
35
+ )