def run_usb_mapping(): try: if mapped_usb: # mylogger.info('download transfer.txt from ftp to disk') # myftp.download_file('C:\\', 'Automation\\Linux\\flags', temp_folder+'\\downloads', 'transfer.txt') # mylogger.info('try to copy transfer.txt from usb key to disk') copyfile_to_usb() else: myftp.upload_new_file(temp_folder + '\\documents', 'Automation\\Linux\\flags', 'test_result.txt', 'Failed') return # fso = Dispatch("Scripting.FileSystemObject") fso = Dispatch("Scripting.FileSystemObject") transfer = temp_folder + '\\downloads\\transfer.txt' transfer_check = fso.BuildPath(mapped_usb, "transfer_check.txt") mylogger.info('try to compare files') compare_result = compare(transfer, transfer_check) if compare_result: myftp.upload_new_file(temp_folder + '\\documents', 'Automation\\Linux\\flags', 'test_result.txt', 'Passed') else: myftp.upload_new_file(temp_folder + '\\documents', 'Automation\\Linux\\flags', 'test_result.txt', 'Failed') except FileNotFoundError: mylogger.error('run_usb_mapping', exc_info=True) except OSError: mylogger.error('run_usb_mapping', exc_info=True)
def copyfile_to_usb(): fso = Dispatch("Scripting.FileSystemObject") src = "C:\\scripts\\usb_mapping.txt" des = fso.BuildPath(get_mapped_usb_key(), "usb_mapping_check.txt") if os.path.exists(src): os.remove(src) if os.path.exists(des): os.remove(des) text = "1234567890!@#$%^&*()qwertyuiopQWERTYUIOP" os.system("echo {} > C:\\scripts\\usb_mapping.txt".format(text)) fso.CopyFile(src, des, True) log.info("copying usb_mapping.txt to USB key") return
def copyfile_to_usb(): try: src = temp_folder + "\\downloads\\transfer.txt" with open(src, 'w') as f: f.write('usbkey redirection test') fso = Dispatch("Scripting.FileSystemObject") des = fso.BuildPath(redirected_usb, "transfer_check.txt") fso.CopyFile(src, des, True) mylogger.info('delay 30s') time.sleep(30) if fso.FileExists(des): mylogger.info('find transfer_check.txt in usb key') else: mylogger.info('There has no file transfer.txt in USB key') except Exception: mylogger.error('copyfile_from_usb', exc_info=True)
def run_usb_redirected(): try: if redirected_usb: # mylogger.info('download transfer.txt from ftp to disk') # myftp.download_file('C:\\', 'Linux\\Log\\Citrix\\flags', temp_folder+'\\Downloads', 'transfer.txt') # mylogger.info('try to copy transfer.txt from disk to usb key') copyfile_to_usb() else: myftp.upload_new_file(temp_folder + '\\Documents', 'Automation\\Linux\\flags', 'test_result.txt', 'Failed') return fso = Dispatch("Scripting.FileSystemObject") transfer = temp_folder + '\\downloads\\transfer.txt' transfer_check = fso.BuildPath(redirected_usb, "transfer_check.txt") mylogger.info('try to compare files') compare_result = compare(transfer, transfer_check) if compare_result: myftp.upload_new_file(temp_folder + '\\documents', 'Automation\\Linux\\flags', 'test_result.txt', 'Passed') else: myftp.upload_new_file(temp_folder + '\\documents', 'Automation\\Linux\\flags', 'test_result.txt', 'Failed') # if fso.FileExists(transfer) and fso.FileExists(transfer_check): # compare_result = compare(transfer, transfer_check) # else: # myftp.upload_new_file('C:\\', temp_folder+'\\Documents', 'Linux/Log\\Citrix\\flags', 'case_status.txt', 'Failed') # return # if compare_result: # # myftp.upload_new_file('C:\\', temp_folder+'\\Documents', 'Linux\\Log\\Citrix\\flags', 'case_status.txt', 'Passed') # else: # # myftp.upload_new_file('C:\\', temp_folder+'\\Documents', 'Linux\\Log\\Citrix\\flags', 'case_status.txt', 'Failed') # myftp.upload_existed_file('C:\\', '\\Users\\automation', 'Linux\\Log\\Citrix\\flags', 'usbkey_test.log') except FileNotFoundError: mylogger.error('run_usb_mapping', exc_info=True) except Exception: mylogger.error('run_usb_mapping', exc_info=True)
def run_usb_mapping(): try: if get_mapped_usb_key(): copyfile_to_usb() fso = Dispatch("Scripting.FileSystemObject") usb_mapping = "C:\\scripts\\usb_mapping.txt" usb_mapping_check = fso.BuildPath(get_mapped_usb_key(), "usb_mapping_check.txt") if fso.FileExists(usb_mapping) and fso.FileExists( usb_mapping_check): compare_result = compare(usb_mapping, usb_mapping_check) log.info('comparing files') if compare_result: log.info("pass, two files are the same") with open("C:\\scripts\\rdp_test.txt") as f: data = f.read() os.system( "echo {}_PASS>C:\\scripts\\rdp_test_result.txt".format( data)) ftp.storbinary( "STOR %s" % "rdp_test_result.txt", open("C:\\scripts\\rdp_test_result.txt", "rb"), 1024) upload_log() log.info("test complete, will sign out in 10 seconds") time.sleep(10) os.system("shutdown -l") return True else: log.error("fail, two files are different") with open("C:\\scripts\\rdp_test.txt") as f: data = f.read() os.system( "echo {}_FAIL>C:\\scripts\\rdp_test_result.txt".format( data)) ftp.storbinary( "STOR %s" % "rdp_test_result.txt", open("C:\\scripts\\rdp_test_result.txt", "rb"), 1024) upload_log() log.info("test complete, will sign out in 10 seconds") time.sleep(10) os.system("shutdown -l") return False else: log.error("fail, can't start comparison") with open("C:\\scripts\\rdp_test.txt") as f: data = f.read() os.system( "echo {}_FAIL>C:\\scripts\\rdp_test_result.txt".format( data)) ftp.storbinary("STOR %s" % "rdp_test_result.txt", open("C:\\scripts\\rdp_test_result.txt", "rb"), 1024) upload_log() log.info("test complete, will sign out in 10 seconds") time.sleep(10) os.system("shutdown -l") return False else: return False except FileNotFoundError: log.error('run_usb_mapping', exc_info=True) except Exception: log.error('run_usb_mapping', exc_info=True)