From fc3367181b5770375b0836876ca0e7257e62b48c Mon Sep 17 00:00:00 2001 From: Simone Margaritelli Date: Fri, 11 Oct 2019 17:24:27 +0200 Subject: [PATCH] fix: better data encapsulation --- pwnagotchi/plugins/default/grid.py | 22 ++++++++++++---------- 1 file changed, 12 insertions(+), 10 deletions(-) diff --git a/pwnagotchi/plugins/default/grid.py b/pwnagotchi/plugins/default/grid.py index 96d290a..2574d57 100644 --- a/pwnagotchi/plugins/default/grid.py +++ b/pwnagotchi/plugins/default/grid.py @@ -76,16 +76,18 @@ def grid_update_data(last_session): pass data = { - 'duration': last_session.duration, - 'epochs': last_session.epochs, - 'train_epochs': last_session.train_epochs, - 'avg_reward': last_session.avg_reward, - 'min_reward': last_session.min_reward, - 'max_reward': last_session.max_reward, - 'deauthed': last_session.deauthed, - 'associated': last_session.associated, - 'handshakes': last_session.handshakes, - 'peers': last_session.peers, + 'session': { + 'duration': last_session.duration, + 'epochs': last_session.epochs, + 'train_epochs': last_session.train_epochs, + 'avg_reward': last_session.avg_reward, + 'min_reward': last_session.min_reward, + 'max_reward': last_session.max_reward, + 'deauthed': last_session.deauthed, + 'associated': last_session.associated, + 'handshakes': last_session.handshakes, + 'peers': last_session.peers, + }, 'uname': subprocess.getoutput("uname -a"), 'brain': brain, 'version': pwnagotchi.version