from django.db import models
from django.conf import settings


class UserOTP(models.Model):

    OTP_TYPES = (
                ('HOTP', 'hotp'),
                ('TOTP', 'totp'),
    )

    user = models.OneToOneField(settings.AUTH_USER_MODEL)
    otp_type = models.CharField(max_length=20, choices=OTP_TYPES)
    secret_key = models.CharField(max_length=100, blank=True)


def is_mfa_enabled(user):
    try:
        user_otp = user.userotp
        return True
    except:
        return False

from django.conf import settings
from django.db import models


class UserOTP(models.Model):

    OTP_TYPES = (
        ('HOTP', 'hotp'),
        ('TOTP', 'totp'),
    )

    user = models.OneToOneField(settings.AUTH_USER_MODEL)
    otp_type = models.CharField(max_length=20, choices=OTP_TYPES)
    secret_key = models.CharField(max_length=100, blank=True)


def is_mfa_enabled(user):
    """
    Determine if a user has MFA enabled
    """
    return hasattr(user, 'userotp')

Remove try/except from determing if mfa is enabled
Remove try/except from determing if mfa is enabled

+ from django.conf import settings
  from django.db import models
- from django.conf import settings
  
  
  class UserOTP(models.Model):
  
      OTP_TYPES = (
-                 ('HOTP', 'hotp'),
+         ('HOTP', 'hotp'),
-                 ('TOTP', 'totp'),
+         ('TOTP', 'totp'),
      )
  
      user = models.OneToOneField(settings.AUTH_USER_MODEL)
      otp_type = models.CharField(max_length=20, choices=OTP_TYPES)
      secret_key = models.CharField(max_length=100, blank=True)
  
  
  def is_mfa_enabled(user):
+     """
+     Determine if a user has MFA enabled
+     """
+     return hasattr(user, 'userotp')
-     try:
-         user_otp = user.userotp
-         return True
-     except:
-         return False
  
Remove try/except from determing if mfa is enabled
## Code Before:
from django.db import models
from django.conf import settings


class UserOTP(models.Model):

    OTP_TYPES = (
                ('HOTP', 'hotp'),
                ('TOTP', 'totp'),
    )

    user = models.OneToOneField(settings.AUTH_USER_MODEL)
    otp_type = models.CharField(max_length=20, choices=OTP_TYPES)
    secret_key = models.CharField(max_length=100, blank=True)


def is_mfa_enabled(user):
    try:
        user_otp = user.userotp
        return True
    except:
        return False

## Instruction:
Remove try/except from determing if mfa is enabled
## Code After:
from django.conf import settings
from django.db import models


class UserOTP(models.Model):

    OTP_TYPES = (
        ('HOTP', 'hotp'),
        ('TOTP', 'totp'),
    )

    user = models.OneToOneField(settings.AUTH_USER_MODEL)
    otp_type = models.CharField(max_length=20, choices=OTP_TYPES)
    secret_key = models.CharField(max_length=100, blank=True)


def is_mfa_enabled(user):
    """
    Determine if a user has MFA enabled
    """
    return hasattr(user, 'userotp')
