Code for How to Implement 2FA in Python Tutorial

totp.py

import pyotp

# Generate a random key. You can also set to a variable e.g key = "CodingFleet"
key = pyotp.random_base32()
# Make Time based OTPs from the key.
totp = pyotp.TOTP(key)

# Print current key.
print(totp.now())

# Enter OTP for verification
input_code = input("Enter your OTP:")
# Verify OTP
print(totp.verify(input_code))

otp_verification.py

# Program 2: Verify TOTP Code with Google Authenticator
import pyotp


def simulate_authentication(key):
    # Simulate the process of authenticating with a TOTP code.
    totp = pyotp.TOTP(key)
    print("Enter the code from your Google Authenticator app to complete authentication.")
    user_input = input("Enter Code: ")
    if totp.verify(user_input):
        print("Authentication successful!")
    else:
        print("Authentication failed. Please try again with the right key.")


# Main Code
# The key should be the same one generated and used to create the QR code in Program 1
user_key = open("2fa.txt").read()  # Reading the key from the file generated in Program 1 (otp_qrcode_and_key.py)
simulate_authentication(user_key)

hotp.py

import pyotp

# Set the key. A variable this time
key = 'Muhammad'
# Make a HMAC-based OTP
hotp = pyotp.HOTP(key)

# Print results
print(hotp.at(0))
print(hotp.at(1))
print(hotp.at(2))
print(hotp.at(3))

# Set counter
counter = 0
for otp in range(4):
    print(hotp.verify(input("Enter Code: "), counter))
    counter += 1

otp_qrcode_and_key.py

# Program 1: Generate and Save TOTP Key and QR Code
import pyotp, qrcode


def generate_otp_key():
    # Generate a random key for TOTP authentication.
    return pyotp.random_base32()


def generate_qr_code(key, account_name, issuer_name):
    # Generate a QR code for TOTP authentication.
    uri = pyotp.totp.TOTP(key).provisioning_uri(name=account_name, issuer_name=issuer_name)
    img = qrcode.make(uri)
    img.save('totp_qr.png')
    print("QR Code generated and saved as 'totp_qr.png'.")


# Main code.
# Generate user key.
user_key = generate_otp_key()
print("Your Two-Factor Authentication Key:", user_key)
# Save key to a file for reference purposes
with open('2fa.txt', 'w') as f:
    f.write(user_key)
# Generate QR Code.
generate_qr_code(user_key, 'Muhammad', 'CodingFleet.com')