Source code for eve_panel.widgets

import ast
import json
import base64

import panel as pn
import param
from eve.io.mongo.validation import Validator
from panel.widgets import LiteralInput


class LiteralSchemaInputBase(LiteralInput):
    """[summary]

    Args:
        LiteralInput ([type]): [description]
    """
    def validate_schema(self, value):
        return True

    def _process_property_change(self, msg):
        msg = super(LiteralSchemaInputBase, self)._process_property_change(msg)
        if msg['value'] == self.value:
            return msg
        new_state = ''
        if 'value' in msg:
            value = msg.pop('value')
            if not self.validate_schema(value):
                new_state = ' (invalid)'
                value = self.value
            msg['value'] = value
            msg['name'] = msg.get('title', self.name).replace(
                self._state, '').replace(new_state, '') + new_state
            self._state = new_state
            self.param.trigger('name')
        return msg


def LiteralSchemaInput(name, schema, type_=None):
    validator = Validator({"value": schema})

    def validate_schema(self, value):
        return validator.validate({"value": value})

    params = {
        "validate_schema": validate_schema,
        "type": type_,
    }
    return type(name + "InputWidget", (LiteralSchemaInputBase, ), params)


def PDFViewer(pdf, width=800, height=500):
    if isinstance(pdf, bytes):
        pdf = base64.b64encode(pdf).decode()
    return pn.pane.HTML(f'<iframe width={width} height={height} src="data:application/pdf;base64,{pdf}" type="application/pdf"></iframe>',
                   width=1000,sizing_mode="stretch_both")

def PNGViewer(png, width=800, height=500):
    if isinstance(png, bytes):
        png = base64.b64encode(png).decode()
    src = f"data:image/png;base64,{png}"
    return pn.pane.HTML(f"<img src='{src}' width={width} height={height}></img>")
                
class FileInputPreview(pn.widgets.FileInput):

    @param.depends('value')
    def preview(self):
        if not self.filename:
            return pn.Column()
        if self.filename.endswith(".png"):
            return PNGViewer(self.value)
        elif self.filename.endswith(".pdf"):
            return PDFViewer(self.value)

    def _repr_mimebundle_(self, include=None, exclude=None):
        return pn.Column(self, self.preview)

WIDGET_MAPPING = {
    "media": {"type": pn.widgets.FileInput, "align": "end"},
}


def get_widget(name, schema):
    if schema.get('type', 'string') == "dict" and "schema" in schema:
        return LiteralSchemaInput(name, schema, dict)
    elif schema.get('type', 'string') == "list" and "schema" in schema:
        return LiteralSchemaInput(name, schema, list)
    else:
        return WIDGET_MAPPING.get(schema.get('type', 'string'), None)

class Progress(param.Parameterized):
    value = param.Integer(0)
    total = param.Integer(100)
    active = param.Boolean(False)
    desc = param.String("Loading")
    unit = param.String("iterations")
    
    def __call__(self, **params):
        self.param.set_param(**params)
        return self
    
    @param.depends("value", "total", "active")
    def view(self):
        perc = int(100*self.value/self.total)
        text = f"{perc}% [{self.value}/{self.total} {self.unit}]"
        ind = pn.indicators.Progress(value=perc, active=self.active, sizing_mode="stretch_width")
        
        return pn.Row(self.desc, ind, text, sizing_mode="stretch_width")
    
    def update(self, inc=1):
        self.value = self.value + inc
    
    def reset(self):
        self.value = 0