96 lines
2.5 KiB
Python
96 lines
2.5 KiB
Python
from typing import Dict, List, Optional, Union
|
|
|
|
from .iobase import pipe_interaction
|
|
from .widgets import Button, Widget
|
|
|
|
|
|
class Form:
|
|
"""
|
|
A container for different widgets
|
|
|
|
Syntax:
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
components: List[Union[Widget, str]],
|
|
title: Optional[str] = None,
|
|
submit_button_name: Optional[str] = None,
|
|
cancel_button_name: Optional[str] = None,
|
|
):
|
|
"""
|
|
components: components in the form, can be widgets (except Button) or strings
|
|
title: title of the form
|
|
"""
|
|
assert any(isinstance(c, Button) for c in components) is False, (
|
|
"Button is not allowed in Form"
|
|
)
|
|
|
|
self._components = components
|
|
self._title = title
|
|
|
|
self._rendered = False
|
|
self._submit = submit_button_name
|
|
self._cancel = cancel_button_name
|
|
|
|
@property
|
|
def components(self) -> List[Union[Widget, str]]:
|
|
"""
|
|
Return the components
|
|
"""
|
|
|
|
return self._components
|
|
|
|
def _in_chatmark(self) -> str:
|
|
"""
|
|
Generate ChatMark syntax for all components
|
|
"""
|
|
lines = []
|
|
|
|
if self._title:
|
|
lines.append(self._title)
|
|
|
|
for c in self.components:
|
|
if isinstance(c, str):
|
|
lines.append(c)
|
|
elif isinstance(c, Widget):
|
|
lines.append(c._in_chatmark())
|
|
else:
|
|
raise ValueError(f"Invalid component {c}")
|
|
|
|
return "\n".join(lines)
|
|
|
|
def _parse_response(self, response: Dict):
|
|
"""
|
|
Parse response from user input
|
|
"""
|
|
for c in self.components:
|
|
if isinstance(c, Widget):
|
|
c._parse_response(response)
|
|
|
|
def render(self):
|
|
"""
|
|
Render to receive user input
|
|
"""
|
|
if self._rendered:
|
|
# already rendered once
|
|
# not sure if the constraint is necessary
|
|
# could be removed if re-rendering is needed
|
|
raise RuntimeError("Widget can only be rendered once")
|
|
|
|
self._rendered = True
|
|
|
|
chatmark_header = "```chatmark"
|
|
chatmark_header += f" submit={self._submit}" if self._submit else ""
|
|
chatmark_header += f" cancel={self._cancel}" if self._cancel else ""
|
|
|
|
lines = [
|
|
chatmark_header,
|
|
self._in_chatmark(),
|
|
"```",
|
|
]
|
|
|
|
chatmark = "\n".join(lines)
|
|
response = pipe_interaction(chatmark)
|
|
self._parse_response(response)
|