chrissewell/pwnagotchi
1
0
Fork 0
mirror of https://github.com/jayofelony/pwnagotchi.git synced 2025-07-01 18:37:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
Jeroen Oudshoorn
2024-01-05 14:07:18 +01:00
parent 2f4b05db85
commit 0931a8c10a
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pwnagotchi/agent.py
View File

@ -14,12 +14,12 @@ from pwnagotchi.automata import Automata
from pwnagotchi.log import LastSession
from pwnagotchi.bettercap import Client
from pwnagotchi.mesh.utils import AsyncAdvertiser
# from pwnagotchi.ai.train import AsyncTrainer
from pwnagotchi.ai.train import AsyncTrainer
RECOVERY_DATA_FILE = '/root/.pwnagotchi-recovery'
class Agent(Client, Automata, AsyncAdvertiser):
class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer):
def __init__(self, view, config, keypair):
Client.__init__(self, config['bettercap']['hostname'],
config['bettercap']['scheme'],
@ -28,7 +28,7 @@ class Agent(Client, Automata, AsyncAdvertiser):
config['bettercap']['password'])
Automata.__init__(self, config, view)
AsyncAdvertiser.__init__(self, config, view, keypair)
# AsyncTrainer.__init__(self, config)
AsyncTrainer.__init__(self, config)
self._started_at = time.time()
self._filter = None if not config['main']['filter'] else re.compile(config['main']['filter'])
@ -129,7 +129,7 @@ class Agent(Client, Automata, AsyncAdvertiser):
time.sleep(1)
def start(self):
# self.start_ai()
self.start_ai()
self._wait_bettercap()
self.setup_events()
self.set_starting()