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)
