Skip to content
This repository was archived by the owner on Apr 25, 2024. It is now read-only.

Commit b89539e

Browse files
author
François Guyot
committed
pass quality checks
1 parent 6fcec55 commit b89539e

File tree

1 file changed

+16
-30
lines changed

1 file changed

+16
-30
lines changed

src/pyk/kcfg/tui.py

Lines changed: 16 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
from __future__ import annotations
22

33
from enum import Enum, auto
4-
from functools import cached_property
54
from typing import TYPE_CHECKING, Union
65

76
from textual.app import App
@@ -306,31 +305,23 @@ def __init__(
306305
self._kcfg_nodes = []
307306
kcfg_show = KCFGShow(kprint)
308307
self._node_idx = {}
309-
seg = kcfg_show.pretty_segments(
310-
self._kcfg, minimize=self._minimize, node_printer=self._node_printer
311-
)
308+
seg = kcfg_show.pretty_segments(self._kcfg, minimize=self._minimize, node_printer=self._node_printer)
312309
for lseg_id, node_lines in seg:
313310
self._kcfg_nodes.append(GraphChunk(lseg_id, node_lines))
314311

315312
def last_idx(self) -> int:
316313
return int(self._selected_chunk.rsplit('_', 1)[1])
317314

318315
def pos_of(self, kcfg_id: str) -> int | None:
319-
# return any(node.id for node in self._kcfg_nodes if node.id == kcfg_id)
320316
for i, node in enumerate(self._kcfg_nodes):
321317
if node.id == kcfg_id:
322318
return i
323319

324-
# def pos_of(self, kcfg_id: str) -> int | None:
325-
# return next((i for i, node in enumerate(self._kcfg_nodes) if node.id == kcfg_id), None)
326-
327-
328320
def next_node_from(self, i: int = 0) -> str | None:
329-
li: list[str] = [node.id for node in self._kcfg_nodes[(i + 1):] if not node.id.startswith("unknown")]
321+
li: list[str] = [node.id for node in self._kcfg_nodes[(i + 1) :] if not node.id.startswith('unknown')]
330322
try:
331-
# return next(li)
332323
return li[0]
333-
except:
324+
except StopIteration:
334325
return None
335326

336327
def next_node(self) -> str | None:
@@ -339,10 +330,10 @@ def next_node(self) -> str | None:
339330
return self.next_node_from(pos)
340331

341332
def prev_node_from(self, i: int) -> str | None:
342-
li = reversed(list(node.id for node in self._kcfg_nodes[:i] if not node.id.startswith("unknown")))
333+
li = reversed([node.id for node in self._kcfg_nodes[:i] if not node.id.startswith('unknown')])
343334
try:
344335
return next(li)
345-
except:
336+
except StopIteration:
346337
return None
347338

348339
def prev_node(self) -> str | None:
@@ -367,23 +358,23 @@ def _resolve_any(self, kcfg_id: str) -> KCFGElem:
367358
node, *_ = kcfg_id[5:].split('_')
368359
node_id = int(node)
369360
return self._kcfg.node(node_id)
370-
elif kcfg_id.startswith('edge_'):
361+
elif kcfg_id.startswith('edge_'):
371362
node, *_ = kcfg_id[5:].split('_')
372363
node_source, node_target, *_ = kcfg_id[5:].split('_')
373364
source_id = int(node_source)
374365
target_id = int(node_target)
375366
return single(self._kcfg.edges(source_id=source_id, target_id=target_id))
376-
elif kcfg_id.startswith('cover_'):
367+
elif kcfg_id.startswith('cover_'):
377368
node_source, node_target, *_ = kcfg_id[6:].split('_')
378369
source_id = int(node_source)
379370
target_id = int(node_target)
380371
return single(self._kcfg.covers(source_id=source_id, target_id=target_id))
381-
elif kcfg_id.startswith('split_'):
372+
elif kcfg_id.startswith('split_'):
382373
node_source, node_target, *_ = kcfg_id[6:].split('_')
383374
source_id = int(node_source)
384375
target_id = int(node_target)
385376
return single(self._kcfg.splits(source_id=source_id, target_id=target_id))
386-
elif kcfg_id.startswith('ndbranch_'):
377+
elif kcfg_id.startswith('ndbranch_'):
387378
node_source, node_target, *_ = kcfg_id[8:].split('_')
388379
source_id = int(node_source)
389380
target_id = int(node_target)
@@ -398,20 +389,15 @@ def on_mount(self) -> None:
398389
next_node = self.next_node_from(0)
399390
if next_node != None:
400391
self._selected_chunk = next_node
401-
self.query_one(f'#{self._selected_chunk}', GraphChunk).set_styles(
402-
'border-left: double red;'
403-
)
392+
self.query_one(f'#{self._selected_chunk}', GraphChunk).set_styles('border-left: double red;')
404393
self.query_one('#node-view', NodeView).update(self._resolve_any(next_node))
405394

406395
def on_graph_chunk_selected(self, message: GraphChunk.Selected) -> None:
407-
try:
408-
self.query_one(f'#{self._selected_chunk}', GraphChunk).set_styles('border: none;')
409-
kcfg_elem = self._resolve_any(message.chunk_id)
410-
self._selected_chunk = message.chunk_id
411-
self.query_one(f'#{self._selected_chunk}', GraphChunk).set_styles('border-left: double red;')
412-
self.query_one('#node-view', NodeView).update(kcfg_elem)
413-
except:
414-
pass
396+
self.query_one(f'#{self._selected_chunk}', GraphChunk).set_styles('border: none;')
397+
kcfg_elem = self._resolve_any(message.chunk_id)
398+
self._selected_chunk = message.chunk_id
399+
self.query_one(f'#{self._selected_chunk}', GraphChunk).set_styles('border-left: double red;')
400+
self.query_one('#node-view', NodeView).update(kcfg_elem)
415401

416402
BINDINGS = [
417403
('f', 'keystroke("f")', 'Fold node'),
@@ -590,7 +576,7 @@ async def action_keystroke(self, key: str) -> None:
590576
elif key == 'q':
591577
await self.action_quit()
592578

593-
self._buffer = list()
579+
self._buffer = []
594580

595581
if key == 'change-window':
596582
self._buffer.append(key)

0 commit comments

Comments
 (0)