#! /usr/bin/env python
# -*- coding: utf-8 -*-
# Date: 2020/4/9 11:36
# @Author:bxh
# @file: login_func.py
import json
import time, datetime


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_serialNo(response):
    if response.status_code == 200:
        jsondata = json.loads(response.text)
        serialno =str(jsondata['data']['question']['serialNo'])

    try:

        with open('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_task_accesstoken(response):
    if response.status_code == 200:
        jsondata = json.loads(response.text)
        access_token =jsondata['data']['accessToken']
    try:
        # 保存token到文件
        with open('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 get_task_accesstoken():
    try:
        # 保存token到文件
        with open("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 teardown_hook_get_task_manager_accesstoken(response):
    if response.status_code == 200:
        jsondata = json.loads(response.text)
        access_token =jsondata['data']['access']['accessToken']
    try:
        # 保存token到文件
        with open('config/task_manager_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 get_task_manager_accesstoken():
    try:
        # 保存token到文件
        with open("config/task_manager_accessToken.csv", 'r') as f:
            accesstoken = f.read()
            print('读取成功，accesstoken：{}'.format(accesstoken))
    except Exception as e:
        print('读取失败', e)
    accesstoken = str(accesstoken)
    return accesstoken


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


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


def get_serialNo():
    try:
        with open("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 teardown_hook_get_resource_id(response):
    global resource_id
    resource_id = ''
    if response.status_code ==200:
        jsondata = json.loads(response.text)
        resource_id = str(jsondata['data']['list'][0]['id'])
    try:
        with open("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():
    global resource_id
    resource_id =''
    try:
        with open("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 get_resource_id_to_int():
    return int(get_resource_id())

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

        with open("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():
    global resource_title
    resource_title = ''
    try:
        with open("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):
    global resource_content
    resource_content = ''
    if response.status_code ==200:
        jsondata = json.loads(response.text)
        resource_content = jsondata['data']['list'][0]['content']
    try:

        with open("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():
    global resource_content
    resource_content = ''
    try:
        with open("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