from flask import Flask, request
from flask_cors import CORS
import json
import pymysql

app = Flask(__name__)
CORS(app, resources=r'/*')


class Model:
    def __init__(self):
        self.db = pymysql.connect(
            host='xxx',
            port=3306,
            user='xxx',
            passwd='xxx',
            db='yiban',
            charset='utf8'
        )
        self.cursor = self.db.cursor()

    def insert(self, qq, cookie):
        self.cursor.execute("SELECT * FROM yiban WHERE qq=%d;" % qq)
        data = self.cursor.fetchall()
        print(data)
        if len(data) == 0:
            self.cursor.execute(
                "insert into yiban values (%d,'%s');" % (qq, cookie))
            self.db.commit()
            return 0
        else:
            self.cursor.execute("UPDATE yiban SET cookie='%s' WHERE qq=%d;" % (cookie, qq))
            self.db.commit()
            return 1


model = Model()


@app.route('/submit', methods=['POST'])
def submit():
    data = request.get_data().decode()
    data = json.loads(data)
    try:
        qq = int(data['qq'])
    except:
        data = {
            'code': 100,
            'msg': '请输入合法QQ号'
        }
        data = json.dumps(data)
        return data

    try:
        cookie = data['cookie']
    except:
        data = {'code': 101, 'msg': '请输入合法的cookie'}
        data = json.dumps(data)
        return data

    try:
        res = model.insert(qq, cookie)
        if res == 0:
            msg = '开始刷取易班网薪'
        else:
            msg = 'cookie 更新成功'
        data = {'code': 104, 'msg': msg}
    except Exception as e:
        print(e)
        data = {'code': 103, 'msg': '发生未知错误，请重新检查输入'}
    return data


app.run('0.0.0.0', port=7777)