# -*- coding: utf-8 -*-
#Author：哈士奇說喵
from selenium import webdriver
import os
import pytesser
import sys,time
from PIL import Image,ImageEnhance

#shift+tab多行縮進(jìn)(左)
reload(sys)
PostUrl = "http://yjsymis.hrbeu.edu.cn/gsmis/indexAction.do"

driver=webdriver.Firefox()
driver.get(PostUrl)


i=0
while 1:#sb登錄系統(tǒng)，即使輸對(duì)所有消息還是登不進(jìn)去的，需要登錄兩次及以上

  i=i+1
  try:
    elem_user = driver.find_element_by_name('id')
    elem_psw = driver.find_element_by_name('password')
    elem_code = driver.find_element_by_name('checkcode')
  except:
    break
  #-------------------對(duì)驗(yàn)證碼進(jìn)行區(qū)域截圖，好吧，這方法有點(diǎn)low------------------
  driver.get_screenshot_as_file('C:\Users\MrLevo\image1.jpg')#比較好理解
  im =Image.open('C:\Users\MrLevo\image1.jpg')
  box = (516,417,564,437) #設(shè)置要裁剪的區(qū)域
  region = im.crop(box)   #此時(shí)，region是一個(gè)新的圖像對(duì)象。
  #region.show()#顯示的話就會(huì)被占用，所以要注釋掉
  region.save("e:/image_code.jpg")

  #-------------------------------------------------------------------

  #--------------ImageGrab.grab()直接可以區(qū)域截圖，但是有bug，截圖不全-------
  '''
  bbox = (780, 0, 1020, 800)
  img = ImageGrab.grab()
  img.save("E:\image_code.jpg")
  img.show()
  '''
  #-------------------------手動(dòng)輸入驗(yàn)證碼：適用范圍更廣，但不夠方便------------------------------
  '''
  response = opener.open(CaptchaUrl)
  picture = response.read()
  with open('e:/image.jpg', 'wb') as local:
    local.write(picture)
  # 保存驗(yàn)證碼到本地

  #------------對(duì)于不能用pytesser+ocr進(jìn)行識(shí)別，手動(dòng)打開圖片手動(dòng)輸入--------
  # 打開保存的驗(yàn)證碼圖片 輸入
  #SecretCode = raw_input('please enter the code: ')
  #----------------------------------------------------------------------
  '''

  #--------------------圖片增強(qiáng)+自動(dòng)識(shí)別簡單驗(yàn)證碼-----------------------------
  #time.sleep(3)防止由于網(wǎng)速，可能圖片還沒保存好，就開始識(shí)別
  def image_file_to_string(file):
    cwd = os.getcwd()
    try :
      os.chdir("C:\Users\MrLevo\Anaconda2\Lib")
      return pytesser.image_file_to_string(file)
    finally:
      os.chdir(cwd)
  im=Image.open("E:\\image_code.jpg")
  imgry = im.convert('L')#圖像加強(qiáng)，二值化
  sharpness =ImageEnhance.Contrast(imgry)#對(duì)比度增強(qiáng)
  sharp_img = sharpness.enhance(2.0)
  sharp_img.save("E:\\image_code.jpg")
  #http://www.cnblogs.com/txw1958/archive/2012/02/21/2361330.html
  #imgry.show()#這是分布測試時(shí)候用的，整個(gè)程序使用需要注釋掉
  #imgry.save("E:\\image_code.jpg")

  code= pytesser.image_file_to_string("E:\\image_code.jpg")#code即為識(shí)別出的圖片數(shù)字str類型
  print code
  #打印code觀察是否識(shí)別正確


  #----------------------------------------------------------------------
  if i <= 2: # 根據(jù)自己登錄特性，我這里是驗(yàn)證碼失敗一次，重填所有，失敗兩次，重填驗(yàn)證碼
    elem_user.send_keys('S315080092')
    elem_psw.send_keys('xxxxxxxxxx')

  elem_code.send_keys(code)
  click_login = driver.find_element_by_xpath("http://img[@src='main_images/images/loginbutton.gif']")
  click_login.click()


#time.sleep(5)#搜索結(jié)果頁面停留片刻
#driver.save_screenshot('C:\Users\MrLevo\image.jpg')
#driver.close()
#driver.quit()

# -*- coding: utf-8 -*-
#Author：哈士奇說喵
from selenium import webdriver
import time
import sys


#shift+tab多行縮進(jìn)(左)
reload(sys)
PostUrl = "http://lib.hrbeu.edu.cn/#"
driver=webdriver.Firefox()
driver.get(PostUrl)

elem_user = driver.find_element_by_name('number')
elem_psw = driver.find_element_by_name('passwd')

#選擇我的圖書館，點(diǎn)擊后才能看到輸入賬號(hào)密碼
click_first = driver.find_element_by_xpath("http://ul[@id='imgmenu']/li[4]")
click_first.click()
elem_user.send_keys('S315080092')
elem_psw.send_keys('xxxxxxxx')

#點(diǎn)擊登錄
click_second = driver.find_element_by_name('submit')
click_second.click()

time.sleep(5)
#登陸后選擇
click_third = driver.find_element_by_xpath("http://*[@id='mainbox']/div/div/ul/li/a")
click_third.click()

time.sleep(5)#搜索結(jié)果頁面停留片刻
#driver.save_screenshot('C:\Users\MrLevo\image.jpg')

driver.close()
driver.quit()