Spaces:
Sleeping
Sleeping
File size: 5,752 Bytes
69601d4 1b179fe 69601d4 1b179fe 69601d4 1b179fe 69601d4 1b179fe 69601d4 1b179fe 69601d4 1b179fe 69601d4 1b179fe 69601d4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 |
from ..models.requestModels import CreatePage, ExportToDashboard
from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
from fastapi.exceptions import HTTPException
from fastapi.responses import JSONResponse
from ..utils.functions import verifyToken
from fastapi import APIRouter, Depends
from supabase import create_client
from urllib.request import urlopen
from typing import Annotated
import uuid
import json
import time
import os
import io
router = APIRouter()
security = HTTPBearer()
client = create_client(
supabase_url = os.environ["SUPABASE_URL"],
supabase_key = os.environ["SUPABASE_KEY"]
)
@router.post("/createPage")
async def createPage(details: CreatePage, credentials: Annotated[HTTPAuthorizationCredentials, Depends(security)]):
try:
if verifyToken(token = credentials.credentials):
pageId = str(uuid.uuid4())
if "dashboardConfig.json" in [x.get("name") for x in client.storage.from_("AnalyticsHub").list(path = details.projectId)]:
fileUrl = os.environ["FILE_URL"].format(projectId = details.projectId, fileName = "dashboardConfig.json").replace(".parquet", "") + f"?cb={int(time.time())}"
dashboardConfig = json.loads(urlopen(fileUrl).read())
dashboardConfig[pageId] = {"name": details.pageName, "widgets": []}
else:
dashboardConfig = {pageId: {"name": details.pageName, "widgets": []}}
with io.BytesIO() as buffer:
buffer.write(json.dumps(dashboardConfig, indent=4).encode("utf-8"))
buffer.seek(0)
_ = client.storage.from_("AnalyticsHub").upload(path = f"{details.projectId}/dashboardConfig.json", file = buffer.getvalue(), file_options = {"upsert": "true"})
return JSONResponse(status_code = 200, content = {"status": "SUCCESS", "pageId": pageId})
else:
return JSONResponse(status_code = 498, content = {"status": "ERROR", "errorDetail": "Invalid Token"})
except Exception as e:
raise HTTPException(status_code = 500, detail = f"Endpoint says: {e}")
@router.get("/getAllPages")
async def getAllPages(projectId: str, credentials: Annotated[HTTPAuthorizationCredentials, Depends(security)]):
try:
if verifyToken(token = credentials.credentials):
if "dashboardConfig.json" in [x.get("name") for x in client.storage.from_("AnalyticsHub").list(path = projectId)]:
fileUrl = os.environ["FILE_URL"].format(projectId = projectId, fileName = "dashboardConfig.json").replace(".parquet", "") + f"?cb={int(time.time())}"
dashboardConfig = json.loads(urlopen(fileUrl).read())
pages = [{"pageName": dashboardConfig[x]["name"], "pageId": x} for x in dashboardConfig.keys()]
else:
pages = list()
return JSONResponse(status_code = 200, content = {"status": "SUCCESS", "pages": pages})
else:
return JSONResponse(status_code = 498, content = {"status": "ERROR", "errorDetail": "Invalid Token"})
except Exception as e:
raise HTTPException(status_code = 500, detail = f"Endpoint says: {e}")
@router.post("/exportToDashboard")
async def exportToDashboard(details: ExportToDashboard, credentials: Annotated[HTTPAuthorizationCredentials, Depends(security)]):
try:
if verifyToken(token = credentials.credentials):
fileUrl = os.environ["FILE_URL"].format(projectId = details.projectId, fileName = "dashboardConfig.json").replace(".parquet", "") + f"?cb={int(time.time())}"
dashboardConfig = json.loads(urlopen(fileUrl).read())
pageDict = dashboardConfig.get(details.page)
widgetId = str(uuid.uuid4())
newWidget = {
"id": widgetId,
"chartType": details.chartType,
"title": details.title,
"xLabels": details.xLabels,
"yLabels": details.yLabels,
"data": details.data,
"layout": details.layout
}
pageDict["widgets"].append(newWidget)
dashboardConfig[details.page] = pageDict
with io.BytesIO() as buffer:
buffer.write(json.dumps(dashboardConfig, indent=4).encode("utf-8"))
buffer.seek(0)
_ = client.storage.from_("AnalyticsHub").upload(path = f"{details.projectId}/dashboardConfig.json", file = buffer.getvalue(), file_options = {"upsert": "true"})
return JSONResponse(status_code = 200, content = {"status": "SUCCESS", "widgetId": widgetId})
else:
return JSONResponse(status_code = 498, content = {"status": "ERROR", "errorDetail": "Invalid Token"})
except Exception as e:
raise HTTPException(status_code = 500, detail = f"Endpoint says: {e}")
@router.get("/getData")
async def getData(projectId: str, page: str, credentials: Annotated[HTTPAuthorizationCredentials, Depends(security)]):
try:
if verifyToken(token = credentials.credentials):
fileUrl = os.environ["FILE_URL"].format(projectId = projectId, fileName = "dashboardConfig.json").replace(".parquet", "") + f"?cb={int(time.time())}"
dashboardConfig = json.loads(urlopen(fileUrl).read())
pageInfo = dashboardConfig.get(page)
pageInfo["id"] = page
return JSONResponse(status_code = 200, content = {"status": "SUCCESS", "pageData": pageInfo})
else:
return JSONResponse(status_code = 498, content = {"status": "ERROR", "errorDetail": "Invalid Token"})
except Exception as e:
raise HTTPException(status_code = 500, detail = f"Endpoint says: {e}")
|