From 9f131a4019eeb419c32769323f1233c6b176d2f9 Mon Sep 17 00:00:00 2001
From: Ricardo <electropepper@protonmail.com>
Date: Thu, 7 Jul 2022 20:30:02 +0000
Subject: [PATCH] Added test.py file

---
 main.py |   5 ++-
 test.py | 125 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 128 insertions(+), 2 deletions(-)
 create mode 100644 test.py

diff --git a/main.py b/main.py
index 11f780f..17deaec 100755
--- a/main.py
+++ b/main.py
@@ -66,6 +66,7 @@ class Menu(object):
 class MyApp(object):
 
     def __init__(self, screen):
+        screen = curses.initscr()
         curses.curs_set(0)
         curses.start_color()
         curses.init_color(0,0,0,0)
@@ -77,7 +78,7 @@ class MyApp(object):
         cols56 = (curses.COLS - 2) * 0.56   # Take 2 characters out from the boarders and then calculate 56%
         cols56 = int(cols56) + 1            # Ditch the fractional part and add 1
 
-        lines = (curses.LINES - 2) * 0.95   # Take 2 characters out from the boarders and then calculate 56%
+        lines = (curses.LINES - 2) * 0.95   # Take 2 characters out from the boarders and then calculate 95%
         lines = int(lines)                  # Ditch the fractional part
 
 
@@ -109,7 +110,7 @@ class MyApp(object):
 #        resistors_items = [("10R", curses.beep), ("10K", curses.beep), ("330R", curses.flash), ("57K", curses.flash)]
 #        resistorsMenu = Menu(resistors_items, parts)
 
-        components_items = [("Resistors", (r = 1)), ("Capacitors", curses.beep), ("Coils", curses.flash), ("IC\'s", curses.flash)]
+        components_items = [("Resistors", curses.beep), ("Capacitors", curses.beep), ("Coils", curses.flash), ("IC\'s", curses.flash)]
         componentsMenu = Menu(components_items, components)
         componentsMenu.display()
 
diff --git a/test.py b/test.py
new file mode 100644
index 0000000..ab49843
--- /dev/null
+++ b/test.py
@@ -0,0 +1,125 @@
+#!/bin/env python3
+
+import urwid as u
+
+class ListItem(u.WidgetWrap):
+
+    def __init__ (self, country):
+
+        self.content = country
+
+        name = country["name"]
+
+        t = u.AttrWrap(u.Text(name), "country", "country_selected")
+
+        u.WidgetWrap.__init__(self, t)
+
+    def selectable (self):
+        return True
+
+    def keypress(self, size, key):
+        return key
+
+class ListView(u.WidgetWrap):
+
+    def __init__(self):
+
+        u.register_signal(self.__class__, ['show_details'])
+
+        self.walker = u.SimpleFocusListWalker([])
+
+        lb = u.ListBox(self.walker)
+
+        u.WidgetWrap.__init__(self, lb)
+
+    def modified(self):
+
+        focus_w, _ = self.walker.get_focus()
+
+        u.emit_signal(self, 'show_details', focus_w.content)
+
+    def set_data(self, countries):
+
+        countries_widgets = [ListItem(c) for c in countries]
+
+        u.disconnect_signal(self.walker, 'modified', self.modified)
+
+        while len(self.walker) > 0:
+            self.walker.pop()
+
+        self.walker.extend(countries_widgets)
+
+        u.connect_signal(self.walker, "modified", self.modified)
+
+        self.walker.set_focus(0)
+
+class DetailView(u.WidgetWrap):
+
+    def __init__ (self):
+        t = u.Text("")
+        u.WidgetWrap.__init__(self, t)
+
+    def set_country(self, c):
+        s = f'Name: {c["name"]}\nPop:  {c["pop"]}\nGDP:  {c["gdp"]}'
+        self._w.set_text(s)
+
+class App(object):
+
+    def unhandled_input(self, key):
+        if key in ('q',):
+            raise u.ExitMainLoop()
+
+    def show_details(self, country):
+        self.detail_view.set_country(country)
+
+    def __init__(self):
+
+        self.palette = {
+            ("bg",               "black",       "white"),
+            ("country",          "black",       "white"),
+            ("country_selected", "black",       "yellow"),
+            ("footer",           "white, bold", "dark red")
+        }
+
+        self.list_view = ListView()
+        self.detail_view = DetailView()
+
+        u.connect_signal(self.list_view, 'show_details', self.show_details)
+
+        footer = u.AttrWrap(u.Text(" Q to exit"), "footer")
+
+        col_rows = u.raw_display.Screen().get_cols_rows()
+        h = col_rows[0] - 2
+
+        f1 = u.Filler(self.list_view, valign='top', height=h)
+        f2 = u.Filler(self.detail_view, valign='top')
+
+        c_list = u.LineBox(f1, title="Countries")
+        c_details = u.LineBox(f2, title="Country Details")
+
+        columns = u.Columns([('weight', 30, c_list), ('weight', 70, c_details)])
+
+        frame = u.AttrMap(u.Frame(body=columns, footer=footer), 'bg')
+
+        self.loop = u.MainLoop(frame, self.palette, unhandled_input=self.unhandled_input)
+
+    def update_data(self):
+
+        l = [] # https://databank.worldbank.org/embed/Population-and-GDP-by-Country/id/29c4df41
+        l.append({"name":"USA",     "pop":"325,084,756",   "gdp":"$ 19.485 trillion"})
+        l.append({"name":"China",   "pop":"1,421,021,791", "gdp":"$ 12.238 trillion"})
+        l.append({"name":"Japan",   "pop":"127,502,725",   "gdp":"$ 4.872 trillion"})
+        l.append({"name":"Germany", "pop":"82,658,409",    "gdp":"$ 3.693 trillion"})
+        l.append({"name":"India",   "pop":"1,338,676,785", "gdp":"$ 2.651 trillion"})
+
+        self.list_view.set_data(l)
+
+    def start(self):
+
+        self.update_data()
+        self.loop.run()
+
+if __name__ == '__main__':
+
+    app = App()
+    app.start()