65 lines
1.5 KiB
Python
65 lines
1.5 KiB
Python
|
from apiflask import FileSchema, Schema, fields
|
||
|
|
from flask import send_file
|
||
|
|
|
||
|
|
from backend import api_bp, dataset_path
|
||
|
|
from backend.service import Service
|
||
|
|
|
||
|
|
|
||
|
|
class FileUpload(Schema):
|
||
|
|
file = fields.File(required=True)
|
||
|
|
|
||
|
|
|
||
|
|
class ColumnInfoDto(Schema):
|
||
|
|
datatype = fields.String()
|
||
|
|
items = fields.List(fields.String())
|
||
|
|
|
||
|
|
|
||
|
|
class TableColumnDto(Schema):
|
||
|
|
name = fields.String()
|
||
|
|
datatype = fields.String()
|
||
|
|
items = fields.List(fields.String())
|
||
|
|
|
||
|
|
|
||
|
|
service = Service(dataset_path)
|
||
|
|
|
||
|
|
|
||
|
|
@api_bp.post("/dataset")
|
||
|
|
@api_bp.input(FileUpload, location="files")
|
||
|
|
def upload_dataset(files_data):
|
||
|
|
uploaded_file = files_data["file"]
|
||
|
|
return service.upload_dataset(uploaded_file)
|
||
|
|
|
||
|
|
|
||
|
|
@api_bp.get("/dataset")
|
||
|
|
@api_bp.output(Schema, example="[string, string, ..., string]")
|
||
|
|
def get_all_datasets():
|
||
|
|
return service.get_all_datasets()
|
||
|
|
|
||
|
|
|
||
|
|
@api_bp.get("/dataset/<string:name>")
|
||
|
|
@api_bp.output(TableColumnDto(many=True))
|
||
|
|
def get_dataset_info(name: str):
|
||
|
|
return service.get_dataset_info(name)
|
||
|
|
|
||
|
|
|
||
|
|
@api_bp.get("/dataset/<string:name>/<string:column>")
|
||
|
|
@api_bp.output(ColumnInfoDto)
|
||
|
|
def get_column_info(name: str, column: str):
|
||
|
|
return service.get_column_info(name, column)
|
||
|
|
|
||
|
|
|
||
|
|
@api_bp.get("/dataset/draw/hist/<string:name>")
|
||
|
|
@api_bp.output(
|
||
|
|
FileSchema(type="string", format="binary"), content_type="image/png", example=""
|
||
|
|
)
|
||
|
|
def get_dataset_hist(name: str):
|
||
|
|
data = service.get_hist(name)
|
||
|
|
data.seek(0)
|
||
|
|
return send_file(
|
||
|
|
data,
|
||
|
|
download_name=f"{name}.hist.png",
|
||
|
|
mimetype="image/png",
|
||
|
|
as_attachment=True,
|
||
|
|
conditional=True,
|
||
|
|
)
|