Add CSPRNG option: --securerandom

RaceRandom.py

@@ -0,0 +1,44 @@
+import random as _random
+from functools import update_wrapper
+
+__all__ = ["use_secure"]
+
+_prng_inst = _random.Random()
+_cprng_inst = _random.SystemRandom()
+
+_mode = "prng"
+
+def use_secure(secure=True):
+    # pylint: disable=global-statement
+    global _mode
+    _mode = "cprng" if secure else "prng"
+
+def _wrap(name):
+    def somefunc(*args, **kwargs):
+        return getattr(_cprng_inst if _mode == "cprng" else _prng_inst, name)(*args, **kwargs)
+    update_wrapper(somefunc, getattr(_prng_inst, name))
+
+    return somefunc
+
+# These are for intellisense purposes only, and will be overwritten below
+choice = _prng_inst.choice
+gauss = _prng_inst.gauss
+getrandbits = _prng_inst.getrandbits
+randint = _prng_inst.randint
+random = _prng_inst.random
+randrange = _prng_inst.randrange
+sample = _prng_inst.sample
+seed = _prng_inst.seed
+shuffle = _prng_inst.shuffle
+uniform = _prng_inst.uniform
+
+for func_name in dir(_random):
+    if not callable(getattr(_random, func_name)):
+        continue
+    if not callable(getattr(_prng_inst, func_name, None)):
+        continue
+    if func_name.startswith('_'):
+        continue
+
+    globals()[func_name] = _wrap(func_name)
+    __all__.append(func_name)