#! /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.txt','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_task_accesstoken(response):
    if response.status_code == 200:
        jsondata = json.loads(response.text)
        access_token =jsondata['data']['accessToken']
    try:
        # 保存token到文件
        with open(path + 'config/task_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 teardown_hook_get_task_authcode(response):
    if response.status_code ==200:
        jsondata = json.loads(response.text)
        auth_code = jsondata['data']['authCode']
    try:

        with open(path + "config/task_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_task_authcode():
    try:
        with open(path + "config/task_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_task_accesstoken():
    try:
        # 保存token到文件
        with open(path + "config/task_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.txt", '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
#————————————————————————————————————————————————————————————————————————————————————————————————————————————
#引用资源模块
def teardown_hook_get_resource_id(response):
    if response.status_code ==200:
        jsondata = json.loads(response.text)
        resource_id = str(jsondata['data']['list'][0]['id'])
    try:

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

def get_resource_id():
    try:
        with open(path + "config/resource_id.txt", 'r') as f:
            resource_id = f.read()
            print('读取成功，resource_id：{}'.format(resource_id))
    except Exception as e:
        print('读取失败', e)
    resource_id = resource_id
    return resource_id

def teardown_hook_get_resource_title(response):
    if response.status_code ==200:
        jsondata = json.loads(response.text)
        resource_title = jsondata['data']['list'][0]['title']
    try:

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

def get_resource_title():
    try:
        with open(path + "config/resource_title.txt", 'r') as f:
            resource_title = f.read()
            print('读取成功，resource_id：{}'.format(resource_title))
    except Exception as e:
        print('读取失败', e)
    resource_title = resource_title
    return resource_title

def teardown_hook_get_resource_content(response):
    if response.status_code ==200:
        jsondata = json.loads(response.text)
        resource_content = jsondata['data']['list'][0]['content']
    try:

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

def get_resource_content():
    try:
        with open(path + "config/resource_content.txt", 'r') as f:
            resource_content = f.read()
            print('读取成功，resource_content：{}'.format(resource_content))
    except Exception as e:
        print('读取失败', e)
    resource_content = resource_content
    return resource_content