new project

fastapi_admin/widgets/inputs.py

@@ -0,0 +1,214 @@
+import abc
+import json
+from enum import Enum as EnumCLS
+from typing import Any, List, Optional, Tuple, Type
+
+from starlette.datastructures import UploadFile
+from tortoise import Model
+
+from fastapi_admin.providers.file_upload import FileUploadProvider
+from fastapi_admin.widgets import Widget
+
+
+class Input(Widget):
+    template = "widgets/inputs/input.html"
+
+    def __init__(self, default: Any = None, null: bool = False, **context):
+        super().__init__(null=null, **context)
+        self.default = default
+
+    async def parse_value(self, value: Any):
+        """
+        Parse value from frontend
+        :param value:
+        :return:
+        """
+        return value
+
+    async def render(self, value: Any):
+        if value is None:
+            value = self.default
+        return await super(Input, self).render(value)
+
+
+class DisplayOnly(Input):
+    """
+    Only display without input in edit or create
+    """
+
+
+class Text(Input):
+    input_type: Optional[str] = "text"
+
+    def __init__(
+        self, default: Any = None, null: bool = False, placeholder: str = "", disabled: bool = False
+    ):
+        super().__init__(
+            null=null,
+            default=default,
+            input_type=self.input_type,
+            placeholder=placeholder,
+            disabled=disabled,
+        )
+
+
+class Select(Input):
+    template = "widgets/inputs/select.html"
+
+    def __init__(self, default: Any = None, null: bool = False, disabled: bool = False):
+        super().__init__(null=null, default=default, disabled=disabled)
+
+    @abc.abstractmethod
+    async def get_options(self):
+        """
+        return list of tuple with display and value
+
+        [("on",1),("off",2)]
+
+        :return: list of tuple with display and value
+        """
+
+    async def render(self, value: Any):
+        options = await self.get_options()
+        self.context.update(options=options)
+        return await super(Select, self).render(value)
+
+
+class ForeignKey(Select):
+    def __init__(
+        self,
+        model: Type[Model],
+        default: Any = None,
+        null: bool = False,
+        disabled: bool = False,
+    ):
+        super().__init__(default=default, null=null, disabled=disabled)
+        self.model = model
+
+    async def get_options(self):
+        ret = await self.get_queryset()
+        options = [(str(x), x.pk) for x in ret]
+        if self.context.get("null"):
+            options = [("", "")] + options
+        return options
+
+    async def get_queryset(self):
+        return await self.model.all()
+
+
+class Enum(Select):
+    def __init__(
+        self,
+        enum: Type[EnumCLS],
+        default: Any = None,
+        enum_type: Type = int,
+        null: bool = False,
+        disabled: bool = False,
+    ):
+        super().__init__(default=default, null=null, disabled=disabled)
+        self.enum = enum
+        self.enum_type = enum_type
+
+    async def parse_value(self, value: Any):
+        return self.enum(self.enum_type(value))
+
+    async def get_options(self):
+        options = [(v.name, v.value) for v in self.enum]
+        if self.context.get("null"):
+            options = [("", "")] + options
+        return options
+
+
+class Email(Text):
+    input_type = "email"
+
+
+class Json(Input):
+    template = "widgets/inputs/json.html"
+
+    def __init__(self, null: bool = False, options: Optional[dict] = None):
+        """
+        options config to jsoneditor, see https://github.com/josdejong/jsoneditor
+        :param options:
+        """
+        super().__init__(null=null)
+        if not options:
+            options = {}
+        self.context.update(options=options)
+
+    async def render(self, value: Any):
+        if value:
+            value = json.dumps(value)
+        return await super().render(value)
+
+
+class TextArea(Text):
+    template = "widgets/inputs/textarea.html"
+    input_type = "textarea"
+
+
+class DateTime(Text):
+    input_type = "datetime"
+
+
+class Date(Text):
+    input_type = "date"
+
+
+class File(Input):
+    input_type = "file"
+
+    def __init__(
+        self,
+        upload_provider: FileUploadProvider,
+        default: Any = None,
+        null: bool = False,
+        disabled: bool = False,
+    ):
+        super().__init__(
+            null=null,
+            default=default,
+            input_type=self.input_type,
+            disabled=disabled,
+        )
+        self.upload_provider = upload_provider
+
+    async def parse_value(self, value: Optional[UploadFile]):
+        if value:
+            return await self.upload_provider.upload(value)
+
+
+class Image(File):
+    input_type = "file"
+
+
+class Radio(Select):
+    template = "widgets/inputs/radio.html"
+
+    def __init__(self, options: List[Tuple[str, Any]], default: Any = None, disabled: bool = False):
+        super().__init__(default=default, disabled=disabled)
+        self.options = options
+
+    async def get_options(self):
+        return self.options
+
+
+class RadioEnum(Enum):
+    template = "widgets/inputs/radio.html"
+
+
+class Switch(Input):
+    template = "widgets/inputs/switch.html"
+
+    async def parse_value(self, value: str):
+        if value == "on":
+            return True
+        return False
+
+
+class Password(Text):
+    input_type = "password"
+
+
+class Number(Text):
+    input_type = "number"