# -*- coding: utf-8 -*-
# filename: media.py
from basic import Basic
import urllib2
import json
import poster.encode
from poster.streaminghttp import register_openers
import time

class Media(object):
    def __init__(self):
        register_openers()
    #上传图片
    def uplaod(self, accessToken, filePath, mediaType):
        openFile = open(filePath, "rb")
        param = {'media': openFile}
        postData, postHeaders = poster.encode.multipart_encode(param)

        postUrl = "https://api.weixin.qq.com/cgi-bin/media/upload?access_token=%s&type=%s" % (accessToken, mediaType)
        request = urllib2.Request(postUrl, postData, postHeaders)
        urlResp = urllib2.urlopen(request)
        print urlResp.read()
    def get(self, accessToken, mediaId):
        postUrl = "https://api.weixin.qq.com/cgi-bin/media/get?access_token=%s&media_id=%s" % (accessToken, mediaId)
        urlResp = urllib2.urlopen(postUrl)

        headers = urlResp.info().__dict__['headers']
        if ('Content-Type: application/json\r\n' in headers) or ('Content-Type: text/plain\r\n' in headers):
            jsonDict = json.loads(urlResp.read())
            print jsonDict
        else:
            buffer = urlResp.read()   #素材的二进制
            mediaFile = file("bobo.jpg", "wb")
            mediaFile.write(buffer)
            print "get successful"

#下面的代码，不要执行，而是放在main.py同意规划
'''   
if __name__ == '__main__':
    myMedia = Media()
    accessToken = Basic().get_access_token()
    if (1==2):
        filePath = "D:/bobo.jpg"   #请安实际填写,废除，该图片已经上传，敏感图片信息
        mediaType = "image"
        myMedia.uplaod(accessToken, filePath, mediaType)
    else:
        mediaId = "2ZsPnDj9XIQlGfws31MUfR5Iuz-rcn7F6LkX3NRCsw7nDpg2268e-dbGB67WWM-N"
        myMedia.get(accessToken, mediaId)
'''