login_func.py 5.8 KB
#! /usr/bin/env python
# -*- coding: utf-8 -*-
# Date: 2020/4/9 11:36
# @Author:bxh
# @file: login_func.py
import  json
import  time,datetime
path = 'E:/task_api/'
def teardown_hook_sleep_N_secs(response,n_secs):
    """ sleep n seconds after request """
    if response.status_code == 200:
        time.sleep(0.3)
    else:
        time.sleep(n_secs)

def teardown_hook_get_cuid(response):
    if response.status_code == 200:
        jsondata = json.loads(response.text)
        cuid =str(jsondata['data']['cuid'])

    try:

        with open(path + 'config/cuid.csv','w+') as f:
            f.write(cuid)
            print('写入成功,cuid:{}'.format(cuid))
            f.close()
    except Exception as e:
        print('写入失败', e)
    return cuid

def teardown_hook_get_serialNo(response):
    if response.status_code == 200:
        jsondata = json.loads(response.text)
        serialno =str(jsondata['data']['question']['serialNo'])

    try:

        with open(path + 'config/serialNo.csv','w+') as f:
            f.write(serialno)
            print('写入成功,serialno:{}'.format(serialno))
            f.close()
    except Exception as e:
        print('写入失败', e)
    return serialno

def teardown_hook_get_accesstoken(response):
    if response.status_code == 200:
        jsondata = json.loads(response.text)
        access_token =jsondata['data']['accessToken']
    try:
        # 保存token到文件
        with open(path + 'config/accessToken.csv','w+') as f:
            f.write(access_token)
            print('写入成功,access_token:{}'.format(access_token))
            f.close()
    except Exception as e:
        print('写入失败', e)
    return access_token

def teardown_hook_get_authcode(response):
    if response.status_code ==200:
        jsondata = json.loads(response.text)
        auth_code = jsondata['data']['authCode']
    try:

        with open(path + "config/authCode.csv", 'w+') as f:
            f.write(auth_code)
            print('写入成功,authCode:{}'.format(auth_code))
    except Exception as e:
        print('写入失败', e)
    return auth_code

def get_authcode():
    try:
        with open(path + "config/authCode.csv", 'r') as f:
            auth_code = f.read()
            print('读取成功,authCode:{}'.format(auth_code))
    except Exception as e:
        print('读取失败', e)
    auth_code = str(auth_code)
    return auth_code

def get_accesstoken():
    try:
        # 保存token到文件
        with open(path + "config/accessToken.csv", 'r') as f:
            accesstoken = f.read()
            print('读取成功,accesstoken:{}'.format(accesstoken))
    except Exception as e:
        print('读取失败', e)
    accesstoken = str(accesstoken)
    return accesstoken

def get_serialNo():
    try:
        with open(path + "config/serialNo.csv", 'r') as f:
            serialno = f.read()
            print('读取成功,serialno:{}'.format(serialno))
    except Exception as e:
        print('读取失败', e)
    serialno = str(serialno)
    return serialno

def get_cuid():
    try:
        with open(path + "config/cuid.csv", 'r') as f:
            cuid = f.read()
            print('读取成功,cuid:{}'.format(cuid))
    except Exception as e:
        print('读取失败', e)
    cuid = str(cuid)
    return cuid

def get_credentials():
    try:
        with open(path + "config/credentials.csv", 'r') as f:
            credentials = f.read()
            print('读取成功,credentials:{}'.format(credentials))
    except Exception as e:
        print('读取失败', e)
    credentials = str(credentials)
    return credentials

def get_muid():
    try:
        with open(path + "config/muid.csv", 'r') as f:
            muid = f.read()
            print('读取成功,muid:{}'.format(muid))
    except Exception as e:
        print('读取失败', e)
    muid = str(muid)
    return muid

def get_cid():
    try:
        with open(path + "config/cid.csv", 'r') as f:
            cid = f.read()
            print('读取成功,muid:{}'.format(cid))
    except Exception as e:
        print('读取失败', e)
    cid = str(cid)
    return cid

t =datetime.datetime.now()
def get_bidStartTime():
    #当前日期
    t1  = t.strftime('%Y-%m-%d 00:00:00')
    # #转为秒级时间戳
    # bidStartTime = time.mktime(time.strptime(t1, '%Y-%m-%d %H:%M:%S'))
    # #转为毫秒级
    # bidStartTime = str(bidStartTime*1000).split(".")[0]
    return t1

def get_bidEndTime():
    # 1天后
    bidEndTime = (t + datetime.timedelta(days=1)).strftime("%Y-%m-%d 00:00:00")

    return bidEndTime

def teardown_hook_get_credentials(response):
    if response.status_code == 200:
        jsondata = json.loads(response.text)
        credentials =str(jsondata['data']['credentials'])

    try:

        with open(path + 'config/credentials.csv','w+') as f:
            f.write(credentials)
            print('写入成功,credentials:{}'.format(credentials))
            f.close()
    except Exception as e:
        print('写入失败', e)
    return credentials

def teardown_hook_get_muid(response):
    if response.status_code == 200:
        jsondata = json.loads(response.text)
        muid =str(jsondata['data']['companys'][0]['muid'])

    try:

        with open(path + 'config/muid.csv','w+') as f:
            f.write(muid)
            print('写入成功,muid:{}'.format(muid))
            f.close()
    except Exception as e:
        print('写入失败', e)
    return muid

def teardown_hook_get_cid(response):
    if response.status_code == 200:
        jsondata = json.loads(response.text)
        cid =str(jsondata['data']['companys'][0]['cid'])

    try:

        with open(path + 'config/cid.csv','w+') as f:
            f.write(cid)
            print('写入成功,muid:{}'.format(cid))
            f.close()
    except Exception as e:
        print('写入失败', e)
    return cid