import requests
import traceback
import logging

login_url = "http://earth.gmapp.env/api/account/login_pwd"


def index_first():
    try:
        r1 = requests.get(login_url)
        return r1.cookies.get_dict()
    except:

        logging.error("index_first:%s" % traceback.format_exc())

        return None


def get_cook():
    data = open("batch_user_gen")
    list = []
    dicts = {}

    for i in data:
        list.append(i.strip('\n').strip(','))

    for i in list:
        cookies = index_first()
        post_dict = {
            'account_type': 2,
            'pwd': '123456',
            'email': i
        }
        response = requests.post(
            url=login_url,
            data=post_dict,
            cookies=cookies
        )
        headers = response.headers
        cook = headers['Set-Cookie'].split(";")
        cook = cook[0].split('=')[1]
        logging.info("response.text :%s" % response.text)
        dicts[i] = cook

    return dicts


dicts = get_cook()

with open("/data/log/cybertron/app/user_session.txt", 'w') as f:
    for key, value in dicts.items():
        f.write(key)
        f.write(",")
        f.write(value)
        f.write("\n")