def parse_action(self):
#保存验证码图片到本地 并识别
htool = HTool()
local_img = htool.save_ercode_img(self.driver, "//img[@id='image']")
res = htool.send_img(local_img)
if res['parse']:
return res['parse']
else:
print('解析验证码失败，重新解析')
sleep(2)
return self.parse_action()
def __init__(self, taxObj):
base_dir = os.getcwd()
sys.path.append(base_dir)
self.htool = HTool()
config = self.htool.rt_config()
self.login_url = 'https://www.nuocity.com/xnw_user_ssoservice/login?service=https%3A%2F%2Fwww.nuocity.com%2Fxnw%2Fzz%2Flogin.jspx%3FreturnUrl%3Dhttps%253A%252F%252Fwww.nuocity.com%252F%26locale%3Dzh_CN'
self.ah_sb_hd_url = config['link']['ah_sb_hd_url']
# 社保账号密码
self.si_account_ret_url = self.htool.get_link_by_env(
'si_account_ret_url')
self.taxObj = taxObj
self.driver = self.taxObj.driver
def __init__(self, taxObj):
base_dir = os.getcwd()
sys.path.append(base_dir)
self.htool = HTool()
config = self.htool.rt_config()
self.login_url = config['link']['ah_sb_login_url']
self.ah_sb_hd_url = config['link']['ah_sb_hd_url']
# 社保账号密码
self.si_account_ret_url = self.htool.get_link_by_env(
'si_account_ret_url')
self.taxObj = taxObj
self.driver = self.taxObj.driver
class Ep(ui.MainLogin):
def __init__(self, parent=None, id=wx.ID_ANY, title='异步登录程序'):
ui.MainLogin.__init__(self, parent)
self.SetIcon(wx.Icon('bitbug_favicon.ico', wx.BITMAP_TYPE_ICO))
panel = wx.Panel(self, wx.ID_ANY)
button = wx.Button(panel, wx.ID_ANY, 'Hide Frame', pos=(60, 60))
sizer = wx.BoxSizer()
sizer.Add(button, 0)
panel.SetSizer(sizer)
self.taskBarIcon = TaskBarIcon(self)
self.htool = HTool()
self.config = self.htool.rt_config()
# bind event
self.Bind(wx.EVT_BUTTON, self.OnHide, button)
self.Bind(wx.EVT_CLOSE, self.OnClose)
self.Bind(wx.EVT_ICONIZE, self.OnIconfiy) # 最小化事件绑定
def OnHide(self, event):
self.Hide()
def OnIconfiy(self, event):
# wx.MessageBox('Frame has been iconized!', 'Prompt')
event.Skip()
def OnClose(self, event):
self.Hide()
def start(self):
self.check_task()
return True
def check_task(self):
if hasattr(self, 'task_check') == False:
self.task_check = threading.Thread(target=self.check_task_win)
self.cond = threading.Condition() # 锁
self.task_check.start()
elif self.task_check.is_alive() == False:
delattr(self, 'task_check')
return self.check_task()
# 添加日志
def add_log(self, text):
log_text = ''
if text != '':
log_text = "%s %s" % (time.strftime("%H:%M",
time.localtime()), text)
self.log_list.InsertItems([log_text], 0)
# 保持运行检查未完成任务
def check_task_win(self):
# print('running_index',self.running_index)
while True:
print('yes')
time.sleep(5)
def __init__(self, parent=None, id=wx.ID_ANY, title='异步登录程序'):
ui.MainLogin.__init__(self, parent)
self.SetIcon(wx.Icon('bitbug_favicon.ico', wx.BITMAP_TYPE_ICO))
panel = wx.Panel(self, wx.ID_ANY)
button = wx.Button(panel, wx.ID_ANY, 'Hide Frame', pos=(60, 60))
sizer = wx.BoxSizer()
sizer.Add(button, 0)
panel.SetSizer(sizer)
self.taskBarIcon = TaskBarIcon(self)
self.htool = HTool()
self.config = self.htool.rt_config()
# bind event
self.Bind(wx.EVT_BUTTON, self.OnHide, button)
self.Bind(wx.EVT_CLOSE, self.OnClose)
self.Bind(wx.EVT_ICONIZE, self.OnIconfiy) # 最小化事件绑定
def __init__(self, parent=None, id=wx.ID_ANY, title='发票辅助工具'):
ui.MainMenu.__init__(self, parent)
self.SetIcon(wx.Icon('mondrian.ico', wx.BITMAP_TYPE_ICO))
panel = wx.Panel(self, wx.ID_ANY)
button = wx.Button(panel, wx.ID_ANY, 'Hide Frame', pos=(60, 60))
sizer = wx.BoxSizer()
sizer.Add(button, 0)
panel.SetSizer(sizer)
self.taskBarIcon = TaskBarIcon(self)
self.htool = HTool()
self.config = self.htool.rt_config()
self.task_arr = []
self.auto_tax_cond = 'bundle-0-0||9'
self.msg_receive = 1
self.run_id = uuid.uuid1()
if self.config['debug']['env'] == 'test':
self.set_test_menu.Check(True)
else:
self.set_pro_menu.Check(True)
if self.config['debug']['browser_show'] == '1':
self.set_bro_show_menu.Check(True)
else:
self.set_bro_hide_menu.Check(True)
self.do_type = 1
self.tax_site = TaxSite(self)
# bind event
self.Bind(wx.EVT_BUTTON, self.OnHide, button)
self.Bind(wx.EVT_CLOSE, self.OnClose)
self.Bind(wx.EVT_ICONIZE, self.OnIconfiy) # 最小化事件绑定
def get_sb_data(self):
htool = HTool()
period = time.strftime("%Y%m", time.localtime())
# next_month = htool.get_1st_of_next_month()
# next_period = next_month.strftime("%Y%m")
post_data = {
"qsrq00": period,
"jzrq00": period,
"aae140": "",
"aae078": "",
"pageIndex": "0",
"pageSize": "999"
}
sb_ret = htool.post_data(self.ah_sb_hd_url, post_data, self.driver)
yj_total = 0
if sb_ret.status_code == 200:
sb_json = json.loads(sb_ret.text)
for it in sb_json['data']:
# print(it['aae002'],next_period)
# if it['aae002'] != next_period:
# continue
yj_total += float(it['aae022']) + float(it['aae020'])
return int(yj_total)
class Ep(ui.MainMenu):
def __init__(self, parent=None, id=wx.ID_ANY, title='发票辅助工具'):
ui.MainMenu.__init__(self, parent)
self.SetIcon(wx.Icon('mondrian.ico', wx.BITMAP_TYPE_ICO))
panel = wx.Panel(self, wx.ID_ANY)
button = wx.Button(panel, wx.ID_ANY, 'Hide Frame', pos=(60, 60))
sizer = wx.BoxSizer()
sizer.Add(button, 0)
panel.SetSizer(sizer)
self.taskBarIcon = TaskBarIcon(self)
self.htool = HTool()
self.config = self.htool.rt_config()
self.task_arr = []
self.auto_tax_cond = 'bundle-0-0||9'
self.msg_receive = 1
self.run_id = uuid.uuid1()
if self.config['debug']['env'] == 'test':
self.set_test_menu.Check(True)
else:
self.set_pro_menu.Check(True)
if self.config['debug']['browser_show'] == '1':
self.set_bro_show_menu.Check(True)
else:
self.set_bro_hide_menu.Check(True)
self.do_type = 1
self.tax_site = TaxSite(self)
# bind event
self.Bind(wx.EVT_BUTTON, self.OnHide, button)
self.Bind(wx.EVT_CLOSE, self.OnClose)
self.Bind(wx.EVT_ICONIZE, self.OnIconfiy) # 最小化事件绑定
def OnHide(self, event):
self.Hide()
def OnIconfiy(self, event):
# wx.MessageBox('Frame has been iconized!', 'Prompt')
event.Skip()
def OnClose(self, event):
self.Hide()
def set_env_pro( self, event ):
self.set_config('debug','env','pro')
def set_env_test( self, event ):
self.set_config('debug','env','test')
def set_bro_show( self, event ):
self.set_config('debug','browser_show','1')
def set_bro_hide( self, event ):
self.set_config('debug','browser_show','0')
def set_tax_0_0( self, event ):
self.add_log('设置自动报税为小规模无收入')
self.auto_tax_action = False
self.auto_tax_cond = 'bundle-0-0||9'
def set_tax_0_1( self, event ):
self.add_log('设置自动报税为小规模有收入')
self.auto_tax_action = False
self.auto_tax_cond = 'bundle-0-1||9'
def set_tax_1_0( self, event ):
self.add_log('设置自动报税为一般纳税人无收入')
self.auto_tax_action = False
self.auto_tax_cond = 'bundle-1-0||9'
def set_tax_1_1( self, event ):
self.add_log('设置自动报税为一般纳税人有收入')
self.auto_tax_action = False
self.auto_tax_cond = 'bundle-1-1||9'
def set_tax_all( self, event ):
self.auto_tax_cond = 'bundle-0-0||9'
self.auto_tax_action = True
# self.auto_tax = True
def msg_receive_1( self, event ):
self.msg_receive = 1
def msg_receive_0( self, event ):
self.msg_receive = 0
def do_type2( self, event ):
self.do_type = 2
def do_type1( self, event ):
self.do_type = 1
# 写入配置
def set_config(self,section,key,value):
set_val = str(value)
self.config.set(section,key,re.sub(r'%','#53',set_val.strip()))
self.config.write(open(self.htool.config_file,'w',encoding='utf-8'))
self.htool = HTool()
self.config = self.htool.rt_config()
def add_kk_task( self, event ):
self.auto_kk = bool(1-self.auto_kk)
self.kk_btn.SetValue(self.auto_kk)
if self.auto_kk == True:
self.kk_btn.SetLabel('正在扣款')
else:
self.kk_btn.SetLabel('扣款')
def add_sb_upload_task(self,event):
self.auto_sbjk_upload = bool(1-self.auto_sbjk_upload)
self.tax_sb_btn.SetValue(self.auto_sbjk_upload)
if self.auto_sbjk_upload == True:
self.tax_sb_btn.SetLabel('正在上传')
else:
self.tax_sb_btn.SetLabel('社保上传')
def add_auto_tax_task(self,event):
self.auto_tax = bool(1-self.auto_tax)
self.report_btn.SetValue(self.auto_tax)
if self.auto_tax == True:
self.report_btn.SetLabel('正在报税')
else:
self.report_btn.SetLabel('报税')
def add_agent_task(self,event):
self.auto_dkfp = bool(1-self.auto_dkfp)
self.agent_btn.SetValue(self.auto_dkfp)
if self.auto_dkfp == True:
self.agent_btn.SetLabel('获取代开')
else:
self.agent_btn.SetLabel('代开发票')
def start(self):
self.running_index = 0
self.auto_kk = False
self.auto_tax_sb = False
self.auto_tax = False
self.auto_dkfp = False
self.auto_upload = False
self.auto_sbjk_upload = False
self.auto_tax_action = False
self.auto_tax_index = 0
# self.set_status('已准备')
print('辅助工具已打开')
self.connect_service()
self.check_task()
return True
def set_status(self,text,i=0):
self.SetStatusText(time.strftime("%H:%M:%S", time.localtime()) + ' ' + text,i)
def connect_service(self):
# self.run_status = bool(1 - self.run_status)
if hasattr(self,'mythread') == False:
self.mythread = threading.Thread(target=self.create_websocket)
self.cond = threading.Condition() # 锁
self.mythread.start()
elif self.check_usb.is_alive() == False:
delattr(self,'mythread')
return self.connect_service()
def check_task(self):
if hasattr(self,'task_check') == False:
self.task_check = threading.Thread(target=self.check_task_win)
self.cond = threading.Condition() # 锁
self.task_check.start()
elif self.task_check.is_alive() == False:
delattr(self,'task_check')
return self.check_task()
# 保持运行检查未完成任务
def check_task_win(self):
# print('running_index',self.running_index)
while True:
# 同步数据
# 有新任务
# print('running_index',self.running_index,len(self.task_arr))
if len(self.task_arr) > 0:
# print(self.running_index,self.task_arr,len(self.task_arr))
print('执行任务',self.task_arr[0])
# self.syn_user_data(self.task_arr[0])
try:
login_tool = HLogin(self.run_id)
task_split = self.task_arr[0].split('||')
if task_split[-1] not in ['10','11','12']:
login_tool.login_wait()
self.syn_user_data(self.task_arr[0])
except Exception as e:
print(e)
pass
self.remove_task()
else:
if self.auto_kk == True:
self.add_log('自动获取未扣款数据')
# print('自动执行批量处理')
self.add_task_kk_bundle()
if self.auto_tax == True:
if self.auto_tax_action == False:
self.add_log('自动获取未报税数据')
self.add_task_tax_bundle(self.auto_tax_cond)
if self.auto_dkfp == True:
self.add_log('自动获取代开发票数据')
self.add_task_dkfp_bundle()
if self.auto_upload == True:
self.add_log('自动获取报税数据')
self.add_task_swsb_bundle()
if self.auto_sbjk_upload == True:
self.add_log('自动获取社保待上传')
self.add_task_sb_upload_bundle()
time.sleep(5)
def create_websocket(self):
def on_message(ws, message):
msg = json.loads(message)
if msg['type'] == 'action' and msg['data'] != '' and self.msg_receive == 1:
print('收到消息',msg)
add_task = self.add_task(msg['data'])
room_id = self.htool.get_cfg_by_env('clien','room_id')
reply = '{"type":"reply","isfree":1,"room_id":"%s","data":"%s","client_name":"%s","request_client_id":"%s"}' % (room_id,add_task,self.config['clien']['name'],msg['request_client_id'])
print('回复消息',reply)
ws.send(reply)
def on_error(ws, error):
self.run_status = False
time.sleep(5)
self.create_websocket()
def on_close(ws):
self.run_status = False
self.add_log('')
self.add_log('连接已断开')
time.sleep(5)
self.create_websocket()
def on_open(ws):
self.run_status = True
self.add_log('')
self.add_log('服务器连接成功')
self.set_status('已准备')
room_id = self.htool.get_cfg_by_env('clien','room_id')
print('room_id',room_id)
ws.send('{"type":"login","room_id":"%s","client_name":"%s"}' % (room_id,self.config['clien']['name']))
websocket.enableTrace(True)
ws = websocket.WebSocketApp("ws://im.itking.cc:12366",
on_message = on_message,
on_error = on_error,
on_close = on_close)
ws.on_open = on_open
self.ws = ws
ws.run_forever()
# 回复浏览器消息
def reply_explore(self,request_client_id,text):
print('回复消息',request_client_id)
room_id = self.htool.get_cfg_by_env('clien','room_id')
self.ws.send('{"type":"reply","room_id":"%s","request_client_id":"%s","client_name":"%s","data":"%s"}' % (room_id,request_client_id,self.uid,text))
# 添加日志
def add_log(self,text):
log_text = ''
if text != '':
log_text = "%s %s" % (time.strftime("%H:%M", time.localtime()),text)
self.log_list.InsertItems([log_text],0)
def syn_user_data(self,corp_data):
# try:
if corp_data == None:
return False
self.tax_site.set_corp(corp_data)
self.tax_site.set_runid(self.run_id)
self.tax_site.open_browser()
if self.tax_site.action == '11':
sb_site = SbExport(self.tax_site)
si_login_ret = sb_site.login({},self.tax_site.corpid)
login_res = si_login_ret['ret']
print(si_login_ret)
elif self.tax_site.action == '12':
xnw_site = Xnw(self.tax_site)
xnw_site.set_runid(self.run_id)
xnw_login_ret = xnw_site.login({},self.tax_site.corpid)
login_res = xnw_login_ret['ret']
# print(xnw_login_ret)
else:
login_res = self.tax_site.login()
# 登录失败，跳过
if login_res == False:
self.htool.set_config('login','run_id','')
self.htool.set_config('login','last_run_time','')
return False
else:
if self.config['debug']['browser_show'] == '0':
self.tax_site.restart_driver()
self.tax_site.page_init()
self.tax_site.driver_auto_action()
# except Exception as e:
# print('执行任务时出错',e)
# 添加任务
def add_task(self,corp_list):
# 单独处理批量操作
if corp_list == 'bundle||5':
self.auto_upload = True
return '自动执行批量扣款'
elif corp_list == 'bundle||-5':
self.auto_upload = False
return '自动扣款任务已暂停'
if corp_list == 'bundle||7':
self.auto_kk = True
return '自动执行批量扣款'
elif corp_list == 'bundle||-7':
self.auto_kk = False
return '自动扣款任务已暂停'
# if corp_list == 'bundle||9':
# self.auto_tax = True
# return '自动执行批量报税'
if 'bundle' in corp_list and '9' in corp_list:
self.auto_tax = True
self.auto_tax_cond = corp_list
return '自动执行批量报税'
elif corp_list == 'bundle||-9':
self.auto_tax = False
return '自动报税任务已暂停'
if corp_list == 'bundle||10':
self.auto_dkfp = True
return '自动执行批量获取代开具发票'
elif corp_list == 'bundle||-10':
self.auto_dkfp = False
return '自动获取代开具发票任务已暂停'
if corp_list == 'bundle||11':
self.auto_dkfp = True
return '自动执行批量社保缴纳信息'
elif corp_list == 'bundle||-11':
self.auto_dkfp = False
return '自动获取社保缴纳任务已暂停'
# if corp_list not in self.task_arr:
# self.corpid, self.corpname, self.credit_code, self.pwd, self.sbrqq, self.sbrqz,self.action = corp_list.split('||',7)
# self.task_list.InsertItems(["%s %s" % (time.strftime("%M:%S", time.localtime()),"收到请求:"+self.corpname + " " + self.action)],0)
# self.task_arr.append(corp_list)
# print('任务信息',corp_list)
# return '任务添加成功'
# else:
# return '该任务已在任务列表里'
_, corpname, _, _, _, _,action = corp_list.split('||',7)
self.task_list.InsertItems(["%s %s" % (time.strftime("%H:%M", time.localtime()),"收到请求:"+corpname + " " + action)],len(self.task_arr))
self.task_arr.extend([corp_list])
print('任务列表',self.task_arr)
return '任务添加成功'
# 通过批处理添加扣款任务
def add_task_kk_bundle(self):
kk_list = self.tax_site.get_kk_data()
# 没有任务了，自动停止批量操作
if len(kk_list['rows']) == 0 :
if self.do_type == 1:
self.add_log('批量请求扣款数据为空，自动暂停')
self.auto_kk = False
self.kk_btn.SetValue(self.auto_kk)
self.kk_btn.SetLabel('扣款')
else:
time.sleep(15)
for re in kk_list['rows']:
post = (re['corpid'],re['corpname'],re['credit_code'],re['tax_pwd_gs'],"","","7")
post_data = '||'.join(post)
self.add_task(post_data)
ret = '剩余可扣款数:' + kk_list['total']
self.set_status(ret)
return ret
# 通过批处理添加报税任务
def add_task_tax_bundle(self,auto_tax_cond):
# print('自动获取报税任务')
self.report_btn.SetValue(self.auto_tax)
tax_list = self.tax_site.get_tax_data(auto_tax_cond)
# print('tax_list',tax_list)
# 没有任务了，自动停止批量操作
if len(tax_list['rows']) == 0:
if self.auto_tax_action == False:
if self.do_type == 1:
ret = '批量请求报税数据为空，自动暂停'
self.add_log(ret)
self.auto_tax = False
self.report_btn.SetValue(self.auto_tax)
self.report_btn.SetLabel('报税')
self.set_status(ret)
else:
time.sleep(15)
else:
time.sleep(60)
cond_arr = ['bundle-0-0||9','bundle-0-1||9','bundle-1-0||9','bundle-1-1||9']
self.auto_tax_index += 1
self.auto_tax_cond = cond_arr[self.auto_tax_index % 4]
ret = '正在获取待申报数据'
self.set_status(ret + self.auto_tax_cond)
if self.auto_tax_index == 4:
self.auto_tax_index = 0
else:
for re in tax_list['rows']:
post = (re['corpid'],re['corpname'],re['credit_code'],re['tax_pwd_gs'],"","","9")
post_data = '||'.join(post)
self.add_task(post_data)
ret = '剩余可报税数:' + tax_list['total']
self.set_status(ret)
return ret
def add_task_dkfp_bundle(self):
tax_list = self.tax_site.get_dkjfp_data()
# print('tax_list',tax_list)
# 没有任务了，自动停止批量操作
if len(tax_list['rows']) == 0:
if self.do_type == 1:
self.add_log('批量请求待上传代开具发票企业为空，自动暂停')
self.auto_dkfp = False
self.agent_btn.SetValue(self.auto_dkfp)
self.agent_btn.SetLabel('代开发票')
else:
time.sleep(15)
for re in tax_list['rows']:
if re['agent_invoice_type'] == '2':
action_type = '10'
post = (re['corpid'],re['corpname'],re['credit_code'],re['tax_pwd_gs'],"","",action_type)
else:
action_type = '12'
if re['xnw_number'] == '' and re['xnw_pwd'] == '':
re['xnw_number'] = re['credit_code']
re['xnw_pwd'] = re['tax_pwd_gs']
post = (re['corpid'],re['corpname'],re['xnw_number'],re['xnw_pwd'],"","",action_type)
post_data = '||'.join(post)
self.add_task(post_data)
ret = '剩余待上传开具发票公司:' + tax_list['total']
self.set_status(ret)
return ret
def add_task_sb_upload_bundle(self):
print('自动获取待上传社保')
kk_data = time.strftime("%Y%m", time.localtime())
tax_list = self.tax_site.get_sb_jk_data(kk_data)
# print('tax_list',tax_list)
# 没有任务了，自动停止批量操作
if len(tax_list['rows']) == 0:
if self.do_type == 1:
self.add_log('批量请求待上传社保企业为空，自动暂停')
self.auto_sbjk_upload = False
self.tax_sb_btn.SetValue(self.auto_sbjk_upload)
self.tax_sb_btn.SetLabel('社保上传')
else:
time.sleep(15)
for re in tax_list['rows']:
post = (re['corpid'],re['corpname'],re['credit_code'],re['tax_pwd_gs'],kk_data,kk_data,"11")
post_data = '||'.join(post)
self.add_task(post_data)
ret = '剩余待上传社保公司:' + tax_list['total']
self.set_status(ret)
return ret
def add_task_swsb_bundle(self):
print('自动获取未上传报税')
tax_list = self.tax_site.get_sw_sb_data()
# print('tax_list',tax_list)
# 没有任务了，自动停止批量操作
if len(tax_list['rows']) == 0:
if self.do_type == 1:
self.add_log('批量请求待上传报税企业为空，自动暂停')
self.auto_upload = False
self.bs_upload_btn.SetValue(self.auto_upload)
self.bs_upload_btn.SetLabel('报税上传')
else:
time.sleep(15)
for re in tax_list['rows']:
print(re)
post = (re['corpid'],re['corpname'],re['credit_code'],re['tax_pwd_gs'],re['sbrqq'],re['sbrqz'],"5")
post_data = '||'.join(post)
self.add_task(post_data)
# ret = '剩余待上传开具发票公司:' + tax_list['total']
# self.set_status(ret)
# return ret
# 执行成功,删除任务
def remove_task(self):
if len(self.task_arr) > 0:
self.task_arr.pop(0)
self.task_list.Delete(0)
def post_link(self,link,post_data):
headers = {'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/52.0.2743.116 Safari/537.36 Edge/15.15063'}
try:
req = requests.post(link, data=post_data, headers=headers)
resp = req.json()
return resp
except:
return {"code":500,"text":"网络异常，请求失败"}
def set_config(self,section,key,value):
set_val = str(value)
self.config.set(section,key,re.sub(r'%','#53',set_val.strip()))
self.config.write(open(self.htool.config_file,'w',encoding='utf-8'))
self.htool = HTool()
self.config = self.htool.rt_config()
class Xnw:
def __init__(self, taxObj):
base_dir = os.getcwd()
sys.path.append(base_dir)
self.htool = HTool()
config = self.htool.rt_config()
self.login_url = 'https://www.nuocity.com/xnw_user_ssoservice/login?service=https%3A%2F%2Fwww.nuocity.com%2Fxnw%2Fzz%2Flogin.jspx%3FreturnUrl%3Dhttps%253A%252F%252Fwww.nuocity.com%252F%26locale%3Dzh_CN'
self.ah_sb_hd_url = config['link']['ah_sb_hd_url']
# 社保账号密码
self.si_account_ret_url = self.htool.get_link_by_env(
'si_account_ret_url')
self.taxObj = taxObj
self.driver = self.taxObj.driver
# 设置运行id
def set_runid(self, runid):
self.run_id = runid
#登录
def login(self, post_data, corpid):
self.driver.get(self.login_url)
#跳过引导
sleep(1)
# self.driver.execute_script("jumpToFpdk()")
# sleep(1)
# self.driver.execute_script("$('.layui-layer-dialog .layui-layer-btn a:eq(0)').click()")
# 尝试登录
ret = self.login_action(8)
post_data['ret'] = ret
return post_data
# 重复尝试登录
def login_action(self, login_times):
driver = self.driver
if login_times == 0:
# self.post_data(self.tax_password_url,{'corpid':self.corpid,'pwd':'1','msg':'登录失败次数过多'})
driver.quit()
return False
login_times -= 1
if len(self.taxObj.credit_code) != 11:
self.driver.execute_script("$('.tab li:eq(1)').click()")
self.driver.execute_script("$('#username2').val('%s')" %
self.taxObj.credit_code)
self.driver.execute_script("$('#password2').val('%s')" %
self.taxObj.pwd)
else:
self.driver.execute_script("$('#username0').val('%s')" %
self.taxObj.credit_code)
self.driver.execute_script("$('#password0').val('%s')" %
self.taxObj.pwd)
verify = XnwVerify()
verify.verify_image(self.driver)
driver.execute_script("$('#qyboxnr input[name=submit]').click()")
sleep(2)
curr_link = driver.current_url
# print('当前链接',curr_link)
if curr_link == 'https://www.nuocity.com/':
return True
# 判断是否有密码错误的提示框出来
try:
pass_err = driver.find_element_by_class_name("layui-layer-dialog")
if pass_err:
# print('pass_err',pass_err)
err_msg = pass_err.find_element_by_xpath(
"./div[@class='layui-layer-content layui-layer-padding']"
).text
print('err_msg', err_msg)
# print('错误信息',err_msg)
if '用户名格式不正确' in err_msg or '手机号格式不正确' in err_msg or '用户名不能为空' in err_msg or '密码不能为空' in err_msg or '您输入的用户名或密码不正确' in err_msg or '您输入的手机号或密码不正确' in err_msg:
print('账号密码错误')
self.htool.post_data(
self.taxObj.tax_password_url, {
'corpid': self.taxObj.corpid,
'xnw_pwd': '0',
'msg': err_msg
})
driver.quit()
return False
# 如果是计算结果错误,可重试
if '图片验证码不正确' in err_msg:
print("验证码识别结果不正确，请等待")
# self.taxObj.remove_task()
driver.execute_script("layer.closeAll()")
return self.login_action(login_times)
else:
return True
except Exception as e:
print('登录解析时发生错误', e)
return False
# 获取申报数据
def get_sb_data(self):
htool = HTool()
period = time.strftime("%Y%m", time.localtime())
# next_month = htool.get_1st_of_next_month()
# next_period = next_month.strftime("%Y%m")
post_data = {
"qsrq00": period,
"jzrq00": period,
"aae140": "",
"aae078": "",
"pageIndex": "0",
"pageSize": "999"
}
sb_ret = htool.post_data(self.ah_sb_hd_url, post_data, self.driver)
yj_total = 0
if sb_ret.status_code == 200:
sb_json = json.loads(sb_ret.text)
for it in sb_json['data']:
# print(it['aae002'],next_period)
# if it['aae002'] != next_period:
# continue
yj_total += float(it['aae022']) + float(it['aae020'])
return int(yj_total)
def match_action(self):
time.sleep(2)
# 获取鼠标位置
bar_element = self.driver.find_element_by_class_name('captcha_slider_bar')
# 鼠标移至指定位置
# a = m.position() #获取当前坐标的位置
start_x = bar_element.location['x'] + 20
stary_y = bar_element.location['y'] + 90
# m.move(start_x, stary_y) #鼠标移动到(x,y)位置
pyautogui.moveTo(x=start_x, y=stary_y)
time.sleep(1)
pic_element = self.driver.find_element_by_class_name('captcha_slider_image_area')
# left = pic_element.location['x']
# top = pic_element.location['y']
# location_x = left + pic_element.size['width'] - 20
# location_y = top + 90
# for _ in range(450):
# self.driver.execute_script("$('.captcha_slider_image_slider').remove()")
# self.get_element_image(pic_element)
# pyautogui.click(location_x,location_y)
# time.sleep(5)
# time.sleep(500)
img_path = self.get_element_image(pic_element)
match_index = self.try_match_img(img_path)
# print('最终匹配',match_index)
pre_img = Image.open("./verify_match/hf_swj/%s.png" % str(match_index))
pre_pixels = pre_img.load()
img = Image.open(img_path)
pixels = img.load()
l_max = 0
r_max = 0
for i in range(img.size[0]):
for j in range(img.size[1]):
# print(pixels[i,j][0] , pre_pixels[i,j][0])
if pixels[i,j][0] + pixels[i,j][1] + pixels[i,j][2] - pre_pixels[i,j][0] - pre_pixels[i,j][1] - pre_pixels[i,j][2] >20:
pixels[i,j] = (0, 0, 0)
if i > img.size[0]/3 and r_max < i:
r_max = i
elif i < img.size[0]/3 and l_max < i:
l_max = i
else:
pixels[i,j] = (255, 255, 255)
first_point = [0,0]
height_max = 0
# 二次降噪处理
for i in range(img.size[0]):
for j in range(img.size[1]):
if i < img.size[0]/7 or i > img.size[0]*0.9:
# 第一个像素点
if pixels[i,j] == (0, 0, 0, 255):
if first_point == [0,0]:
first_point = [i,j]
if j > height_max:
height_max = j
else:
if pixels[i,j] == (0, 0, 0, 255) and self.check_pixels(pixels,i,j,(255, 255, 255, 255)) == True:
pixels[i,j] = (255, 255, 255)
# print('first_point',first_point)
pic_width = 49
# print('height_max',height_max)
# print('width',pic_width)
# 计算距离
match_max_x = 0
for i in range(img.size[0]):
start_x = i + pic_width
start_y = first_point[1] + 25
if start_x >= img.size[0]:
break
if pixels[start_x,start_y] == (0, 0, 0, 255) and pixels[start_x+3,start_y+3] != (0, 0, 0, 255) and pixels[start_x+3,start_y-3] != (0, 0, 0, 255) and start_x > match_max_x:
match_max_x = start_x
for r in range(img.size[1]):
pixels[match_max_x - pic_width + 2,r] = (255, 0, 0)
pixels[match_max_x,r] = (255, 0, 0)
# img.save(img_path)
try:
os.remove(img_path)
except(FileNotFoundError):
print("文件不存在")
# 调整距离
end_fix = round((match_max_x - 220) * 0.07)
pyautogui.mouseDown()
pyautogui.dragRel(xOffset=match_max_x - pic_width + 2 + end_fix,yOffset=0,duration=1,button='left',mouseDownUp=False)
pyautogui.mouseUp()
# print('match_max_x',match_max_x)
# print('end_fix',end_fix)
# time.sleep(80000)
try:
pass_err = self.driver.find_element_by_id("captcha")
if pass_err:
print('验证码失败，重试')
htool = HTool()
htool.set_config('login','last_run_time',round(time.time()))
return self.match_action()
except:
pass
class SbExport:
def __init__(self, taxObj):
base_dir = os.getcwd()
sys.path.append(base_dir)
self.htool = HTool()
config = self.htool.rt_config()
self.login_url = config['link']['ah_sb_login_url']
self.ah_sb_hd_url = config['link']['ah_sb_hd_url']
# 社保账号密码
self.si_account_ret_url = self.htool.get_link_by_env(
'si_account_ret_url')
self.taxObj = taxObj
self.driver = self.taxObj.driver
#登录
def login(self, post_data, corpid):
# 获取社保账号密码
get_acc_ret = self.htool.post_data(self.si_account_ret_url,
{'corpid': corpid})
sb_account = sb_pwd = ''
post_data['ret'] = False
print('获取社保账号密码结果', get_acc_ret.text)
if get_acc_ret.status_code == 200:
tax_json = json.loads(get_acc_ret.text)
if 'shebao_number' not in tax_json or 'shebao_pwd' not in tax_json or tax_json[
'shebao_number'] == '' or tax_json['shebao_pwd'] == '':
post_data['comp_status'] = 0
post_data['content'] = '社保账号或密码不正确，请及时修改'
post_ret = self.htool.post_data(
self.taxObj.tax_password_url, {
'corpid': self.taxObj.corpid,
'shebao_pwd': '0',
'msg': '报税密码错误，请及时修改'
})
print('密码错误上传结果', post_ret.text)
self.driver.quit()
return post_data
sb_account = tax_json['shebao_number']
sb_pwd = tax_json['shebao_pwd']
else:
post_data['content'] = '获取社保账号密码时发生错误'
self.htool.post_data(
self.taxObj.tax_password_url, {
'corpid': self.taxObj.corpid,
'shebao_pwd': '0',
'msg': '报税密码错误，请及时修改'
})
self.driver.quit()
return post_data
self.sb_account = sb_account
self.sb_pwd = sb_pwd
self.driver.get(self.login_url)
#跳过引导
sleep(3)
# 尝试登录6次
ret = self.login_action(3)
post_data['ret'] = ret
return post_data
# 重复尝试登录
def login_action(self, login_times):
driver = self.driver
if login_times == 0:
# self.post_data(self.tax_password_url,{'corpid':self.corpid,'pwd':'1','msg':'登录失败次数过多'})
driver.quit()
return False
login_times -= 1
# try:
# tax_code_input = driver.find_element_by_id("username")
# pwd_input = driver.find_element_by_id("password")
verify_code = driver.find_element_by_xpath(
"//form[@id='form1']//input[@name='verify']")
# tax_code_input.send_keys(self.sb_account)
driver.execute_script("$('.wsbsLoginBg #username').val('%s')" %
self.sb_account)
# pwd_input.send_keys(self.sb_pwd)
driver.execute_script("$('.wsbsLoginBg #password').val('%s')" %
self.sb_pwd)
parse_code = self.parse_action()
verify_code.send_keys(parse_code)
# print('解析结果',parse_code)
driver.execute_script('login()')
sleep(3)
curr_link = driver.current_url
# print('当前链接',curr_link)
# 判断是否有密码错误的提示框出来
try:
if 'companyInfo' not in curr_link:
err_msg = driver.find_element_by_xpath(
"//form[@id='form1']/div[@class='wsbsLoginBg']/div/label"
).text
print('错误信息', err_msg)
if '密码不正确' in err_msg or '用户名不存在' in err_msg or '没有信息' in err_msg:
print('账号密码错误')
self.htool.post_data(
self.taxObj.tax_password_url, {
'corpid': self.taxObj.corpid,
'shebao_pwd': '0',
'msg': err_msg
})
driver.quit()
return False
# 如果是计算结果错误,可重试
if '验证码有误' in err_msg:
self.driver.execute_script("window.location.reload()")
print("验证码识别结果不正确，请等待")
# self.taxObj.remove_task()
return self.login_action(login_times)
else:
return True
except:
print('登录解析时发生错误')
return False
# 解析验证码，直到获取到一个结果为止
def parse_action(self):
#保存验证码图片到本地 并识别
htool = HTool()
local_img = htool.save_ercode_img(self.driver, "//img[@id='image']")
res = htool.send_img(local_img)
if res['parse']:
return res['parse']
else:
print('解析验证码失败，重新解析')
sleep(2)
return self.parse_action()
# 获取申报数据
def get_sb_data(self):
htool = HTool()
period = time.strftime("%Y%m", time.localtime())
# next_month = htool.get_1st_of_next_month()
# next_period = next_month.strftime("%Y%m")
post_data = {
"qsrq00": period,
"jzrq00": period,
"aae140": "",
"aae078": "",
"pageIndex": "0",
"pageSize": "999"
}
sb_ret = htool.post_data(self.ah_sb_hd_url, post_data, self.driver)
yj_total = 0
if sb_ret.status_code == 200:
sb_json = json.loads(sb_ret.text)
for it in sb_json['data']:
# print(it['aae002'],next_period)
# if it['aae002'] != next_period:
# continue
yj_total += float(it['aae022']) + float(it['aae020'])
return int(yj_total)