|
|
|
|
| import wx
|
|
|
| class MainFrame(wx.Frame):
|
|
|
| MAX_WIDTH = 1000
|
| MAX_HEIGHT = 1000
|
| MIN_WIDTH = 300
|
| MIN_HEIGHT = 200
|
|
|
| def __init__(self, *args, hardware_info=None, result_queue=None, **kwargs):
|
| super().__init__(*args, **kwargs)
|
| self.hardware_info = hardware_info or {}
|
| self.result_queue = result_queue
|
|
|
| self.selected_components = {
|
| 'cpu': True,
|
| 'ram': True,
|
| 'gpu': True,
|
| 'network': True,
|
| 'drives': []
|
| }
|
|
|
|
|
| self.Bind(wx.EVT_CLOSE, self.on_close)
|
|
|
|
|
| self.checkboxes = {}
|
|
|
| self.panel = wx.Panel(self)
|
| vbox = wx.BoxSizer(wx.VERTICAL)
|
|
|
| info = wx.StaticText(self.panel, label="Window will self-close in 10 sec")
|
| vbox.Add(info, 0, wx.ALL, 10)
|
|
|
| auto_close_info = wx.StaticText(
|
| self.panel,
|
| label="Drive monitoring (read/write) every second\nThreshold 2MB"
|
| )
|
| vbox.Add(auto_close_info, 0, wx.LEFT | wx.BOTTOM, 10)
|
|
|
| tray_info = wx.StaticText(
|
| self.panel,
|
| label="(Hover over tray icon to see further information)"
|
| )
|
| vbox.Add(tray_info, 0, wx.LEFT | wx.BOTTOM, 10)
|
|
|
|
|
| if hardware_info.get('cpu_info'):
|
| cpu_info = hardware_info['cpu_info']
|
| cpu_label = (f"CPU Info: Logical cores: {cpu_info.get('logical_cores')}, "
|
| f"Physical cores: {cpu_info.get('physical_cores')}, "
|
| f"Frequency: {cpu_info.get('frequency', 'N/A')} MHz")
|
| checkbox = wx.CheckBox(self.panel, label=cpu_label)
|
| checkbox.SetValue(True)
|
| vbox.Add(checkbox, 0, wx.LEFT | wx.BOTTOM, 10)
|
| self.checkboxes['cpu'] = [checkbox]
|
|
|
|
|
| if hardware_info.get('ram_info'):
|
| ram_info = hardware_info['ram_info']
|
| ram_label = (f"RAM Info: Total: {ram_info.get('total_gb', 'N/A')} GB, "
|
| f"Available: {ram_info.get('available_gb', 'N/A')} GB")
|
| checkbox = wx.CheckBox(self.panel, label=ram_label)
|
| checkbox.SetValue(True)
|
| vbox.Add(checkbox, 0, wx.LEFT | wx.BOTTOM, 10)
|
| self.checkboxes['ram'] = [checkbox]
|
|
|
|
|
| if hardware_info.get('gpu_info'):
|
| self.add_section(vbox, "GPU Info:", [
|
| f"{gpu['name']} ({gpu['memory_total_mb']} MB VRAM / MaxTemp: {gpu['max_temp']} °C)"
|
| for gpu in hardware_info['gpu_info']
|
| ], category="gpu")
|
|
|
|
|
| if hardware_info.get('network_adapters'):
|
| self.add_section(vbox, "Active Network Adapters:", hardware_info['network_adapters'], category="network")
|
|
|
|
|
| if hardware_info.get('drive_map'):
|
| self.add_drive_section(vbox)
|
|
|
|
|
| self.submit_button = wx.Button(self.panel, label="Submit and Close (10)")
|
| self.submit_button.Bind(wx.EVT_BUTTON, self.on_submit)
|
| vbox.Add(self.submit_button, 0, wx.ALL | wx.CENTER, 10)
|
|
|
| self.panel.SetSizer(vbox)
|
| self.panel.Layout()
|
| self.submit_button.SetFocus()
|
| self.SetDefaultItem(self.submit_button)
|
|
|
| best_size = self.panel.GetBestSize()
|
| width = min(max(best_size.width, self.MIN_WIDTH), self.MAX_WIDTH)
|
| height = min(max(best_size.height, self.MIN_HEIGHT), self.MAX_HEIGHT)
|
| self.SetClientSize((width, height))
|
| self.SetPosition((200,100))
|
| self.SetTitle("SmartTaskTool by Sevenof9")
|
|
|
| self.countdown_timer = 10
|
| self.timer = wx.Timer(self)
|
| self.Bind(wx.EVT_TIMER, self.update_countdown, self.timer)
|
| self.timer.Start(1000)
|
|
|
| def update_countdown(self, event):
|
| self.countdown_timer -= 1
|
| if self.countdown_timer <= 0:
|
| self.submit_values()
|
| self.timer.Stop()
|
| else:
|
| self.submit_button.SetLabel(f"Submit and Close ({self.countdown_timer})")
|
|
|
| def add_drive_section(self, vbox):
|
| drive_map = self.hardware_info.get('drive_map', {})
|
| vbox.Add(wx.StaticText(self.panel, label="Detected Drives:"), 0, wx.LEFT | wx.TOP, 10)
|
| for dev in sorted(drive_map.keys()):
|
| parts = drive_map[dev]
|
| for part_info in sorted(parts, key=lambda x: x['letter']):
|
| letter = part_info.get('letter', 'N/A')
|
| label = part_info.get('label', 'N/A')
|
| checkbox_label = f"{dev} - {letter} ({label})"
|
| checkbox = wx.CheckBox(self.panel, label=checkbox_label)
|
| checkbox.SetValue(True)
|
| vbox.Add(checkbox, 0, wx.LEFT | wx.BOTTOM, 10)
|
|
|
| if 'drives' not in self.checkboxes:
|
| self.checkboxes['drives'] = []
|
| self.checkboxes['drives'].append((dev, letter, checkbox))
|
|
|
| def submit_values(self):
|
| selected_components = {
|
| 'cpu': False,
|
| 'ram': False,
|
| 'gpu': False,
|
| 'network': False,
|
| 'drives': []
|
| }
|
|
|
| checkbox_categories = ['cpu', 'ram', 'gpu', 'network']
|
| for category in checkbox_categories:
|
| if category in self.checkboxes:
|
| selected_components[category] = any(checkbox.GetValue() for checkbox in self.checkboxes[category])
|
|
|
| if 'drives' in self.checkboxes:
|
| for dev, part, checkbox in self.checkboxes['drives']:
|
| if checkbox.GetValue():
|
| selected_components['drives'].append((dev, part))
|
|
|
|
|
| selected_components['cpu_info'] = self.hardware_info.get('cpu_info', {})
|
| selected_components['ram_info'] = self.hardware_info.get('ram_info', {})
|
| selected_components['gpu_info'] = self.hardware_info.get('gpu_info', [])
|
| selected_components['network_adapters'] = self.hardware_info.get('network_adapters', [])
|
| selected_components['drive_map'] = self.hardware_info.get('drive_map', {})
|
|
|
| print("[DEBUG] Auswahl:", selected_components)
|
| if self.result_queue:
|
| self.result_queue.put(selected_components)
|
| self.Close()
|
|
|
| def on_submit(self, event):
|
| print("[DEBUG] Submit-Button geklickt")
|
| self.submit_values()
|
|
|
| def add_section(self, vbox, title, items, category):
|
| vbox.Add(wx.StaticText(self.panel, label=title), 0, wx.LEFT | wx.TOP, 10)
|
| for item in items:
|
| checkbox = wx.CheckBox(self.panel, label=item)
|
| checkbox.SetValue(True)
|
| vbox.Add(checkbox, 0, wx.LEFT | wx.BOTTOM, 10)
|
|
|
| if category not in self.checkboxes:
|
| self.checkboxes[category] = []
|
|
|
| if category == "drives":
|
|
|
| parts = item.split(":")
|
| if len(parts) == 2:
|
| dev, part = parts
|
| self.checkboxes[category].append((dev.strip(), part.strip(), checkbox))
|
| else:
|
| self.checkboxes[category].append(checkbox)
|
|
|
| def get_selected_components(self):
|
| selected = {
|
| 'cpu': False,
|
| 'ram': False,
|
| 'gpu': False,
|
| 'network': False,
|
| 'drives': []
|
| }
|
|
|
| for category, checkboxes in self.checkboxes.items():
|
| if category == 'drives':
|
|
|
| drive_map = self.hardware_info.get('drive_map', {})
|
| selected_drives = []
|
| for entry in self.checkboxes:
|
| if len(entry) == 3:
|
| category, checkbox, item = entry
|
| if category == "drives" and checkbox.GetValue():
|
| if ":" in item:
|
| dev, part = item.split(":", 1)
|
| selected_drives.append((dev.strip(), part.strip()))
|
|
|
| selected['drives'] = selected_drives
|
| else:
|
|
|
| if any(checkbox.GetValue() for checkbox in checkboxes):
|
| selected[category] = True
|
|
|
|
|
| selected['cpu_info'] = self.hardware_info.get('cpu_info', {})
|
| selected['ram_info'] = self.hardware_info.get('ram_info', {})
|
| selected['gpu_info'] = self.hardware_info.get('gpu_info', [])
|
| selected['network_adapters'] = self.hardware_info.get('network_adapters', [])
|
| selected_components['drive_map'] = self.hardware_info.get('drive_map', {})
|
|
|
| return selected
|
|
|
| def on_close(self, event):
|
| print("[DEBUG] Fenster wird geschlossen")
|
|
|
| if hasattr(self, 'timer') and self.timer.IsRunning():
|
| self.timer.Stop()
|
|
|
| if self.result_queue:
|
|
|
| selected_components = {
|
| 'cpu': False,
|
| 'ram': False,
|
| 'gpu': False,
|
| 'network': False,
|
| 'drives': [],
|
| 'cpu_info': self.hardware_info.get('cpu_info', {}),
|
| 'ram_info': self.hardware_info.get('ram_info', {}),
|
| 'gpu_info': self.hardware_info.get('gpu_info', []),
|
| 'network_adapters': self.hardware_info.get('network_adapters', []),
|
| 'drive_map': self.hardware_info.get('drive_map', {})
|
| }
|
| self.result_queue.put(selected_components)
|
|
|
| self.Destroy()
|
|
|
| app = wx.GetApp()
|
| if app:
|
| app.ExitMainLoop()
|
|
|
|
|
|
|
|
|
|
|
| if __name__ == "__main__":
|
| app = wx.App(False)
|
|
|
|
|
| hardware_info = {
|
| 'cpu_info': {
|
| 'logical_cores': 8,
|
| 'physical_cores': 4,
|
| 'frequency': 3200,
|
| },
|
| 'ram_info': {
|
| 'total_gb': 16,
|
| 'available_gb': 10
|
| },
|
| 'gpu_info': [
|
| {'name': 'NVIDIA GTX 1080', 'memory_total_mb': 8192, 'max_temp': 84}
|
| ],
|
| 'network_adapters': ['Ethernet', 'Wi-Fi'],
|
| 'drive_map': {
|
| 'Disk0': [{'letter': 'C:', 'label': 'System'}, {'letter': 'D:', 'label': 'Recovery'}],
|
| 'Disk1': [{'letter': 'E:', 'label': 'Data'}]
|
| }
|
|
|
|
|
| }
|
|
|
| frame = MainFrame(None, hardware_info=hardware_info)
|
| frame.Show()
|
| app.MainLoop()
|
|
|
