fix(CornerShot): Improved shot generation at scale

zeronetworks · Feb 4, 2021 · 0b6993d · 0b6993d
1 parent c206a62
commit 0b6993d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/cornershot/cornershot.py b/cornershot/cornershot.py
@@ -136,6 +136,7 @@ def _shots_manager(self):
         self.total_shots = 0
 
     def open_fire(self,blocking=True):
+        self.lock_and_load()
         num_threads = min(self.total_shots,self.workers)
 
         if self.total_shots > 0:
@@ -153,10 +154,10 @@ def read_results(self):
         return self.results
 
     def lock_and_load(self):
-        self.total_shots = self.remaining_shots()
+        self.total_shots = len(self.shot_list)
 
     def remaining_shots(self):
-        return len(self.shot_list)
+        return self.total_shots
 
     def _get_suitable_shots(self, target_port, destination_port):
         class_list = []

diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
 setup(
     name='cornershot',
     python_requires='>=3',
-    version='0.2.1',
+    version='0.2.2',
     description='Library to test network connectivity',
     long_description_content_type='text/markdown',
     long_description=long_description,