导出ERP本月的考勤记录

  1 from selenium import webdriver
  2 from selenium.webdriver.common.by import By
  3 import time
  4 
  5 time_start=time.time()
  6 
  7 
  8 #default setting of firefox
  9 import os
 10 fp = webdriver.FirefoxProfile()
 11 fp.set_preference("browser.download.folderList",0)#0代表下载到浏览器默认下载路径
 12 fp.set_preference("browser.download.manager.showhenStarting",False)
 13 fp.set_preference("browser.download.dir",os.getcwd())
 14 fp.set_preference("browser.helperApps.neverAsk.saveToDisk","application/vnd.ms-excel")#下载文件类型
 15 
 16 
 17 driver = webdriver.Firefox(firefox_profile = fp)
 18 
 19 
 20 #login
 21 driver.get(‘http://xxxxxxxx/login‘)
 22 driver.find_element(By.ID, "txtUserID").send_keys("xxxx")
 23 driver.find_element(By.ID, "txtPwd").send_keys("xxxx")
 24 driver.find_element(By.ID, "submit_btn").click()
 25 
 26 
 27 #open aim tab
 28 import time
 29 driver.find_element(By.CSS_SELECTOR, ".panel:nth-child(3) .panel-title").click()
 30 
 31 
 32 from selenium.webdriver.support.wait import WebDriverWait  
 33 from selenium.webdriver.support import expected_conditions as EC 
 34 def shortSleep(aimLocator):
 35     WebDriverWait(driver, 10, 0.1).until(EC.presence_of_element_located(aimLocator))  
 36 
 37 
 38 def fakeSleep(func,*args,timeout = 3,tik = 0.1):
 39     import time
 40     time_start=time.time()
 41     while timeout>0:
 42         time.sleep(tik)
 43         try:
 44             func(*args)
 45             timeout = 0
 46         except:
 47             pass
 48     time_end = time.time()
 49     print(‘time cost‘,round(time_end-time_start,2),‘s‘)
 50 
 51 
 52 #btnC1Locator = (By.ID, "_easyui_tree_12")
 53 def locAimTreeNode():
 54     driver.find_element(By.ID, "_easyui_tree_12").click()
 55     driver.find_element(By.CSS_SELECTOR, "#\\_easyui_tree_18 > .tree-title").click()
 56 
 57 
 58 fakeSleep(locAimTreeNode)
 59 print("locAimTreeNode-end")
 60 
 61 
 62 def switchFrame():
 63     driver.switch_to.frame(1)
 64 
 65 
 66 fakeSleep(switchFrame)
 67 print("switchFrame-end")
 68 
 69 
 70 #switch to new tab
 71 def searchMounthSum():
 72     import editElement
 73     date_show_text = driver.find_element(By.CSS_SELECTOR, ".item > .text")
 74     editElement.setText(driver,date_show_text,"本月")
 75     #change search days
 76     date_inner_value = driver.find_element(By.XPATH, "//*[@id=‘MySearchPanel‘]/div/div[2]/div[1]/span[2]/a/input")
 77     import datetime
 78     startDate = datetime.datetime.now().strftime(‘%Y-%m-01‘)
 79     currentDate = datetime.datetime.now().strftime(‘%Y-%m-%d‘)
 80     searchStr = startDate+‘@‘+currentDate
 81     editElement.setAttribute(driver,date_inner_value,"value",searchStr)
 82     driver.find_element(By.CSS_SELECTOR, ".tbar > div:nth-child(1) > button:nth-child(1) > span:nth-child(2)").click()
 83 
 84 
 85 fakeSleep(searchMounthSum)
 86 print("searchMounthSum-end")
 87 
 88 
 89 def tableFinished():
 90     rowsCount = driver.find_elements(By.XPATH, "/html/body/div[1]/div/div/div[3]/div/div/div[4]/div[1]/div[2]/div[2]/table/tbody/tr")
 91     if len(rowsCount)<1:
 92         raise RuntimeError(‘not ready yet‘)
 93 fakeSleep(tableFinished)
 94 print("tableFinished-end")
 95 
 96 
 97 #export
 98 def export():
 99     import editElement
100     ffNeedMakeThisItemVis =driver.find_element(By.CSS_SELECTOR, "div.sciyon-combo:nth-child(4)")
101     editElement.setAttribute(driver,ffNeedMakeThisItemVis,"style","overflow: auto; width: 100px; left: 1210.6px; top: 622.2px; display: true;")
102     driver.find_element(By.CSS_SELECTOR, "div.sciyon-combo:nth-child(4) > div:nth-child(1)").click()
103 fakeSleep(export)
104 print("export-end")
105 
106 
107 #check meg box
108 import win32gui
109 import win32con
110 import win32api
111 def click(xPos,yPos):
112     # 将鼠标移动到坐标处
113     win32api.SetCursorPos((xPos, yPos))
114     # 左点击
115     win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, 200, 200, 0, 0)
116     win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, 200, 200, 0, 0)
117 
118 
119 def catchPrintWindow():
120     dialog = win32gui.FindWindow(‘MozillaDialogClass‘,None)  # 对话框
121     windowName = win32gui.GetWindowText(dialog)
122     if windowName == ‘‘:
123         raise RuntimeError(‘not catched till now‘)
124     print("窗口名称："+windowName)
125     #print("btn名称："+win32gui.GetWindowText(btnhld)) #发现无法取到下级控件，无句柄
126     #(643, 201)-(1201, 596), 558x395
127     win32gui.SetWindowPos(dialog, win32con.HWND_TOPMOST, 0,0,420,400, win32con.SWP_SHOWWINDOW)#偷懒，使用按键精灵录制的绝对坐标
128     time.sleep(1)
129     click(64,254)#问题点：没有考虑win10内容缩放、屏幕分辨率，缩放这个估计有难度，可能需要管理员权限
130     click(230,359)
131 fakeSleep(catchPrintWindow)
132 print("catchPrintWindow-end")
133 
134 
135 print("Script End")
136 time_end=time.time()
137 print(‘time cost‘,time_end-time_start,‘s‘)