* BUGFIX: Fixed CLI bug

Author: lewis-chambers

src/iotswarm/devices.py

@@ -4,7 +4,14 @@
 import logging
 from iotswarm import __version__ as package_version
 from iotswarm.queries import CosmosTable
-from iotswarm.db import BaseDatabase, CosmosDB, Oracle, LoopingCsvDB, LoopingSQLite3
+from iotswarm.db import (
+    BaseDatabase,
+    MockDB,
+    CosmosDB,
+    Oracle,
+    LoopingCsvDB,
+    LoopingSQLite3,
+)
 from iotswarm.messaging.core import MessagingBaseClass, MockMessageConnection
 from iotswarm.messaging.aws import IotCoreMQTTConnection
 from typing import List
@@ -306,8 +313,11 @@ async def run(self):
                     )
                     self.cycle += 1
 
-                    if self.swarm is not None:
-                        self.swarm.write_self(replace=True)
+                    if isinstance(
+                        self.data_source, (LoopingCsvDB, LoopingSQLite3, MockDB)
+                    ):
+                        if self.swarm is not None:
+                            self.swarm.write_self(replace=True)
             else:
                 self._instance_logger.warning(f"No data found.")
 

src/iotswarm/scripts/cli.py

@@ -303,8 +303,10 @@ async def _mqtt_resume_session():
 
         for i in range(len(swarm.devices)):
             swarm.devices[i].connection = connection
-            swarm.devices[i].max_cycles = max_cycles
-            swarm.devices[i].sleep_time = sleep_time
+            if max_cycles is not None:
+                swarm.devices[i].max_cycles = max_cycles
+            if sleep_time is not None:
+                swarm.devices[i].sleep_time = sleep_time
 
         click.echo("Loaded swarm from pickle")
 
@@ -452,8 +454,10 @@ async def _mqtt_resume_session():
 
         for i in range(len(swarm.devices)):
             swarm.devices[i].connection = connection
-            swarm.devices[i].max_cycles = max_cycles
-            swarm.devices[i].sleep_time = sleep_time
+            if max_cycles is not None:
+                swarm.devices[i].max_cycles = max_cycles
+            if sleep_time is not None:
+                swarm.devices[i].sleep_time = sleep_time
 
         click.echo(swarm.devices[0].cycle)
         click.echo("Loaded swarm from pickle")