Merge pull request #1333 from cglewis/master

check if 'data' exists in the response
faucetsdn · Sep 17, 2019 · 9fd65fd · 9fd65fd
2 parents 6ae72d1 + b3ff3b7
commit 9fd65fd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/poseidon/main.py b/poseidon/main.py
@@ -725,7 +725,9 @@ def process(self):
                 self.logger.debug(
                     'Faucet event: {0}'.format(self.faucet_event))
             elif found_work:
-                ml_returns = self.format_rabbit_message(item)['data']
+                msg = self.format_rabbit_message(item)
+                if 'data' in msg:
+                    ml_returns = msg['data']
                 if ml_returns:
                     self.logger.info(
                         'ML results: {0}'.format(ml_returns))