Upload files to ''

fuck12306.py

@@ -0,0 +1,160 @@
+#!/usr/bin/python
+# #  FileName    : fuck12306.py
+# #  Author      : MaoMao Wang <andelf@gmail.com>
+# #  Created     : Mon Mar 16 22:08:41 2015 by ShuYu Wang
+# #  Copyright   : Feather (c) 2015
+# #  Description : fuck fuck 12306
+# #  Time-stamp: <2016-04-10 16:28:41 andelf>
+
+import re
+# hack CERTIFICATE_VERIFY_FAILED
+# https://github.com/mtschirs/quizduellapi/issues/2
+import ssl
+import urllib
+
+import requests
+from PIL import Image
+from PIL import ImageFilter
+
+if hasattr(ssl, '_create_unverified_context'):
+    ssl._create_default_https_context = ssl._create_unverified_context
+
+UA = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_2) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/41.0.2272.89 Safari/537.36"
+
+pic_url = "https://kyfw.12306.cn/otn/passcodeNew/getPassCodeNew?module=login&rand=sjrand&0.21191171556711197"
+
+
+def get_img():
+    resp = urllib.urlopen(pic_url)
+    raw = resp.read()
+    with open("./tmp.jpg", 'wb') as fp:
+        fp.write(raw)
+
+    return Image.open("./tmp.jpg")
+
+
+def get_sub_img(im, x, y):
+    assert 0 <= x <= 3
+    assert 0 <= y <= 2
+    WITH = HEIGHT = 68
+    left = 5 + (67 + 5) * x
+    top = 41 + (67 + 5) * y
+    right = left + 67
+    bottom = top + 67
+
+    return im.crop((left, top, right, bottom))
+
+
+def baidu_image_upload(im):
+    url = "http://image.baidu.com/pictureup/uploadshitu?fr=flash&fm=index&pos=upload"
+
+    im.save("./query_temp_img.png")
+    raw = open("./query_temp_img.png", 'rb').read()
+
+    files = {
+        'fileheight': "0",
+        'newfilesize': str(len(raw)),
+        'compresstime': "0",
+        'Filename': "image.png",
+        'filewidth': "0",
+        'filesize': str(len(raw)),
+        'filetype': 'image/png',
+        'Upload': "Submit Query",
+        'filedata': ("image.png", raw)
+    }
+
+    resp = requests.post(url, files=files, headers={'User-Agent': UA})
+
+    #  resp.url
+    redirect_url = "http://image.baidu.com" + resp.text
+    return redirect_url
+
+
+def baidu_stu_lookup(im):
+    redirect_url = baidu_image_upload(im)
+
+    # print redirect_url
+    resp = requests.get(redirect_url)
+
+    html = resp.text
+
+    return baidu_stu_html_extract(html)
+
+
+def baidu_stu_html_extract(html):
+    pattern = re.compile(r"'multitags':\s*'(.*?)'")
+    matches = pattern.findall(html)
+    if not matches:
+        return '[ERROR?]'
+
+    tags_str = matches[0]
+
+    result = list(filter(None, tags_str.replace('\t', ' ').split()))
+
+    return '|'.join(result) if result else '[UNKOWN]'
+
+
+def ocr_question_extract(im):
+    # git@github.com:madmaze/pytesseract.git
+    global pytesseract
+    try:
+        import pytesseract
+    except:
+        print
+        "[ERROR] pytesseract not installed"
+        return
+    im = im.crop((127, 3, 260, 22))
+    im = pre_ocr_processing(im)
+    # im.show()
+    return pytesseract.image_to_string(im, lang='chi_sim').strip()
+
+
+def pre_ocr_processing(im):
+    im = im.convert("RGB")
+    width, height = im.size
+
+    white = im.filter(ImageFilter.BLUR).filter(ImageFilter.MaxFilter(23))
+    grey = im.convert('L')
+    impix = im.load()
+    whitepix = white.load()
+    greypix = grey.load()
+
+    for y in range(height):
+        for x in range(width):
+            greypix[x, y] = min(255, max(255 + impix[x, y][0] - whitepix[x, y][0],
+                                         255 + impix[x, y][1] -
+                                         whitepix[x, y][1],
+                                         255 + impix[x, y][2] - whitepix[x, y][2]))
+
+    new_im = grey.copy()
+    binarize(new_im, 150)
+    return new_im
+
+
+def binarize(im, thresh=120):
+    assert 0 < thresh < 255
+    assert im.mode == 'L'
+    w, h = im.size
+    for y in range(0, h):
+        for x in range(0, w):
+            if im.getpixel((x, y)) < thresh:
+                im.putpixel((x, y), 0)
+            else:
+                im.putpixel((x, y), 255)
+
+
+if __name__ == '__main__':
+    im = get_img()
+    # im = Image.open("./tmp.jpg")
+    try:
+        print
+        'OCR Question:', ocr_question_extract(im)
+    except Exception as e:
+        print
+        '<OCR failed>', e
+    for y in range(2):
+        for x in range(4):
+            im2 = get_sub_img(im, x, y)
+
+            result = baidu_stu_lookup(im2)
+            print(y, x), result

test.py

@@ -0,0 +1,49 @@
+# coding: utf-8
+'''
+Created on 2016年9月9日
+@author: liuyuqi
+'''
+from time import sleep
+
+from splinter.browser import Browser
+
+# from selenium import webdriver
+# options = webdriver.ChromeOptions()
+# options.add_experimental_option("excludeSwitches",["ignore-certificate-errors"])
+# driver = webdriver.Chrome(chrome_options=options)
+
+executable_path = {'executable_path': 'C:\Program Files (x86)\Google\Chrome\Application'}
+
+username = u"979315539@qq.com"
+passwd = u""
+ticket_url = "https://kyfw.12306.cn/otn/leftTicket/init"
+login_url = "https://kyfw.12306.cn/otn/login/init"
+initmy_url = "https://kyfw.12306.cn/otn/index/initMy12306"
+
+b = Browser(driver_name="chrome")
+# b = Browser('chrome', **executable_path)
+starts = u"%u4E0A%u6D77%2CSHH"
+ends = u"%u8425%u53E3%u4E1C%2CYGT"
+
+dtime = u"2016-02-01"
+order = 0
+pa = u"000"
+
+
+# 账号登录
+def login():
+    b.find_by_text(u"登录").click()
+    sleep(3)
+    b.fill("loginUserDTO.user_name", username)
+    sleep(1)
+    b.fill("userDTO.password", passwd)
+    sleep(1)
+
+
+# 打开12306网站
+b.visit(ticket_url)
+# 点击登录按钮登录网站
+login()
+print(u"等待验证码，自行输入...")
+# 购买票
+# b.visit(ticket_url)

test2.py

@@ -0,0 +1,20 @@
+# coding: utf-8
+'''
+Created on 2016年9月9日
+@author: liuyuqi
+'''
+from splinter.browser import Browser
+
+executable_path = {
+    'executable_path': 'C:/Program Files (x86)/Google/Chrome/Application'}
+
+username = u"979315539@qq.com"
+passwd = u""
+ticket_url = "https://kyfw.12306.cn/otn/leftTicket/init"
+login_url = "https://kyfw.12306.cn/otn/login/init"
+initmy_url = "https://kyfw.12306.cn/otn/index/initMy12306"
+
+chromedriver = "C:/Program Files (x86)/Google/Chrome/Application/chromedriver.exe"
+
+b = Browser()
+b.visit("http://www.qq.com")