Guards-List/ShavtzakScreen.py at master · Kfirul/Guards-List · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
from kivy.uix.boxlayout import BoxLayout
from kivymd.uix.toolbar import MDTopAppBar
from kivy.uix.screenmanager import Screen
from kivy.uix.scrollview import ScrollView
from kivymd.uix.list import OneLineListItem, MDList


class ShavtzakScreen(Screen):
    def __init__(self, shavtzak_instance, **kwargs):
        super(ShavtzakScreen, self).__init__(**kwargs)
        self.shavtzak_instance = shavtzak_instance
        self.shavtzak_instance.createGuardsList()

        # Create a toolbar for the screen
        self.toolbar = MDTopAppBar(
            title="Shavtzak Screen",
            md_bg_color=(0.2, 0.7, 0.5, 1),
            right_action_items=[["arrow-right", lambda x: self.back()]]
        )
        self.toolbar.pos_hint = {"top": 1}
        self.add_widget(self.toolbar)

        # Create a layout to hold the information
        self.layout = BoxLayout(orientation="vertical", spacing=10, padding=10)
        # Add the layout to the screen
        self.add_widget(self.layout)
        self.bind(on_enter=self.update_result)

    def update_result(self, *args):
        self.shavtzak_instance.createGuardsList()
        self.layout.clear_widgets()

        # Create horizontal layout for guards list and positions list
        self.horizontal_layout = BoxLayout(orientation="horizontal", spacing=10)

        positions_list = ScrollView(size_hint=(0.5, 0.9))
        positions_list_inner = MDList()

        positions_label = OneLineListItem(text="Positions List:", theme_text_color="Secondary")
        positions_list_inner.add_widget(positions_label)

        for position in self.shavtzak_instance.positionList:
            item = OneLineListItem(text=position.name)
            item.bind(on_release=lambda x, p=position: self.on_position_click(p))
            positions_list_inner.add_widget(item)

        positions_list.add_widget(positions_list_inner)  # Add the MDList to the ScrollView

        self.horizontal_layout.add_widget(positions_list)

        self.guards_list = MDList()

        guards_label = OneLineListItem(text="Guards List:", theme_text_color="Secondary")
        self.guards_list.add_widget(guards_label)

        # Create a vertical layout for guards (initially hidden)
        self.guards_layout = ScrollView(size_hint=(0.75, 0.9), opacity=0)
        self.guards_list_inner = MDList()
        self.guards_layout.add_widget(self.guards_list_inner)
        self.horizontal_layout.add_widget(self.guards_layout)

        self.layout.add_widget(self.horizontal_layout)

    def on_position_click(self, position):
        # Display guards for the selected position
        self.display_guards_list(position)

    def display_guards_list(self, position):
        # Clear existing guards list
        self.guards_list_inner.clear_widgets()

        # Add the position name as the title
        title_item = OneLineListItem(text=f"Guards List for {position.name}", theme_text_color="Secondary",
                                     font_style="H6")
        self.guards_list_inner.add_widget(title_item)

        # Add guards for this position
        for guard in position.guardList:
            item = OneLineListItem(text=f" {guard}", theme_text_color="Primary", divider=None)
            self.guards_list_inner.add_widget(item)

        # Show guards layout
        self.guards_layout.opacity = 1

    def back(self):
        print("Going back to the main screen")
        # Switch back to the main screen
        self.manager.current = "main_screen"