yunRequest.py

__author__ = 'wudaown'
# -*- coding: utf-8 -*-

import urllib.request
import http.cookiejar
import pickle

class   opener:
    def userAgent(self):
        self.head = {
                'User-Agent':'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/43.0.2357.124 Safari/537.36',
            }
        return self.head

    def imageHeader(self,url='http://p1.yunfile.com/fs/3wusahan4gzh0ikobng1268/'):
        # url = input("Please enter the url: ")
        # url = 'http://p1.yunfile.com/fs/awus9hanfgzhfikoang4098/'
        self.url = url
        self.imageHead = {
            'Accept':'image/webp,*/*;q=0.8',
            'Accept-Encoding':'gzip, deflate, sdch',
            'Accept-Language':'zh-CN,zh;q=0.8,en;q=0.6',
            'Connection':'keep-alive',
            'Cookie':'language=zh_cn; validCodeUrl="http://page1.yunfile.com:8880/view?module=service&action=queryValidCode"; referer=http%3A%2F%2Fadf.ly%2FY9hDS; JSESSIONID=D51B8AAED9C3445F1D5ECF1D51E80297',
            'Referer':self.url
        }
        return self.imageHead

    def urlOpener(self,pageUrl):
        cj = http.cookiejar.CookieJar()
        pageOpen = urllib.request.build_opener(urllib.request.HTTPCookieProcessor(cj))
        self.header = []
        for k,v in self.head.items():
            ele = (k,v)
            self.header.append(ele)
        pageOpen.addheaders = self.header
        self.pageUrl = pageUrl
        url_return = pageOpen.open(self.pageUrl)
        url_return = url_return.read().decode('utf-8')
        return url_return

    def captchaOpener(self,captchaUrl='http://p1.yunfile.com/verifyimg/getPcv.html'):
        cj = http.cookiejar.CookieJar()
        captchaOpen = urllib.request.build_opener(urllib.request.HTTPCookieProcessor(cj))
        # self.header = header
        self.captchaUrl = captchaUrl
        self.header = []
        for k , v in self.head.items():
            ele = (k,v)
            self.header.append(ele)
        for k ,v in self.imageHead.items():
            ele = (k,v)
            self.header.append(ele)
        captchaOpen.addheaders = self.header
        captcha_return = captchaOpen.open(self.captchaUrl).read()
        with open('verify.jpg','wb') as f:
            # pickle.dump(captcha_return,f)
            f.write(captcha_return)
        return captcha_return