def prisoner_add(page):
"""罪犯信息导入"""
# 生成模板
list = xlsx_read('./tmp_file/import.xlsx', '添加新犯')
xls_update('./tmp_file/新学员导入模板.xls', list, 2)
file_path = os.path.abspath('./')
page.driver.find_element_by_link_text('基础信息管理').click()
page.driver.find_element_by_link_text('戒毒人员信息管理').click()
page.driver.switch_to_frame(1)
page.fun_btn_nth5.click()
page.driver.switch_to.default_content()
page.driver.switch_to_frame(2)
page.btn_file_upload.click()
page.input_file_upload.send_keys(file_path + '/tmp_file/新学员导入模板.xls')
sleep(1)
page.base_prisoner_file_upload_submit.click()
sleep(2)
result = page.base_prisoner_file_upload_result
for item in result:
assert "有误" not in item.text
page.base_prisoner_add_submit.click()
sleep(1)
try:
page.driver.switch_to.default_content()
result = page.base_prisoner_add_submit_result.text
except:
page.driver.switch_to_frame(2)
result = page.win_error.text
return result
def prisoner_switch_level(page):
# 生成模板
js_ver = 'window.scrollTo(0,1000);'
list = xlsx_read('./tmp_file/import.xlsx', '转级')
xls_update('./tmp_file/批量转级模板.xls', list, 1)
page.driver.find_element_by_link_text('基础信息管理').click()
page.driver.find_element_by_link_text('戒毒人员转级管理').click()
page.driver.switch_to_frame(1)
page.prisoner_switch_area.click()
page.driver.switch_to.default_content()
page.driver.switch_to_frame(2)
page.prisoner_switch_area_upload.click()
page.input_file_upload.send_keys(
os.path.abspath('./') + '/tmp_file/批量转级模板.xls')
sleep(2)
page.prisoner_switch_level_upload_submit.click()
sleep(2)
res = page.prisoner_switch_area_upload_list
cursor1 = conn("116.63.68.108", "lecentMysql", "lecentMysql#1234",
"gd_9081")
bol = True
index = 0
for item in res:
if bol:
bol = False
continue
text = item.text.split(maxsplit=6)
config_list = query_one(
cursor1,
"select t.`Name` from prisoner as p,treat_level as t where p.TreatLevelID = t.ID and "
"p.`Code`='{}'".format(list[index][1]))
# 断言监区是否正确
assert str(text[5]) == str(config_list)
index += 1
close()
page.prisoner_switch_level_cont.send_keys('转级测试')
page.driver.execute_script(js_ver)
page.btn_next.click()
sleep(1)
page.btn_next.click()
sleep(1)
res = page.prisoner_switch_area_submit_result.text
return res
def money_in(page, is_check):
# 生成上账模板
list = xlsx_read('./tmp_file/import.xlsx', '上下账')
xls_update('./tmp_file/资金上账模板.xls', list, 1)
page.driver.find_element_by_link_text('资金业务管理').click()
page.driver.find_element_by_link_text('资金上账').click()
page.driver.switch_to_frame(1)
page.money_manage_in_add.click()
page.driver.switch_to.default_content()
page.driver.switch_to_frame(2)
page.money_manage_in_add_class.click()
page.money_manage_in_add_class_check.click()
page.money_manage_in_add_type.click()
page.money_manage_in_add_type_check.click()
# 上传文件
page.money_manage_in_file_upload.click()
page.input_file_upload.send_keys(
os.path.abspath('./') + '/tmp_file/资金上账模板.xls')
sleep(3)
# 点击开始上传
page.money_manage_in_file_submit.click()
page.money_manage_in_cont.send_keys('测试上账')
page.btn_next.click()
js = 'window.scrollTo(0,1000);'
page.driver.execute_script(js)
page.btn_next.click()
page.money_manage_in_close.click() # 上账完成
page.driver.switch_to.default_content()
page.driver.switch_to_frame(1)
page.win_refresh.click()
sleep(1)
money_in_check(page, is_check, 1)
def money_out(page, flag):
# 生成上账模板
list = xlsx_read('./tmp_file/import.xlsx', '上下账')
xls_update('./tmp_file/资金下账模板.xls', list, 1)
js = 'window.scrollTo(0,1000);'
page.driver.find_element_by_link_text('资金业务管理').click()
page.driver.find_element_by_link_text('资金下账').click()
page.driver.switch_to_frame(1)
page.money_manage_in_add.click()
page.driver.switch_to.default_content()
page.driver.switch_to_frame(2)
page.money_manage_out_add_class.click()
page.money_manage_out_add_class_check.click()
page.money_manage_out_add_type.click()
page.money_manage_out_add_type_check.click()
# 点击 上传文件
page.money_manage_in_file_upload.click()
page.money_manage_out_file_check.send_keys(
os.path.abspath('./') + '/tmp_file/资金下账模板.xls')
sleep(2)
# 点击开始上传
page.money_manage_out_file_submit.click()
sleep(3)
page.money_manage_out_cont.send_keys('测试下账thuu')
page.driver.execute_script(js)
page.btn_next.click()
sleep(1)
page.driver.execute_script(js)
page.btn_next.click()
sleep(1)
page.money_manage_out_close.click()
# 审核
page.driver.switch_to.default_content()
page.driver.switch_to_frame(1)
page.win_refresh.click()
sleep(1)
# 审核下账
if flag == '0':
page.money_manage_out_check.click()
page.money_manage_out_check_cont.send_keys('审核通过')
page.money_manage_out_check_submit.click()
page.win_prisoner_confirm.click()
# 提交下账
page.money_manage_out_submit.click()
sleep(1)
page.win_prisoner_confirm.click()
res = ""
page.win_refresh.click()
# res = page.money_manage_out_result.text
while True:
res = page.money_manage_out_result.text
sleep(1)
if "处理中" in res:
page.win_refresh.click()
else:
break
sleep(2)
page.driver.switch_to.default_content()
page.win_min_close.click()
sleep(1)
page.driver.find_element_by_link_text('资金业务管理').click()
return res