def test_combo_box(self):
def update(control: ctrl.Select, parent: ctrl.Select):
selected = parent.items[parent.selected]
control.items = [f"{selected} 1", f"{selected} 2"]
app = ds.app()
cb = app.select(["Hello", "World"])
self.assertTrue(isinstance(cb._derive_model(), ctrl.DefaultListModel))
c1 = app.select(handler=update, depends=cb)
controller = Controller(app.controls, [])
views = controller.list()
v = ty.cast(ctrl.SelectView, self.get_by_id(cb.get_id(), views))
v1 = ty.cast(ctrl.SelectView, self.get_by_id(c1.get_id(), views))
self.assertEqual(0, v.selected)
self.assertEqual(["Hello 1", "Hello 2"], v1.titles)
v.selected = 1
print(views)
views = controller.list(views)
print(views)
v = ty.cast(ctrl.SelectView, self.get_by_id(cb.get_id(), views))
v1 = ty.cast(ctrl.SelectView, self.get_by_id(c1.get_id(), views))
self.assertEqual(1, v.selected)
self.assertEqual(["World 1", "World 2"], v1.titles)
def test_simple_check_box(self):
app = ds.app()
c1 = app.checkbox()
controller = Controller(app.controls, [])
views = controller.list()
v1 = ty.cast(ctrl.CheckboxView, self.get_by_id(c1.get_id(), views))
self.assertEqual(False, v1.selected)
def test_handler_called_only_once(self):
count = 0
def update_c2(control: ctrl.Control, text_area: ctrl.Input):
nonlocal count
if count == 0:
count += 1
else:
print(count)
raise RuntimeError()
control.text = str(int(text_area.text) * 2)
def update_c3_c4(control: ctrl.Control, text_area: ctrl.Input):
control.text = str(int(text_area.text) * 2)
app = ds.app()
c1 = app.input(text="10")
c2 = app.input(handler=update_c2, depends=c1)
c3 = app.input(handler=update_c3_c4, depends=c2)
c4 = app.input(handler=update_c3_c4, depends=c2)
controller = Controller(app.controls, [])
views = controller.list()
self.assertEqual(1, count)
v2 = ty.cast(ctrl.InputView, self.get_by_id(c2.get_id(), views))
v3 = ty.cast(ctrl.InputView, self.get_by_id(c3.get_id(), views))
v4 = ty.cast(ctrl.InputView, self.get_by_id(c4.get_id(), views))
self.assertEqual("20", v2.text)
self.assertEqual("40", v3.text)
self.assertEqual("40", v4.text)
def test_immutability_of_controller(self):
app = ds.app()
app.input(text="Some initial data")
controller = Controller(app.controls, [])
views = controller.list()
self.assertEqual(views[0].text, "Some initial data")
views[0].text = "Some other data"
updated_views = controller.list(views)
self.assertEqual(updated_views[0].text, "Some other data")
views = controller.list()
self.assertEqual(views[0].text, "Some initial data")
def test_combo_box_model(self):
class City:
def __init__(self, id, title):
self.id = id
self.title = title
def __repr__(self):
return self.title
class Country:
def __init__(self, code, title):
self.code = code
self.title = title
def __repr__(self):
return self.title
data = {"US": [City(0, "New York"), City(1, "San Francisco"), City(2, "Boston")],
"DE": [City(10, "Munich"), City(11, "Berlin"), City(12, "Hamburg")]}
def list_cities_from_db_by_code(country: Country) -> ty.List[City]:
return data[country.code]
def list_countries_from_db(self):
self.items = [Country("US", "United States"), Country("DE", "German")]
def update_cities(control: ctrl.Select, parent: ctrl.Select):
country = ty.cast(Country, parent.get_model().element(parent.selected))
control.items = list_cities_from_db_by_code(country)
app = ds.app()
countries = app.select(handler=list_countries_from_db)
cities = app.select(handler=update_cities, depends=countries)
controller = Controller(app.controls, [])
views = controller.list()
v1 = ty.cast(ctrl.SelectView, self.get_by_id(countries.get_id(), views))
v2 = ty.cast(ctrl.SelectView, self.get_by_id(cities.get_id(), views))
self.assertEqual(["United States", "German"], v1.titles)
self.assertEqual(0, v1.selected)
self.assertEqual(["New York", "San Francisco", "Boston"], v2.titles)
v1.selected = 1
views = controller.list(views)
v1 = ty.cast(ctrl.SelectView, self.get_by_id(countries.get_id(), views))
v2 = ty.cast(ctrl.SelectView, self.get_by_id(cities.get_id(), views))
self.assertEqual(1, v1.selected)
self.assertEqual(["Munich", "Berlin", "Hamburg"], v2.titles)
def test_update_error(self):
def update(control: ctrl.Control, text_area: ctrl.Input):
raise ValueError()
app = ds.app()
c1 = app.input("10")
c2 = app.input(handler=update, depends=c1)
controller = Controller(app.controls, [])
try:
controller.list(controller.list())
self.fail()
except ctrl.UpdateError as e:
self.assertEqual(c2.get_id(), e.id)
def test_title_override(self):
class Item:
def __init__(self, id, title):
self.id = id
self.title = title
def __repr__(self):
return self.title
app = ds.app()
app.select([Item(1, "hello"), Item(2, "world")], title=lambda x: x.title.upper())
controller = Controller(app.controls, [])
views = controller.list()
items_view = ty.cast(ctrl.SelectView, views[0])
self.assertEqual(["HELLO", "WORLD"], items_view.titles)
def test_controller_apply(self):
def update(control, text_field):
control.text = str(int(text_field.text) * 2)
app = ds.app()
c1 = app.input(text="10")
c2 = app.input(handler=update, depends=c1)
def test():
return 30
o1 = app.output(data=test())
controller = Controller(app.controls, [])
views = controller.list()
self.assertEqual(30, views[2].data)
def test_dependant_check_box(self):
def get_selected(self: ctrl.Checkbox, c2: ctrl.Input):
self.selected = int(c2.value()) > 5
app = ds.app()
c1 = app.input(text="10")
c2 = app.checkbox(handler=get_selected, depends=[c1])
controller = Controller(app.controls, [])
views = controller.list()
v1 = ty.cast(ctrl.InputView, self.get_by_id(c1.get_id(), views))
v2 = ty.cast(ctrl.CheckboxView, self.get_by_id(c2.get_id(), views))
self.assertEqual("10", v1.text)
self.assertEqual(True, v2.selected)
v1.text = "5"
views = controller.list(views)
v1 = ty.cast(ctrl.InputView, self.get_by_id(c1.get_id(), views))
v2 = ty.cast(ctrl.CheckboxView, self.get_by_id(c2.get_id(), views))
self.assertEqual("5", v1.text)
self.assertEqual(False, v2.selected)
def test_simple_update(self):
def update(control: ctrl.Control, text_field: ctrl.Input):
control.text = str(int(text_field.text) * 2)
app = ds.app()
c1 = app.input(text="10")
c2 = app.input(handler=update, depends=c1)
controller = Controller(app.controls, [])
views = controller.list()
self.assertEqual(2, len(views))
ids = [v.id for v in views]
self.assertIn(c1.get_id(), ids)
self.assertIn(c2.get_id(), ids)
v1 = self.get_by_id(c1.get_id(), views)
v2 = self.get_by_id(c2.get_id(), views)
if isinstance(v1, ctrl.InputView):
self.assertEqual("10", v1.text)
else:
self.fail()
if isinstance(v2, ctrl.InputView):
self.assertEqual("20", v2.text)
else:
self.fail()
views = controller.list(views)
v1 = self.get_by_id(c1.get_id(), views)
v2 = self.get_by_id(c2.get_id(), views)
if isinstance(v1, ctrl.InputView):
self.assertEqual("10", v1.text)
else:
self.fail()
if isinstance(v2, ctrl.InputView):
self.assertEqual("20", v2.text)
def test_controller_event(self):
app = ds.app()
def i1_handler(self):
print("i1_handler")
if self.text is None:
self.text = "Andrey"
i1 = app.input(handler=i1_handler)
def i2_handler(self, i1):
print("i2_handler")
self.text = i1.text.upper()
i2 = app.input(handler=i2_handler, depends=[i1])
cb1 = app.checkbox()
def m1_handler(self, i2, cb1):
print("m1_handler")
self.text = i2.text.upper() if cb1.selected else i2.text.lower()
app.markdown(handler=m1_handler, depends=[i2, cb1], require_apply=True)
controller = Controller(app.controls, [])
print("Event: empty")
views = controller.list()
for view in views:
print(str(view.pack()))
print("Event: apply")
views = controller.list(views, event=ctrl.Event(type="apply", source=None))
for view in views:
print(str(view.pack()))
print("Event: change i1")
views[0].text = 'Sergey'
views = controller.list(views, event=ctrl.Event(type="change", source=i1.get_id()))
for view in views:
print(str(view.pack()))
def test_multiple_combo_box(self):
def update(control: ctrl.Select, parent: ctrl.Select):
selected = [parent.items[s] for s in parent.selected]
control.items = [f"{selected} 1", f"{selected} 2"]
app = ds.app()
cb = app.select(["Hello", "World"], multiple=True)
self.assertTrue(isinstance(cb._derive_model(), ctrl.DefaultListModel))
c1 = app.select(handler=update, depends=cb)
controller = Controller(app.controls, [])
views = controller.list()
v = ty.cast(ctrl.SelectView, self.get_by_id(cb.get_id(), views))
v1 = ty.cast(ctrl.SelectView, self.get_by_id(c1.get_id(), views))
self.assertEqual([], v.selected)
self.assertEqual(["[] 1", "[] 2"], v1.titles)
v.selected = [1]
print(views)
views = controller.list(views)
print(views)
v = ty.cast(ctrl.SelectView, self.get_by_id(cb.get_id(), views))
v1 = ty.cast(ctrl.SelectView, self.get_by_id(c1.get_id(), views))
self.assertEqual([1], v.selected)
self.assertEqual(["['World'] 1", "['World'] 2"], v1.titles)
v.selected = [0, 1]
print(views)
views = controller.list(views)
print(views)
v = ty.cast(ctrl.SelectView, self.get_by_id(cb.get_id(), views))
v1 = ty.cast(ctrl.SelectView, self.get_by_id(c1.get_id(), views))
self.assertEqual([0, 1], v.selected)
self.assertEqual(["['Hello', 'World'] 1", "['Hello', 'World'] 2"], v1.titles)
def test_tqdm(self):
tqdm_state = {}
class Handler(TqdmHandler):
def close(self, tqdm: tqdm):
tqdm_state[tqdm.n] = {"desc": tqdm.desc, "n": tqdm.n, "total": tqdm.total,
"elapsed": tqdm.format_dict["elapsed"]}
def display(self, tqdm: tqdm):
tqdm_state[tqdm.n] = {"desc": tqdm.desc, "n": tqdm.n, "total": tqdm.total,
"elapsed": tqdm.format_dict["elapsed"]}
set_tqdm_handler(Handler())
def output_handler(output):
for _ in trange(3, desc="Calculating data"):
time.sleep(0.25)
output.data = "success"
app = ds.app()
app.output(handler=output_handler)
controller = Controller(app.controls, [])
views = controller.list()
def apply():
controller.list(views)
apply()
for i in range(4):
self.assertEqual(i, tqdm_state[i]["n"])
self.assertEqual(3, tqdm_state[i]["total"])
self.assertEqual("Calculating data", tqdm_state[i]["desc"])
set_tqdm_handler(None)
def test_file_uploader(self):
app = ds.app()
file_uploader = app.uploader()
self.assertTrue(isinstance(file_uploader.uploads, list))
controller = Controller(app.controls, [])
views = controller.list()
file_uploader_view = ty.cast(ctrl.UploaderView, self.get_by_id(file_uploader.get_id(), views))
self.assertTrue(isinstance(file_uploader_view.uploads, list))
self.assertEqual({"uploads": []}, file_uploader_view._pack())
today = date.today()
file_uploader_view.uploads.append(ctrl.Upload("some_file_id", "some_file_name", 123, today))
packed_view = {"id": file_uploader._id, "colspan": 2, "container": "main",
"enabled": True, "label": None, "optional": False, "visible": True, "rowspan": 1,
"type": "UploaderView", "uploads": [
{"id": "some_file_id", "file_name": "some_file_name", "length": 123,
"created_date": today.strftime("%Y-%m-%d")}]}
self.assertEqual(packed_view, file_uploader_view.pack())
file_uploader._apply(file_uploader_view) # TODO: Clean me
value = file_uploader.value()
self.assertTrue(isinstance(value, list))
self.assertEqual(len(value), 1)
self.assertEqual(value[0].id, "some_file_id")
self.assertEqual(value[0].file_name, "some_file_name")
self.assertEqual(value[0].length, 123)
self.assertEqual(value[0].created_date, today)
unpacked_view = ctrl.unpack_view(packed_view)
self.assertTrue(isinstance(unpacked_view, ctrl.UploaderView))
self.assertTrue(isinstance(unpacked_view.uploads, list))
self.assertEqual(len(unpacked_view.uploads), 1)
unpacked_upload = unpacked_view.uploads[0]
self.assertEqual(unpacked_upload.id, "some_file_id")
self.assertEqual(unpacked_upload.file_name, "some_file_name")
self.assertEqual(unpacked_upload.length, 123)
self.assertEqual(unpacked_upload.created_date, today)
def encode(self, app: 'Application', description: ty.Optional[str], params: ty.Optional[ty.Dict]) -> FrameData:
containers = []
if app._sidebar:
containers.append(Container(app._sidebar.id, app._sidebar.layout, app._sidebar.columns))
containers.append(Container(app.id, app.layout, app.columns))
controller = Controller(app.controls, containers)
stage_dir = util.create_path(self._temp_dir)
_stage_deps(app.deps(), stage_dir)
_serialize(controller, stage_dir / "controller.pickle")
archived = util.create_filename(self._temp_dir)
filename = shutil.make_archive(archived, self._archive, stage_dir)
settings = {"cloudpickle": cloudpickle.__version__, "archive": self._archive}
return FrameData(FileContent(Path(filename)),
MediaType("application/octet-stream", "application/python"),
description, params, settings)
def test_markdown(self):
app = ds.app()
app.markdown(text="Hello, **this** is Markdown")
controller = Controller(app.controls, [])
views = controller.list()
self.assertTrue(isinstance(views[0].data, _Markdown))
self.assertEquals("Hello, **this** is Markdown", views[0].data.text)
app = ds.app()
t1 = app.input(text="Markdown")
def m_handler(self: ctrl.Markdown, t1: ctrl.Input):
self.text = "Hello, **this** is " + t1.text
app.markdown(handler=m_handler, depends=[t1])
controller = Controller(app.controls, [])
views = controller.list()
self.assertTrue(isinstance(views[1].data, _Markdown))
self.assertEquals("Hello, **this** is Markdown", views[1].data.text)