tasks.py 1.25 KB
import requests
import traceback
import logging
from celery import shared_task

login_url = "http://earth.igengmei.com/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


@shared_task(name="majia.tasks.get_cook")
def get_cook():
    data = open("srv/apps/physical/majia/majia_email.txt", "r")
    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
        print(response.text)
        cook = headers['Set-Cookie'].split(";")
        cook = cook[0].split('=')[1]
        logging.info("response.text :%s" % response.text)
        dicts[i] = cook

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