fake_cstor_v1_v2_server/flask_v1_v2.py

import sys
reload(sys)
sys.setdefaultencoding('utf-8')
import hmac
import hashlib
from flask import Flask, request
import json
import os
import time

GAMEID="123456" ### REMEMBER TO CHANGE ###
USERID="1234" ### REMEMBER TO CHANGE ####
REFUND_SERVER="http://onlyellow2.cstor.cn:18888/launch_refund" ### REMEMBER TO CHANGE ###

app=Flask(__name__)

result = os.popen('curl --silent --fail 169.254.169.254/latest/dynamic/instance-identity/document |grep account |awk -F:\' \' \'{print $2}\' ')
res = result.read()


@app.route('/', methods=['GET'])
def first_flask():
    return 'Hello World'


@app.route('/v1', methods=['POST'])
def post_request1():
    time.sleep(5)
    key = res[1:-3]+"v1"
    data_list = []
    data = json.loads(request.get_data(as_text=True))
    msg = ""
    for k, v in data.items():
        if k.lower() == "refund_id":
            os.popen('curl -i -H "Accept:application/json" -X POST -d "{\\"user_id\\":%s, \\"game_id\\":%s, \\"refund_id\\":\\"%s\\"}" %s'%(USERID, GAMEID, v, REFUND_SERVER))
            os.popen('curl -i -H "Accept:application/json" -X POST -d "{\\"user_id\\":\\"%s\\", \\"game_id\\":\\"%s\\", \\"refund_id\\":\\"%s\\"}" %s'%(USERID, GAMEID, v, REFUND_SERVER))
            # backup to file
            os.system('echo %s >> /root/refund.log'%(v))
            msg = "Success Get Refund ID"

        elif k.lower() == "msg":
            msg = hmac.new(key, str(v), hashlib.md5).hexdigest()

    return msg


@app.route('/v2', methods=['POST'])
def post_request2():
    time.sleep(5)
    key = res[1:-3]+"v2"
    data_list = []
    data = json.loads(request.get_data(as_text=True))
    msg = ""
    for k, v in data.items():
        if k.lower() == "refund_id":
            os.popen('curl -i -H "Accept:application/json" -X POST -d "{\\"user_id\\":%s, \\"game_id\\":%s, \\"refund_id\\":\\"%s\\"}" %s'%(USERID, GAMEID, v, REFUND_SERVER))
            os.popen('curl -i -H "Accept:application/json" -X POST -d "{\\"user_id\\":\\"%s\\", \\"game_id\\":\\"%s\\", \\"refund_id\\":\\"%s\\"}" %s'%(USERID, GAMEID, v, REFUND_SERVER))
            # backup to file
            os.system('echo %s >> /root/refund.log'%(v))
            msg = "Success Get Refund ID"

        elif k.lower() == "msg":
            msg = hmac.new(key, str(v), hashlib.md5).hexdigest()

    return msg



if __name__ == '__main__':
    app.run('0.0.0.0', 5000)