Added type annotations + some fixes to get them correct

One functional change: `CryptoOperation.read_infile()` now reads bytes
from `sys.stdin` instead of text. This is necessary to be consistent with
the rest of the code, which all deals with bytes.

1 files changed, 2 insertions, 2 deletions

diff --git a/rsa/parallel.py b/rsa/parallel.py
index ef9f07f..e81bcad 100644
--- a/rsa/parallel.py
+++ b/rsa/parallel.py
@@ -30,7 +30,7 @@ import rsa.prime
 import rsa.randnum
 
 
-def _find_prime(nbits, pipe):
+def _find_prime(nbits: int, pipe) -> None:
     while True:
         integer = rsa.randnum.read_random_odd_int(nbits)
 
@@ -40,7 +40,7 @@ def _find_prime(nbits, pipe):
             return
 
 
-def getprime(nbits, poolsize):
+def getprime(nbits: int, poolsize: int) -> int:
     """Returns a prime number that can be stored in 'nbits' bits.
 
     Works in multiple threads at the same time.