# coding=utf-8

import requests

import json

import MySQLdb

from datetime import datetime

from flask import Flask, redirect, request

app = Flask(__name__)

# 全局变量token

token = {}

def get_ticket():

    # App Key：13866668888   app key就是url中的 YOUR_CLIENT_ID

    # YOUR_REGISTERED_REDIRECT_URI就是 http://test.chinasoft.com

    client_id = 13866668888

    redirect_url = "http://test.chinasoft.com/a"

    url = 'https://api.weibo.com/oauth2/authorize?client_id={app_key}&response_type=code&redirect_uri={redirect_url}'.format(app_key = client_id, redirect_url = redirect_url)

    return url

def get_token(code):

    # App Secret：6dc11f2039

    redirect_url = "http://test.chinasoft.com/a"

    client_secret = "6dc11f2039"

    client_id = 13866668888

    url = "https://api.weibo.com/oauth2/access_token?client_id={client_id}&client_secret={client_secret}&grant_type=authorization_code&redirect_uri={redirect_url}&code={code}".format(client_id = client_id,client_secret = client_secret,redirect_url = redirect_url,code = code)

    resp = requests.post(url)

    global token

    token = resp.json()

    return token

def get_info(access_token, uid):

    url = "https://api.weibo.com/2/users/show.json"

    resp = requests.get(url, params = {

        'access_token':access_token,

        'uid':uid

    })

    return resp.json()

def get_conn():

    """ 获取mysql的连接"""

    try:

        conn = MySQLdb.connect(

            host="localhost",

            port=3306,

            user="root",

            password="root",

            db="user_grade",

        )

    except MySQLdb.Error as e:

        print("Error %d:%s" % (e.args[0], e.args[1]))

    return conn

def weibo_share(access_token):

    url = "https://api.weibo.com/2/statuses/share.json"

    resp = requests.post(url, {

        'access_token':access_token,

        'status':'现在是北京时间:{0} http://test.chinasoft.com'.format(datetime.now())

    })

    return resp.json()

@app.route('/a')

def index():

    code = request.args.get('code', None)

    # 根据code获取token

    token = get_token(code)

    # 获取用户信息

    user_info = get_info(token['access_token'], token['uid'])

    third_id = user_info['id']

    nickname = user_info['screen_name']

    headimg = user_info['profile_image_url']

    # 获取数据库的连接，将用户信息存入到数据库中

    # create table user(third_id int,nickname varchar(255),headimg varchar(1000));

    conn = get_conn()

    cur = conn.cursor()

    # 注意 sql中的 values中的字段一定要加 单引号，否则会报错

    sql = "insert into user(third_id,nickname,headimg) values('{third_id}','{nickname}','{headimg}')".format(third_id = third_id,nickname = nickname,headimg = headimg)

    cur.execute(sql)

    conn.autocommit(True)

    return json.dumps(user_info)

@app.route('/weibo')

def weibo():

    ticket = get_ticket()

    return redirect(ticket)

@app.route('/share')

def share():

    ticket = get_ticket()

    rest = weibo_share(token['access_token'])

    return json.dumps(rest)

if __name__ == '__main__':

    app.run(port=80, debug=True)