tworu/dzgui
1
0 Fork 0
mirror of https://github.com/aclist/dztui.git synced 2026-10-31 23:22:44 +01:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat: _sort_unique_maps()

Browse Source
This commit is contained in:
aclist committed 2026-02-27 23:24:30 +09:00
1 parent 940dd41fdf
commit 3b40ea1846
1 file changed
+17 -7
Download Patch File Download Diff File Expand all files Collapse all files
dzgui/model/servers.py
+17 -7
View File
@@ -204,7 +204,7 @@ class ServerModelManager:
self.thread_man.set_cleanup_func(StoredFunc(self._cleanup_on_failure)) self.thread_man.set_cleanup_func(StoredFunc(self._cleanup_on_failure))
return return
record = Servers.parse_json([response]) records = Servers.parse_json([response])
proxy_man = self._get_proxy_man() proxy_man = self._get_proxy_man()
raw_model = proxy_man.get_control() raw_model = proxy_man.get_control()
@@ -217,11 +217,16 @@ class ServerModelManager:
if raw_model is None: if raw_model is None:
return return
raw_model.append(record[0]) # NOTE: expected to only contain one item
raw_model.append(records[0])
# TODO: if all filters are already applied, strange behavior may occur # TODO: if all filters are already applied, strange behavior may occur
# -> need to insert and update per current filters # -> need to insert and reupdate tre per current filters
# for example, non-empty will only show up in empty because it is not cached
proxy_man.filter(FilterMode.INITIAL) proxy_man.filter(FilterMode.INITIAL)
self._sort_unique_maps(records)
self.thread_man.set_cleanup_func(StoredFunc(self._cleanup_single_ip)) self.thread_man.set_cleanup_func(StoredFunc(self._cleanup_single_ip))
def _dump_history(self) -> None: def _dump_history(self) -> None:
@@ -276,8 +281,10 @@ class ServerModelManager:
self.first_iteration = False self.first_iteration = False
def _cleanup_on_success(self) -> None: def _cleanup_on_success(self) -> None:
self.tv.set_model(None) proxy = self._get_proxy_man().get_proxy_model()
self.tv.set_model(self.to_insert) self.tv.set_model(proxy)
# self.tv.set_model(None)
# self.tv.set_model(self.to_insert)
# TODO: make sure control model len is N + 1 # TODO: make sure control model len is N + 1
# inserting a row serializes file on disk, updates control model for that tab, and updates model # inserting a row serializes file on disk, updates control model for that tab, and updates model
@@ -325,11 +332,14 @@ class ServerModelManager:
self.to_insert = manager.get_proxy_model() self.to_insert = manager.get_proxy_model()
# TODO: abstract for all methods # TODO: abstract for all methods
u_maps = set([row[1] for row in data]) self._sort_unique_maps(data)
self._set_new_maps(sorted(u_maps))
self.thread_man.set_cleanup_func(StoredFunc(self._cleanup_on_success)) self.thread_man.set_cleanup_func(StoredFunc(self._cleanup_on_success))
def _sort_unique_maps(self, data: list) -> None:
u_maps = set([row[1] for row in data])
self._set_new_maps(u_maps)
def _set_new_maps(self, maps: list[str]) -> None: def _set_new_maps(self, maps: list[str]) -> None:
self.new_maps = maps self.new_maps = maps