remove magic number

freedomofpress · Jul 7, 2020 · 602e438 · 602e438
1 parent 46a5d87
commit 602e438
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/securedrop/create-dev-data.py b/securedrop/create-dev-data.py
@@ -4,6 +4,7 @@
 import datetime
 import os
 import argparse
+import math
 from itertools import cycle
 
 from flask import current_app
@@ -53,7 +54,7 @@ def main(staging=False):
         if NUM_SOURCES == "ALL":
             # We want all strings, 14 sources will give all the strings based
             # on current string count.
-            NUM_SOURCES = 14
+            NUM_SOURCES = math.ceil(len(strings))
         # Add test sources and submissions
         num_sources = int(NUM_SOURCES)
         for i in range(1, num_sources + 1):