def read_CompanyInfo():
log.d('读取用例公司信息excel文件')
dir = os.path.dirname(__file__)
parent_path = os.path.dirname(dir)
filename = parent_path + '/testCases/csae_document.xlsx' # 根据项目所在路径，找到用例所在的相对项目的路径
wb = xlrd.open_workbook(filename=filename) # 打开文件
company_name_index = int
tax_id_index = int
current_account_year_index = int
current_account_month_index = int
user_name_index = int
user_pwd_index = int
host_type_index = int
sheet1 = wb.sheet_by_name('用例信息') # 通过索引获取表格
for index in range(sheet1.ncols):
if ('company_name' == sheet1.cell_value(0, index)):
company_name_index = index
if ('tax_id' == sheet1.cell_value(0, index)):
tax_id_index = index
if ('current_account_year' == sheet1.cell_value(0, index)):
current_account_year_index = index
if ('current_account_month' == sheet1.cell_value(0, index)):
current_account_month_index = index
if ('user_name' == sheet1.cell_value(0, index)):
user_name_index = index
if ('user_pwd' == sheet1.cell_value(0, index)):
user_pwd_index = index
if ('host_type' == sheet1.cell_value(0, index)):
host_type_index = index
company_name = sheet1.cell_value(1, company_name_index)
tax_id = sheet1.cell_value(1, tax_id_index)
current_account_year = int(sheet1.cell_value(1,
current_account_year_index))
current_account_month = int(
sheet1.cell_value(1, current_account_month_index))
# user_name = sheet1.cell_value(1, user_name_index)
ctype = sheet1.cell(1, user_name_index).ctype # 表格的数据类型
if ctype == 2 and sheet1.cell_value(1, user_name_index) % 1 == 0: # 如果是整形
user_name = int(sheet1.cell_value(1, user_name_index))
else:
user_name = sheet1.cell_value(1, user_name_index)
ctype = sheet1.cell(1, user_pwd_index).ctype # 表格的数据类型
if ctype == 2 and sheet1.cell_value(1, user_pwd_index) % 1 == 0: # 如果是整形
user_pwd = int(sheet1.cell_value(1, user_pwd_index))
else:
user_pwd = sheet1.cell_value(1, user_pwd_index)
host_type = sheet1.cell_value(1, host_type_index)
config.set_caseCompanyName(company_name)
config.set_caseTaxId(tax_id)
config.set_caseCurrentAccountYear(current_account_year)
config.set_caseCurrentAccountMonth(current_account_month)
config.set_host(host_type)
config.set_userName(user_name)
config.set_userPwd(user_pwd)
return 0
def run(taxId):
log.d('cs-info 查询企业',
config.domain_cs_info + '/cs-info/customer/findByTaxNo')
kv = {'taxNo': taxId}
response = requests.get(config.domain_cs_info +
'/cs-info/customer/findByTaxNo',
params=kv,
allow_redirects=False)
response.encoding = 'utf-8'
if response.status_code == 200:
ret = json.loads(response.text)
if ret['code'] == '200':
year = ret['data']['currHjyYear']
month = ret['data']['currHjyMonth']
log.i('无年月查询成功')
ret, company, taxNo, id, uid, accountSystem, year, month, invoiceNo = runWithDate(
taxId, year, month)
if ret != 0:
return 1
else:
return ret, company, taxNo, id, uid, accountSystem, year, month, invoiceNo
else:
log.e('查询失败 ：', ret['msg'])
else:
log.e('登录失败 http请求失败：', response.status_code)
return 1
def read_balanceSheetList():
log.d('资产负债表excel文件')
filename = str
dir_or_files = os.listdir(config.FILE_DOWNLOAD)
if len(dir_or_files) == 0:
log.d(config.FILE_DOWNLOAD, '文件夹为空')
return 1, None
for dir_file in dir_or_files:
filename = config.FILE_DOWNLOAD + "\\" + dir_file
# log.i(filename)
wb = xlrd.open_workbook(filename=filename) # 打开文件
sheet1 = wb.sheet_by_name('资产负债表') # 通过索引获取表格
lists = dict()
for index in range(sheet1.nrows):
if (index < 3):
continue
if sheet1.cell_value(index, 1) != '' and sheet1.cell_value(index,
1) != '11':
line = str(int(sheet1.cell_value(index, 1)))
qm = sheet1.cell_value(index, 2)
nc = sheet1.cell_value(index, 3)
lists['r' + line + 'qm'] = qm if qm != '0' else '0.00'
lists['r' + line + 'nc'] = nc if nc != '0' else '0.00'
if sheet1.cell_value(index, 5) != '':
line = str(int(sheet1.cell_value(index, 5)))
qm = sheet1.cell_value(index, 6)
nc = sheet1.cell_value(index, 7)
lists['r' + line + 'qm'] = qm if qm != '0' else '0.00'
lists['r' + line + 'nc'] = nc if nc != '0' else '0.00'
return 0, lists
def runWithDate(taxId, year, month):
log.d('cs-info 带年月参数查询企业')
kv = {'taxNo': taxId, 'year': int(year), 'month': int(month)}
response = requests.get(config.domain_cs_info +
'/cs-info/customer/findByTaxNo',
params=kv,
allow_redirects=False)
response.encoding = 'utf-8'
if response.status_code == 200:
ret = json.loads(response.text)
if ret['code'] == '200':
year = ret['data']['currHjyYear']
month = ret['data']['currHjyMonth']
id = ret['data']['id']
uid = ret['data']['uid']
accountSystem = ret['data']['accountSystem']
invoiceNo = ret['data']['invoiceNo']
taxNo = ret['data']['taxNo']
company = ret['data']['company']
log.i('带年月参数查询成功', company, taxNo, id, uid, accountSystem, year,
month, invoiceNo)
return 0, company, taxNo, id, uid, accountSystem, year, month, invoiceNo
else:
log.e('带年月参数查询失败', taxId, ret['msg'])
else:
log.e('登录失败 http请求失败：', response.status_code)
return 1
def run(driver):
log.d('CRM登录', config.domain + "/#/login")
try:
if toPage(driver, config.domain + "/#/login"):
log.e('CRM登录失败-进入CRM登录页超时')
return 1
driver.find_element_by_name('username').send_keys(config.userName)
driver.find_element_by_name('password').send_keys(config.userPwd)
time.sleep(ACTION_WAIT_SLEEP_LONG)
driver.find_element_by_name('btn_login').send_keys(Keys.ENTER)
times = 0
maxTimes = int(LOAD_PAGE_TIMEOUT / WHILE_WAIT_SLEEP)
while times < maxTimes:
if driver.current_url == config.domain + "/#/dashboard":
time.sleep(ACTION_WAIT_SLEEP_LONG)
log.i('CRM登录成功')
return 0
times = times + 1
time.sleep(WHILE_WAIT_SLEEP)
log.e('CRM登录失败-登录超时')
return 1
except BaseException as e:
r = requests.get(driver.current_url, allow_redirects=False)
log.exception('CRM登录异常', r.status_code)
return 1
def mouseLeftClick(_driver, x, y):
current_x=config.current_x
offset_x = x - current_x
config.current_x = x;
current_y=config.current_y
offset_y = y - current_y
config.current_y = y
log.d('鼠标左键点击', x, y, offset_x, offset_y)
ActionChains(_driver).move_by_offset(offset_x, offset_y).click().perform()
def mouseMove(_driver, x, y):
current_x=config.current_x
offset_x = x - current_x
config.current_x = x;
current_y=config.current_y
offset_y = y - current_y
config.current_y = y
log.d('鼠标移动', x, y, offset_x, offset_y)
ActionChains(_driver).move_by_offset(offset_x, offset_y).perform() # 鼠标左键点击， 200为x坐标， 100为y坐标 多次使用时偏移累加
def mouseLocationReset(_driver):
current_x=config.current_x
offset_x = 0 - current_x
config.current_x = 0;
current_y=config.current_y
offset_y = 0 - current_y
config.current_y = 0
log.d('mouseLocationReset', offset_x, offset_y)
ActionChains(_driver).move_by_offset(offset_x,
offset_y).perform() # 鼠标左键点击， 200为x坐标， 100为y坐标
def mouseLeftDoubleClick(_driver, x, y):
current_x=config.current_x
offset_x = x - current_x
config.current_x = x;
current_y=config.current_y
offset_y = y - current_y
config.current_y = y
log.d('鼠标左键双击', x, y, offset_x, offset_y)
ActionChains(_driver).move_by_offset(offset_x,
offset_y).double_click().perform() # 鼠标左键点击， 200为x坐标， 100为y坐标 多次使用时偏移累加
def select_only(self, nid, key):
dt = datetime.datetime.now().strftime("%Y-%m-%d")
select_sql = SELECT.format(
s="id", table="product") + "WHERE dt = 0 and search_key = '" + str(
key) + "' and nid = '" + str(
nid) + "' and create_time > '" + str(
dt) + " 00:00:00.000000' and create_time < '" + str(
dt) + " 23:59:59.999999'"
log.d("mysql select:\n%s" % select_sql)
num = self.cursor.execute(select_sql)
return num, self.cursor.fetchall()
def mouseRightClick(_driver, x, y):
current_x = config.current_x
offset_x = x - current_x
config.current_x = x;
current_y = config.current_y
offset_y = y - current_y
config.current_y = y
log.d('鼠标右键点击', x, y, offset_x, offset_y)
ActionChains(_driver).move_by_offset(offset_x,
offset_y).context_click().perform() # 鼠标左键点击， 200为x坐标， 100为y坐标
def runClear(driver):
log.d('清空往来单位')
try:
if toThird.run(driver, config.caseCompanyName, config.caseTaxId):
return 1
if commonSelenium.toPage(
driver,
config.domain + "/cs-third/con/contactsunit/contactsunitlist"):
return 1
while 1:
dels = driver.find_elements_by_id('deleteContact')
if len(dels) == 0:
if 'active-kh' in driver.find_element_by_id(
'company').get_attribute('class'):
driver.find_element_by_id('people').click()
continue
else:
break
dels[0].click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
times = 0
maxTimes = int(config.LOAD_PAGE_TIMEOUT / config.WHILE_WAIT_SLEEP)
while times < maxTimes:
if driver.find_element_by_class_name(
'bootbox-body').text == '确认删除?':
btns = driver.find_elements_by_class_name('btn-primary')
for btn in btns:
if btn.text == 'OK':
btn.click()
log.i(
"往来单位- 删除",
driver.find_element_by_class_name(
'kh-text').text)
break
if times == maxTimes:
log.e("往来单位-删除等待确认超时")
return 1
times = times + 1
time.sleep(config.WHILE_WAIT_SLEEP)
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
log.i("往来单位-删除完成")
return 0
except:
r = requests.get(driver.current_url, allow_redirects=False)
log.exception('往来单位-删除异常', r.status_code)
return 1
def insert(self, bean):
insert_sql = INSERT.format(
table="product",
columns=
"id,create_time,name, standard_price,shop_price,comment_num, pay_num, sale_num, month_sale_num, collect_num, nid,cid,shop_id, url,search_key, raw_html",
values=
"'%s','%s','%s', '%s','%s','%s', '%s', '%s', '%s', '%s', '%s','%s','%s', '%s', '%s', '%s'"
%
(uuid1(), datetime.datetime.now(), bean.name, bean.standard_price,
bean.price, bean.comment_num, bean.pay_num, bean.sale_num,
bean.month_sale_num, bean.collect_num, bean.nid, bean.cid,
bean.shop_id, bean.url, bean.search_key, bean.raw_html))
log.d("mysql insert:\n%s" % insert_sql)
self.cursor.execute(insert_sql)
def read_subsidiaryLedgerList():
log.d('明细账excel文件')
filename = str
dir_or_files = os.listdir(config.FILE_DOWNLOAD)
if len(dir_or_files) == 0:
log.d(config.FILE_DOWNLOAD, '文件夹为空')
return 1
for dir_file in dir_or_files:
filename = config.FILE_DOWNLOAD + "\\" + dir_file
log.i(filename)
wb = xlrd.open_workbook(filename=filename) # 打开文件
sheet1 = wb.sheet_by_name('明细账') # 通过索引获取表格
lastKmCode = ''
lists = dict()
subsidiaryLedgers = []
for index in range(sheet1.nrows):
if (index < 3):
continue
kmCode = sheet1.cell_value(index, 0).split('_')[0]
kmName = sheet1.cell_value(index, 0).split('_')[1]
date = sheet1.cell_value(index, 1)
voucherNo = sheet1.cell_value(index, 2)
summary = sheet1.cell_value(index, 3)
debitAmount = sheet1.cell_value(index, 4)
creditAmount = sheet1.cell_value(index, 5)
direction = sheet1.cell_value(index, 6)
qmYue = sheet1.cell_value(index, 7)
sl = SubsidiaryLedger(kmCode, kmName, date, voucherNo, summary,
debitAmount, creditAmount, direction, qmYue)
# subsidiaryLedgers.append(sl)
# log.d(kmCode,kmName,date, voucherNo, summary, debitAmount,creditAmount,direction,qmYue)
# log.d(kmCode,kmName,date, voucherNo, summary, sl.debitAmount,sl.creditAmount,direction,sl.qmYue)
if (lastKmCode == ''):
lastKmCode = kmCode
# log.d('第一次' )
if (lastKmCode == kmCode):
subsidiaryLedgers.append(sl)
else:
lists[lastKmCode] = subsidiaryLedgers
subsidiaryLedgers = []
subsidiaryLedgers.append(sl)
if (index == sheet1.nrows - 1):
lists[kmCode] = subsidiaryLedgers
# log.d('最后一次', len(lists))
subsidiaryLedgers = []
lastKmCode = kmCode
return 0, lists
def run(driver):
log.d('期末结转')
alert = None
try:
if toThird.run(driver, config.caseCompanyName, config.caseTaxId):
return 1
if commonSelenium.toPage(driver,
config.domain + "/cs-third/jz/qmjz/initqmjz"):
return 1
driver.find_element_by_class_name('calculationthismonth').click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
driver.find_element_by_class_name('btn-primary').click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
# try:
# alert = WebDriverWait(driver, 10, poll_frequency=0.5).until(
# lambda x: x.find_element_by_class_name('btn-primary'))
#
# alert.click()
# return 0
# except TimeoutException:
# log.w('TimeoutException')
# return 1
# except:
# log.exception('期末结转异常')
# return 1
times = 0
maxTimes = int(config.LOAD_PAGE_TIMEOUT / config.WHILE_WAIT_SLEEP)
while times < maxTimes:
btn = driver.switch_to.active_element
if btn.tag_name == 'button' and btn.text == '确定':
btn.click()
time.sleep(config.ACTION_WAIT_SLEEP_SHORT)
break
if times == maxTimes:
log.e("期末结转等待确认超时")
return 1
times = times + 1
log.i("期末结转完成")
return 0
except:
r = requests.get(driver.current_url, allow_redirects=False)
log.exception('期末结转', r.status_code)
return 1
def run():
log.d('建账')
try:
params = {
'companyName':
config.caseCompanyName,
'taxidCode':
config.caseTaxId,
'startDate':
str(config.caseCurrentAccountYear) + '-' +
str(config.caseCurrentAccountMonth),
'accountSystem':
'20191227-145623-870-66558',
'orgId':
1,
'auditerName':
"shyuan",
'inputUserName':
"******",
'taxType':
2
}
log.i('API 建账', config.domain)
response = requests.get(config.domain + '/cs-third/mod/customer/jz',
params=params,
allow_redirects=False)
response.encoding = 'utf-8'
if response.status_code == 200:
ret = json.loads(response.text)
if str(ret['code']) == '200':
log.i('API 建账成功：', response.text)
else:
log.e('API 建账失败：', response.text)
return 1
else:
log.e('API 建账失败HTTP：', response.status_code)
return 1
return 0
except:
log.exception('API 建账', r.status_code)
return 1
def run(userName, pwd):
log.d('cs-info 登录', config.domain_cs_info)
kv = {'userName': userName, 'pwd': pwd}
response = requests.get(config.domain_cs_info + '/cs-info/user/login',
params=kv,
allow_redirects=False)
response.encoding = 'utf-8'
if response.status_code == 200:
ret = json.loads(response.text)
if ret['code'] == '200':
log.i('登录成功')
return 0, ret['data']['id']
else:
log.e('登录失败 ：', ret['msg'])
else:
log.e('登录失败 http请求失败：', response.status_code)
return 1
def runAdd(driver, partnerName):
log.d('新增往来单位')
try:
if toThird.run(driver, config.caseCompanyName, config.caseTaxId):
return 1
if commonSelenium.toPage(
driver,
config.domain + "/cs-third/con/contactsunit/contactsunitlist"):
return 1
# driver.find_element_by_id('company').click()
# driver.find_element_by_id('people').click()
driver.find_element_by_id('add').click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
driver.find_element_by_id('partnerName').send_keys(partnerName)
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
driver.find_element_by_id('contactFormBtn').click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
count = 0
while (count < config.LOAD_PAGE_TIMEOUT):
names = driver.find_elements_by_class_name('kh-text')
deletes = driver.find_elements_by_id('deleteContact')
for index in range(len(names)):
if names[index].text == partnerName:
log.i('新增往来单位成功')
return 0
elementsError = driver.find_elements_by_class_name('error')
if len(elementsError) == 2 and elementsError[1].text != '':
log.e('新增往来单位失败', elementsError[1].text)
break
count = count + 1
time.sleep(config.WHILE_WAIT_SLEEP)
log.e('新增往来单位失败, 添加后未找到')
return 1
except:
r = requests.get(driver.current_url, allow_redirects=False)
log.exception('新增往来单位异常', r.status_code)
return 1
def run(driver):
log.d('信息采集')
dicts = dict
try:
if toThird.run(driver, config.caseCompanyName, config.caseTaxId):
time.sleep(config.FAIL_WAIT_SLEEP)
return 1
if commonSelenium.toPage(
driver, config.domain +
"/cs-third/fil/fileManagement/fileManagementList"):
return 1
driver.find_element_by_id('currentDate').click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
months = driver.find_elements_by_class_name('month')
for month in months:
if str(config.caseCurrentAccountMonth) + '月' in month.text:
ActionChains(driver).move_to_element(month).click().perform()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
break
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
rows = driver.find_elements_by_xpath(
"//table[@id='customerTable']/tbody/tr")
for i in range(len(rows)):
if i < 1:
continue
tdsSum = rows[i].find_elements_by_tag_name("td")
if tdsSum[1].text != '' and tdsSum[1].text != '11':
line = tdsSum[1].text
log.i('信息采集页面接口对比通过')
return 0
except:
r = requests.get(driver.current_url, allow_redirects=False)
log.exception('信息采集', r.status_code)
return 1
def run(taxNo, type, count):
'''
:param taxNo:
:param type: 文件类型 1：收票；2：出票；3：内部流转
:param count: 上传数量
:return:
'''
log.d('cs-info 文件上传')
userName = '******'
pwd = 'lxhw_013',
retLogin, userId = csInfoLogin.run(userName, pwd)
if retLogin == 0:
retFindByTaxNo, company, taxNo, id, uid, accountSystem, year, month, invoiceNo = csInfoFindByTaxNo.run(
taxNo)
log.d(company, taxNo, id, uid, accountSystem, year, month, invoiceNo)
if retFindByTaxNo == 0:
kv = {
'id': id,
'uid': uid,
'userId': userId,
'company': company,
'taxNo': taxNo,
'year': int(year),
'month': int(month),
'type': type,
'invoiceNo': invoiceNo,
'origin': '1',
'status': '1',
'accountSystem': accountSystem}
case_dir = os.path.dirname(os.path.dirname(__file__)) + '/testCases/tax.jpg'
for i in range(count):
files = {
'files': ('tax' + str(int(time.time())) +str(random.randint(100000,999999)) + '.jpg', # file是请求参数，要与接口文档中的参数名称一致
open(case_dir, 'rb'), # 已二进制的形式打开文件
'application/msword') # 上传文件的MIME文件类型，这个必须要有
} # 上传的文件
response = requests.post(config.domain_cs_info + '/cs-info/file/upload', params=kv, files=files,
allow_redirects=False)
response.encoding = 'utf-8'
if response.status_code == 200:
ret = json.loads(response.text)
if ret['code'] == '200':
log.d('上传文件成功', i)
else:
log.e('上传文件失败 ：', i, ret['msg'])
else:
log.e('上传文件失败 http请求失败：', i, response.status_code)
return 1
log.i('上传文件完成')
return 0
# else:
# log.e('上传文件失败 ：查询企业失败' )
# else:
# log.e('上传文件失败 ：登录失败')
return 1
def read_BalanceBaseCode(path):
log.d('读取用例公司信息excel文件')
dir = os.path.dirname(__file__)
# parent_path = os.path.dirname(dir)
# filename = parent_path + '/testCases/湖北达喜供应链管理有限公司_余额表_202004-202004.xls' # 根据项目所在路径，找到用例所在的相对项目的路径
wb = xlrd.open_workbook(filename=path) # 打开文件
bm_index = 0
mc_index = 1
lists = dict()
sheet1 = wb.sheet_by_name('科目表') # 通过索引获取表格
for index in range(sheet1.nrows):
if index < 5:
continue
bm = sheet1.cell_value(index, bm_index)
mc = sheet1.cell_value(index, mc_index)
if bm == '':
continue
lists[bm] = Kemuyueb(bm, mc, 0, 0, 0, 0, 0, 0)
return 0, lists
def read_Balance(path):
log.d('读取用例公司信息excel文件')
dir = os.path.dirname(__file__)
parent_path = os.path.dirname(dir)
# filename = parent_path + '/testCases/湖北达喜供应链管理有限公司_余额表_202004-202004.xls' # 根据项目所在路径，找到用例所在的相对项目的路径
wb = xlrd.open_workbook(filename=path) # 打开文件
bm_index = 0
mc_index = 1
qcD_index = 2
qcC_index = 3
bqD_index = 4
bqC_index = 5
qmD_index = 6
qmC_index = 7
lists = dict()
sheet1 = wb.sheet_by_name('余额表') # 通过索引获取表格
for index in range(sheet1.nrows):
if index < 5:
continue
bm = sheet1.cell_value(index, bm_index)
mc = sheet1.cell_value(index, mc_index)
qcD = sheet1.cell_value(index, qcD_index)
qcC = sheet1.cell_value(index, qcC_index)
bqD = sheet1.cell_value(index, bqD_index)
bqC = sheet1.cell_value(index, bqC_index)
qmD = sheet1.cell_value(index, qmD_index)
qmC = sheet1.cell_value(index, qmC_index)
if index == sheet1.nrows - 1:
lists['sum'] = Kemuyueb('sum', mc, qcD, qcC, bqD, bqC, qmD, qmC)
else:
if bm == '':
continue
lists[bm] = Kemuyueb(bm, mc, qcD, qcC, bqD, bqC, qmD, qmC)
# log.i(bm, mc, qcD, qcC, bqD, bqC, qmD, qmC)
return 0, lists
def toPage(driver, url):
"""
:param driver: aaa
:param url:ddd
:return: 0 跳转成功
"""
if (driver.current_url == url):
# driver.refresh()
return 0
time.sleep(config.ACTION_WAIT_SLEEP_SHORT)
driver.get(url)
time.sleep(config.ACTION_WAIT_SLEEP_SHORT)
times = 0
maxTimes = int(LOAD_PAGE_TIMEOUT / WHILE_WAIT_SLEEP)
while (times < maxTimes):
if (driver.current_url == url):
return 0
times = times + 1
log.d('页面加载等待', maxTimes, times)
time.sleep(WHILE_WAIT_SLEEP)
log.e('页面加载超时',driver.current_url, url)
return 1
def read_balanceList():
log.d('余额表excel文件')
filename = str
dir_or_files = os.listdir(config.FILE_DOWNLOAD)
if len(dir_or_files) == 0:
log.d(config.FILE_DOWNLOAD, '文件夹为空')
return 1, None
for dir_file in dir_or_files:
filename = config.FILE_DOWNLOAD + "\\" + dir_file
log.i(filename)
wb = xlrd.open_workbook(filename=filename) # 打开文件
sheet1 = wb.sheet_by_name('余额表') # 通过索引获取表格
lists = dict()
for index in range(sheet1.nrows):
if (index < 5):
continue
accountCode = sheet1.cell_value(index, 0)
accountName = sheet1.cell_value(index, 1)
beginningBalanceDebit = round(sheet1.cell_value(index, 2))
beginningBalanceCrebit = round(sheet1.cell_value(index, 3))
currentAmountDebit = round(sheet1.cell_value(index, 4))
currentAmountCrebit = round(sheet1.cell_value(index, 5))
endingBalanceDebit = round(sheet1.cell_value(index, 6))
endingBalanceCrebit = round(sheet1.cell_value(index, 7))
if index == sheet1.nrows - 1:
lists['sum'] = Kemuyueb('sum', accountName, beginningBalanceDebit,
beginningBalanceCrebit, currentAmountDebit,
currentAmountCrebit, endingBalanceDebit,
endingBalanceCrebit)
else:
lists[accountCode] = Kemuyueb(
accountCode, accountName, beginningBalanceDebit,
beginningBalanceCrebit, currentAmountDebit,
currentAmountCrebit, endingBalanceDebit, endingBalanceCrebit)
return 0, lists
def run(driver):
try:
log.d(' 进入首页')
if toPage(driver, 'https://www.mi.com/'):
log.e(' 进入首页超时')
return 1
driver.find_element_by_xpath(
"//div[@id = 'J_siteUserInfo']/a[1]").click()
driver.find_element_by_class_name('btn-primary').click()
log.d(' 进入登录')
driver.find_element_by_id('username').send_keys('*****@*****.**')
driver.find_element_by_id('pwd').send_keys('lqg346353848')
driver.find_element_by_id('login-button').send_keys(Keys.ENTER)
log.d(' 进入进入购物车')
# driver.find_element_by_id('J_miniCartBtn').click()
time.sleep(2)
ActionChains(driver).move_to_element(
driver.find_element_by_id('J_miniCartBtn')).perform()
# if toPage(driver, 'https://static.mi.com/cart/'):
# log.e('进入购物车超时')
# return 1
driver.refresh()
cartListBody = driver.find_element_by_id('J_cartListBody').text
log.i('购物车列表:', cartListBody)
totalPrice = driver.find_element_by_id('J_cartTotalPrice').text
log.i('总价:', totalPrice)
return 1
except BaseException as e:
r = requests.get(driver.current_url, allow_redirects=False)
log.exception('查询异常', r.status_code)
return 1
def run(driver):
log.d('费用报销单')
try:
if commonSelenium.toPage(
driver,
config.domain + "/cs-third//third/expensesClaimSheer/list"):
log.e('加载费用报销单页面超时')
return 1
for index in range(1):
inputs = driver.find_elements_by_class_name('summary')
inputs[0].send_keys('123123123')
driver.switch_to.active_element.send_keys(Keys.TAB)
driver.switch_to.active_element.send_keys("123123")
driver.switch_to.active_element.send_keys(Keys.TAB)
driver.switch_to.active_element.send_keys('1')
textOld = driver.find_elements_by_xpath(
"//table[@id = 'bankStatementTable']/tbody/tr[1]/td[1]")
driver.find_element_by_id('save_btn').click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
driver.refresh()
textNew = driver.find_elements_by_xpath(
"//table[@id = 'bankStatementTable']/tbody/tr[1]/td[1]")
if textNew != textOld:
log.d('费用报销单成功')
log.d('费用报销单结束')
return 0
except:
r = requests.get(driver.current_url, allow_redirects=False)
log.exception('新增费用报销单异常', r.status_code)
return 1
def run(driver):
log.d('录入凭证')
try:
ret, documents, msg = read_documentInput()
if ret != 0:
log.e('加载凭证文件失败', msg)
return 1
if len(documents) == 0:
log.w('加载凭证文件数量为0')
return 1
spCount = 0
cpCount = 0
nblzCount = 0
for document in documents:
if document[0].type == 1:
spCount = spCount + 1
if document[0].type == 2:
cpCount = cpCount + 1
if document[0].type == 3:
nblzCount = nblzCount + 1
document = documents[0]
if commonSelenium.toPage(
driver, config.domain +
"/cs-third/cer/certificate/toCertificateInput"):
log.w('加载凭证录入页面超时')
return 1
driver.find_element_by_id('uniformCreditCode').send_keys(
config.caseTaxId)
driver.find_element_by_id('currentDate').click()
years = config.caseCurrentAccountYear - int(
driver.find_elements_by_class_name('datepicker-switch')[1].text)
if years < 0:
for i in range(-1 * years): # <<
ActionChains(driver).move_to_element(
driver.find_elements_by_class_name('prev')
[1]).click().perform()
time.sleep(ACTION_WAIT_SLEEP_SHORT)
if years > 0:
for i in range(years): # >>
ActionChains(driver).move_to_element(
driver.find_elements_by_class_name('next')
[1]).click().perform()
time.sleep(ACTION_WAIT_SLEEP_SHORT)
months = driver.find_elements_by_class_name('month')
for month in months:
if str(config.caseCurrentAccountMonth) in month.text:
ActionChains(driver).move_to_element(month).click().perform()
break
isOpenOriginCertificate = 0
isCsInfoFileUploadSp = 0
isCsInfoFileUploadCp = 0
isCsInfoFileUploadNblz = 0
for documentIndex in range(len(documents)):
document = documents[documentIndex]
ActionChains(driver).move_to_element(
driver.find_elements_by_class_name('iCheck-helper')[
document[0].type - 1]).click().perform()
driver.find_element_by_class_name("btn-success").send_keys(
Keys.ENTER)
# print('原始凭证',driver.find_element_by_class_name('allcount').text.split(' ')[1] )
if isOpenOriginCertificate == 0: # 打开原始凭证
isOpenOriginCertificate = 1
driver.find_element_by_id("originCertificate").send_keys(
Keys.ENTER)
driver.find_element_by_id(
"originCertificateConfirm").send_keys(Keys.ENTER)
handles = driver.window_handles
time.sleep(ACTION_WAIT_SLEEP_LONG)
driver.switch_to.window(handles[0])
# switch back to main screen
time.sleep(ACTION_WAIT_SLEEP_LONG)
count = int(
driver.find_element_by_class_name('allcount').text.split(' ')
[1][4:])
if isCsInfoFileUploadSp == 0 and document[
0].type == 1 and count < spCount:
isCsInfoFileUploadSp = 1
log.w('原始凭证收票数量不足', '当前数量：', count, '需要数量', spCount)
if csInfoFileUpload.run(config.caseTaxId, document[0].type,
spCount - count): # 文件上传失败
return 1
if isCsInfoFileUploadCp == 0 and document[
0].type == 2 and count < cpCount:
isCsInfoFileUploadCp = 1
log.w('原始凭证出票数量不足', '当前数量：', count, '需要数量', cpCount)
if csInfoFileUpload.run(config.caseTaxId, document[0].type,
cpCount - count): # 文件上传失败
return 1
if isCsInfoFileUploadNblz == 0 and document[
0].type == 3 and count < nblzCount:
isCsInfoFileUploadNblz = 1
log.w('原始凭证内部流转数量不足', '当前数量：', count, '需要数量', nblzCount)
if csInfoFileUpload.run(config.caseTaxId, document[0].type,
nblzCount - count): # 文件上传失败
return 1
divBtn = driver.find_element_by_id('divBtn')
ActionChains(driver).move_to_element(divBtn).perform()
time.sleep(ACTION_WAIT_SLEEP_LONG)
smallClasses = driver.find_elements_by_xpath(
"//ul[@class = 'dropspan-ul']/li")
for i in range(len(smallClasses)):
ActionChains(driver).move_to_element(smallClasses[i]).perform()
templates = driver.find_elements_by_xpath(
"//ul[@class = 'dropspan-ul']/li[" + str(i + 1) +
"]/ul/li")
isbreak = 0
for j in range(len(templates)):
if templates[j].text == document[0].TEMPLATED_NAME:
ActionChains(driver).move_to_element(
templates[j]).click().perform()
isbreak = 1
break
if isbreak == 1:
break
if i == len(smallClasses) - 1:
print('模板未找到', document[0].TEMPLATED_NAME)
return 1, '模板未找到', document[0].TEMPLATED_NAME
time.sleep(ACTION_WAIT_SLEEP_SHORT)
time.sleep(ACTION_WAIT_SLEEP_LONG)
ret_getFile, dataTemplates, msg = getTemplateSubjectById(
document[0].TEMPLATED_ID)
if ret_getFile != 0 or len(dataTemplates) == 0:
log.e('加载模板失败', msg)
return 1
# ActionChains(driver).move_to_element(
# driver.find_element_by_class_name('sz-zkm')).click().perform()
# time.sleep(ACTION_WAIT_SLEEP_LONG)
sj = 1
hasBank = 0
for detail in document:
for template in dataTemplates:
isSum = 0
if template.kmCode in detail.account_code:
if (template.subjectType == 1 or template.subjectType
== 2) and template.jdType == 0:
isSum = 1
if template.subjectType == 3 and template.jdType == 1:
isSum = 1
if (template.subjectType == 4
or template.subjectType == 5) and sj == 1:
isSum = 1
sj == 0
id = 'ts' + str(template.tsId)
if isSum == 0:
elementInput = driver.find_element_by_id(id)
# if elementInput.text!='':
clearElement(elementInput)
if template.jdType == 0:
elementInput.send_keys(
str(detail.credit_amount))
else:
elementInput.send_keys(str(
detail.debit_amount))
# else:
if getFeatureCdByCode(template.kmCode) == 2: # 银行
hasBank = 1
# sz_zkm=driver.find_element(By.CLASS_NAME,'sz-zkm')
sz_zkm = driver.find_element_by_class_name(
'sz-zkm')
print(
'location',
driver.find_element_by_class_name(
'sz-zkm').location['x'],
driver.find_element_by_class_name(
'sz-zkm').location['y'])
# mouseRightClick(driver,driver.find_element_by_class_name('sz-zkm').location['x'],
# driver.find_element_by_class_name('sz-zkm').location['y'])
# driver.execute_script(
# "arguments[0].setAttribute('style', arguments[1]);",
# driver.find_element_by_class_name('sz-zkm'),
# "border: 1px solid red;" # 边框border:2px; red红色
# )
# ActionChains(driver).move_to_element(
# driver.find_element_by_class_name('sz-zkm')).context_click().perform()
time.sleep(ACTION_WAIT_SLEEP_LONG)
driver.find_element_by_class_name('sz-zkm').click()
time.sleep(ACTION_WAIT_SLEEP_LONG)
try:
alert = WebDriverWait(
driver, 1, poll_frequency=0.2).until(
lambda x: x.find_element_by_id(
"zkmNoData"))
log.w(alert.text)
if '暂无子科目数据' in alert.text and alert.is_displayed(
):
addNewBank(driver, detail.account_name)
else:
chooseBank(driver, detail)
except TimeoutException:
log.w(alert.text)
chooseBank(driver, detail)
except:
log.exception('凭证录入-点击银行等待异常')
return 1
elif getFeatureCdByCode(
template.kmCode) == 4 or getFeatureCdByCode(
template.kmCode) == 5: # 往来
count = 0
# ActionChains(driver).move_to_element(
# driver.find_element_by_class_name('sz-wldw')).click().perform()
element = driver.find_element_by_css_selector(
'div[class*="sz-wldw"]')
driver.execute_script("arguments[0].click();",
element)
# driver.find_element_by_class_name('sz-wldw').click()
time.sleep(ACTION_WAIT_SLEEP_LONG)
if hasBank == 1:
elements = driver.find_elements_by_class_name(
'select2-selection')
times = 0
maxTimes = int(LOAD_PAGE_TIMEOUT /
WHILE_WAIT_SLEEP)
while times < maxTimes:
time.sleep(WHILE_WAIT_SLEEP)
if len(elements) == 2:
break
elements = driver.find_elements_by_class_name(
'select2-selection')
times = times + 1
if times == maxTimes:
log.e("等待弹出框超时——选择往来")
return 1
time.sleep(WHILE_WAIT_SLEEP)
for i in range(len(elements)):
aria_labelledby = elements[
i].get_attribute('aria-labelledby')
if aria_labelledby == 'select2-wldw-container':
ActionChains(driver).move_to_element(
elements[i]).click().perform()
else:
select2_selection = driver.find_element_by_class_name(
'select2-selection')
ActionChains(driver).move_to_element(
driver.find_element_by_class_name(
'select2-selection')).click().perform(
)
time.sleep(ACTION_WAIT_SLEEP_LONG)
driver.find_element_by_class_name(
'select2-search__field').send_keys(
detail.partner_name)
time.sleep(ACTION_WAIT_SLEEP_LONG)
times = 0
maxTimes = int(LOAD_PAGE_TIMEOUT /
WHILE_WAIT_SLEEP)
while times < maxTimes:
partner = driver.find_element_by_class_name(
'select2-results__option').text
if detail.partner_name == partner or '公司--' + detail.partner_name == partner:
driver.switch_to.active_element.send_keys(
Keys.DOWN)
driver.switch_to.active_element.send_keys(
Keys.ENTER)
break
if partner == 'No results found':
log.w('往来单位不存在', detail.partner_name)
ActionChains(driver).move_to_element(
driver.find_element_by_class_name(
'select2-selection')).click(
).perform()
time.sleep(config.ACTION_WAIT_SLEEP_SHORT)
driver.find_element_by_id(
'wldw_add_sub').click()
time.sleep(config.ACTION_WAIT_SLEEP_SHORT)
driver.find_element_by_id(
'partnerName').send_keys(
detail.partner_name)
time.sleep(config.ACTION_WAIT_SLEEP_SHORT)
driver.find_element_by_id(
'add_wldw_sub').click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
if detail.partner_name not in driver.find_element_by_class_name(
'select2-selection').text:
log.i('添加往来_关联科目失败',
detail.partner_name)
return 1
# waitToast(driver,'新增往来单位','保存成功')
break
times = times + 1
if times == maxTimes:
log.e(documentIndex, '录入凭证——添加往来等待超时',
detail.document_id)
return 1
time.sleep(WHILE_WAIT_SLEEP)
time.sleep(ACTION_WAIT_SLEEP_LONG)
driver.find_element_by_id('wldw_sub').click()
currentImgIdOld = driver.get_cookie('current_img_id')['value']
driver.find_element_by_id("remark").send_keys("selenium 录入")
time.sleep(ACTION_WAIT_SLEEP_LONG)
driver.find_element_by_id('save').click()
time.sleep(ACTION_WAIT_SLEEP_LONG)
times = 0
maxTimes = int(LOAD_PAGE_TIMEOUT / WHILE_WAIT_SLEEP)
while times < maxTimes:
try:
alert = WebDriverWait(driver, 1, poll_frequency=0.2).until(
lambda x: x.find_element_by_id("gritter-notice-wrapper"
))
# log.w(alert.text)
if '保存成功' in alert.text:
log.i(documentIndex, '录入凭证成功')
break
except TimeoutException:
pass
except:
log.exception('录入凭证异常')
return 1
currentImgIdNew = driver.get_cookie('current_img_id')['value']
log.d('点击保存等待', times, 'currentImgIdNew', currentImgIdNew,
'currentImgIdOld', currentImgIdOld)
if currentImgIdNew != currentImgIdOld:
# print('currentImgIdNew', currentImgIdNew, 'currentImgIdOld', currentImgIdOld)
log.i(documentIndex, '录入凭证成功', detail.document_id)
break
times = times + 1
if times == maxTimes:
log.e(documentIndex, '录入凭证保存等待超时', detail.document_id)
return 1
time.sleep(WHILE_WAIT_SLEEP)
time.sleep(ACTION_WAIT_SLEEP_LONG)
log.i('录入凭证结束')
return 0
except BaseException as e:
r = requests.get(driver.current_url, allow_redirects=False)
log.exception('录入凭证异常', r.status_code)
return 1
def run(driver):
log.d('资产负债表')
dictsApi = {}
try:
params = {
'origin': 'zidh',
'taxNo': config.caseTaxId,
'year': config.caseCurrentAccountYear,
'month': config.caseCurrentAccountMonth
}
log.i('API 资产负债表获取', config.domain_api)
response = requests.get(config.domain_api + '/api/report/balancesheet',
params=params,
allow_redirects=False)
response.encoding = 'utf-8'
if response.status_code == 200:
ret = json.loads(response.text)
if str(ret['code']) == '200':
dictsApi = ret['data']
else:
log.e('API 资产负债表获取失败：', response.text)
return 1
else:
log.e('API 资产负债表获取失败HTTP：', response.status_code)
return 1
log.i('API 资产负债表获取成功')
if toThird.run(driver, config.caseCompanyName, config.caseTaxId):
time.sleep(config.FAIL_WAIT_SLEEP)
return 1
if commonSelenium.toPage(
driver,
config.domain + "/cs-third/cer/balanceSheet/balanceSheetList"):
return 1
driver.find_element_by_id('currentDate').click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
months = driver.find_elements_by_class_name('month')
for month in months:
if str(config.caseCurrentAccountMonth) + '月' in month.text:
ActionChains(driver).move_to_element(month).click().perform()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
break
time.sleep(config.ACTION_WAIT_SLEEP_SHORT)
shutil.rmtree(config.FILE_DOWNLOAD)
os.mkdir(config.FILE_DOWNLOAD)
driver.find_element_by_id('downloadBtn').click()
times = 0
maxTimes = int(config.LOAD_PAGE_TIMEOUT / config.WHILE_WAIT_SLEEP)
while times < maxTimes:
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
if '数据导出并下载成功' in driver.find_element_by_class_name(
'bootbox-body').text:
btns = driver.find_elements_by_class_name('btn-primary')
for btn in btns:
if btn.text == "OK":
btn.click()
break
if times == maxTimes:
log.e("资产负债表-资产负债表下载超时")
return 1
times = times + 1
log.w("资产负债表-资产负债表下载等待", times)
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
retCode, dictsDownload = excelTools.read_balanceSheetList()
if retCode != 0:
time.sleep(config.FAIL_WAIT_SLEEP)
log.e('资产负债表下载失败：')
return 1
if len(dictsDownload) != len(dictsApi) - 1:
log.e('API资产负债表数量与下载数量不一致 dictsDownload:', str(len(dictsDownload)),
'listsApi:', str(len(dictsApi)))
return 1
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
rows = driver.find_elements_by_xpath(
"//table[@id='customerTable']/tbody/tr")
for i in range(len(rows)):
if i < 1:
continue
tdsSum = rows[i].find_elements_by_tag_name("td")
if tdsSum[1].text != '' and tdsSum[1].text != '11':
line = tdsSum[1].text
qm = tdsSum[2].text.replace(',', '')
nc = tdsSum[3].text.replace(',', '')
# log.e('tdsSum[1]：\t', line)
if dictsApi['r' + line + 'qm'] == dictsDownload[
'r' + line + 'qm'] == qm == '' and dictsApi[
'r' + line + 'nc'] == dictsDownload[
'r' + line + 'nc'] == nc == '':
pass
elif not (float(dictsApi['r' + line + 'qm']) == float(
dictsDownload['r' + line + 'qm']) == float(qm)
and float(dictsApi['r' + line + 'nc']) == float(
dictsDownload['r' + line + 'nc']) == float(nc)):
log.e('比较失败——行次', line, '\n页面：\t'
'期末余额', qm, '年初余额', nc, '\nAPI：\t',
dictsApi['r' + line + 'qm'], '年初余额',
dictsApi['r' + line + 'nc'], '\n文件：\t',
dictsDownload['r' + line + 'qm'], '年初余额',
dictsDownload['r' + line + 'nc'])
return 1
# else:
# log.d('行次',line,'期末余额', qm, '年初余额', nc )
if tdsSum[5].text != '':
line = tdsSum[5].text
qm = tdsSum[6].text.replace(',', '')
nc = tdsSum[7].text.replace(',', '')
# log.e('tdsSum[5]：\t', line)
if dictsApi['r' + line + 'qm'] == dictsDownload[
'r' + line + 'qm'] == qm == '' and dictsApi[
'r' + line + 'nc'] == dictsDownload[
'r' + line + 'nc'] == nc == '':
pass
elif not (float(dictsApi['r' + line + 'qm']) == float(
dictsDownload['r' + line + 'qm']) == float(qm)
and float(dictsApi['r' + line + 'nc']) == float(
dictsDownload['r' + line + 'nc']) == float(nc)):
log.e('比较失败——行次', line, '\n页面：\t'
'期末余额', qm, '年初余额', nc, '\nAPI：\t',
dictsApi['r' + line + 'qm'], '年初余额',
dictsApi['r' + line + 'nc'], '\n文件：\t',
dictsDownload['r' + line + 'qm'], '年初余额',
dictsDownload['r' + line + 'nc'])
return 1
# else:
# log.d('行次', line, '期末余额', qm, '年初余额', nc)
if dictsApi['r30qm'] != dictsApi['r53qm'] or dictsApi[
'r30nc'] != dictsApi['r53nc']:
log.e('资产负债不平衡')
return 1
log.i('资产负债表页面接口对比通过')
return 0
except:
r = requests.get(driver.current_url, allow_redirects=False)
log.exception('资产负债表', r.status_code)
return 1
def runBack(driver):
log.d('反结账')
try:
if toThird.run(driver, config.caseCompanyName, config.caseTaxId):
time.sleep(config.FAIL_WAIT_SLEEP)
return 1
if commonSelenium.toPage(
driver,
config.domain + "/cs-third/st/settle/toSettleAccounts"):
return 1
lastSettleDate = driver.find_element_by_class_name('marked_words').text
if lastSettleDate == '上次结账: 无数据':
log.i(lastSettleDate, '测试会计月:', config.caseCurrentAccountMonth)
return 0
lastSettleDate = lastSettleDate.replace('上次结账至:', '')
driver.find_element_by_id('dateTime').click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
# years = document[0].year - int(driver.find_elements_by_class_name('datepicker-switch')[1].text)
#
# if years < 0:
# for i in range(-1 * years): # <<
# ActionChains(driver).move_to_element(driver.find_elements_by_class_name('prev')[1]).click().perform()
# time.sleep(ACTION_WAIT_SLEEP_SHORT)
# if years > 0:
# for i in range(years): # >>
# ActionChains(driver).move_to_element(driver.find_elements_by_class_name('next')[1]).click().perform()
# time.sleep(ACTION_WAIT_SLEEP_SHORT)
lastSettleYear = int(lastSettleDate.split('-')[0])
lastSettleMonth = int(lastSettleDate.split('-')[1])
months = driver.find_elements_by_class_name('month')
if config.caseCurrentAccountYear * 100000 + config.caseCurrentAccountMonth <= lastSettleYear * 100000 + lastSettleMonth:
for month in months:
if str(lastSettleMonth) + '月' in month.text:
ActionChains(driver).move_to_element(
month).click().perform()
times = 0
maxTimes = int(config.LOAD_PAGE_TIMEOUT /
config.WHILE_WAIT_SLEEP)
while times < maxTimes:
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
if driver.find_element_by_id('checkBtn').text == '反结账':
driver.find_element_by_id('checkBtn').click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
alert = driver.find_element_by_class_name(
'bootbox-body')
if alert.text == '反结账成功':
log.i("反结账成功", str(lastSettleDate), '月')
btns = driver.find_elements_by_class_name(
'btn-primary')
for btn in btns:
if btn.text == 'OK':
btn.click()
if runBack(driver) == 0:
return 0
else:
log.w(driver.find_element_by_id('checkBtn').text)
times = times + 1
if times == maxTimes:
log.e('反结账月份选择后等待超时')
return 1
else:
log.w(lastSettleDate, "反结账")
return 0
return 1
except:
r = requests.get(driver.current_url, allow_redirects=False)
log.exception('反结账', r.status_code)
return 1
def run(driver):
log.d('结账')
try:
if toThird.run(driver, config.caseCompanyName, config.caseTaxId):
time.sleep(config.FAIL_WAIT_SLEEP)
return 1
if commonSelenium.toPage(
driver,
config.domain + "/cs-third/st/settle/toSettleAccounts"):
return 1
lastSettleMonth = driver.find_element_by_class_name(
'marked_words').text
if lastSettleMonth != '上次结账: 无数据' and int(
lastSettleMonth.split('-')
[1]) >= config.caseCurrentAccountMonth:
log.w(lastSettleMonth, "不需结账")
return 0
driver.find_element_by_id('checkBtn').click()
times = 0
maxTimes = int(config.LOAD_PAGE_TIMEOUT / config.WHILE_WAIT_SLEEP)
while times < maxTimes:
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
if driver.find_element_by_id('checkBtn').text == '继续结账':
driver.find_element_by_id('checkBtn').click()
time.sleep(config.ACTION_WAIT_SLEEP_LONG)
alert = driver.find_element_by_class_name('bootbox-body')
if alert.text == '结账成功':
log.i("结账成功")
btns = driver.find_elements_by_class_name('btn-primary')
for btn in btns:
if btn.text == 'OK':
btn.click()
return 0
else:
log.i("结账成功22222", alert.text)
elif driver.find_element_by_id('checkBtn').text == '重新检查':
log.e("马上检查未通过")
lis = driver.find_elements_by_xpath(
"//div[@class='ckPannel']/div[3]/div/ul/li")
for i in range(len(lis)):
li = driver.find_element_by_xpath(
"//div[@class='ckPannel']/div[3]/div/ul/li[" +
str(i + 1) + "]/i")
if li.get_attribute('class') == 'describe_i error':
log.w(
driver.find_element_by_xpath(
"//div[@class='ckPannel']/div[3]/div/ul/li[" +
str(i + 1) + "]/span[1]").text,
driver.find_element_by_xpath(
"//div[@class='ckPannel']/div[3]/div/ul/li[" +
str(i + 1) + "]/span[2]").text)
return 1
if times == maxTimes:
log.e("结账失败-马上检查超时")
return 1
times = times + 1
except:
r = requests.get(driver.current_url, allow_redirects=False)
log.exception('结账', r.status_code)
return 1
