# cf. https://github.com/PyCQA/bandit/blob/b1411bfb43795d3ffd268bef17a839dee954c2b1/examples/weak_cryptographic_key_sizes.py

import os
from Crypto.PublicKey import RSA as pycrypto_rsa
from Cryptodome.PublicKey import RSA as pycryptodomex_rsa

# ruleid:insufficient-rsa-key-size
pycrypto_rsa.generate(bits=2048)
# ruleid:insufficient-rsa-key-size
pycryptodomex_rsa.generate(bits=2048)

# ok:insufficient-rsa-key-size
pycrypto_rsa.generate(bits=3072)
# ok:insufficient-rsa-key-size
pycryptodomex_rsa.generate(bits=3072)

# ok:insufficient-rsa-key-size
pycrypto_rsa.generate(4096)
# ok:insufficient-rsa-key-size
pycryptodomex_rsa.generate(4096)

# ruleid:insufficient-rsa-key-size
pycrypto_rsa.generate(bits=1024)
# ruleid:insufficient-rsa-key-size
pycryptodomex_rsa.generate(bits=1024)

# ruleid:insufficient-rsa-key-size
pycrypto_rsa.generate(512)
# ruleid:insufficient-rsa-key-size
pycryptodomex_rsa.generate(512)

pycrypto_rsa.generate(os.getenv("KEY_SIZE"))
pycryptodomex_rsa.generate(os.getenv("KEY_SIZE"))