#!/usr/bin/env python3
# _*_coding: utf-8
import os
import yaml
import requests
import time
from time import sleep

current_path = os.path.abspath(os.path.dirname(__file__))  # 获取当前目录的路径
config_dir = current_path + "/choujiang.yml"    # 获取配置文件路径 

def post_http(url, headers, data):    # Post请求方式
    response = requests.post(url=url, headers=headers, data=data)
    print(response)
    print(f"\n响应码: {response.status_code}")
    print(f"\n响应结果: {response.text}")
    return response

def get_http(url, headers,data=None):  # Get请求方式

    response = requests.get(url=url, headers=headers, data=data)
    print(response)
    print(f"\n响应码: {response.status_code}")
#    print(f"\n响应结果: {response.text}")
    return response

def read_yaml():      # 打开读取yaml配置文件内容
    with open(config_dir, ‘r‘,  encoding="utf-8" ) as f:
        content = f.read()
        contents = yaml.load(content, Loader=yaml.FullLoader)
    return contents

def get_url(api_name):   # 传入url地址
    content=read_yaml()
    new_url =‘http://‘ + content[api_name][‘host‘] + content[api_name][‘url‘]
    print(f"\n请求地址:{new_url}")
    return new_url

def get_headers(api_name):   # 获取请求头
    content = read_yaml()
    new_headers = content[api_name][‘headers‘]
    #print(f"\n请求头: {new_headers}")
    return new_headers

def get_data(api_name):  # 获取数据
    content = read_yaml()
    new_data = content[api_name][‘data‘]
    return new_data

def get_method(api_name): # 获取请求方法
    content = read_yaml()
    new_method = content[api_name][‘method‘]
    return new_method

def get_time():
    # 以毫秒为单位的时间戳
    get_now_milli_time = int(time.time() * 1000)
    print(get_now_milli_time)

def send_request(api_name):   # 发送请求
    try:
        # 获取接口请求参数
        method = get_method(api_name)
        url = get_url(api_name)
        headers = get_headers(api_name)
        # 区分Get和Post方法
        if method == "Get":
            response = get_http(url, headers, data=None)
        else:
            data = get_data(api_name)
            response = post_http(url, headers, data)
        return response
    except Exception as e:
        print(f"\n接口访问出错啦: {e}")

if __name__ == ‘__main__‘:   # 执行接口
    totalData=send_request(‘prizes‘).json()
    print("第一次抽奖")
    counts =totalData[‘data‘][‘counts‘]
    print(f"总抽奖次数:{counts}")

    num = 1
    while num <= counts:
        num += 1
        print(f"抽奖次数:{num}次")
        data = send_request(‘draws‘).json()
        if data[‘code‘] == ‘1003‘:
            print(f"抽奖次数:{num}次")
            totalData = send_request(‘prizes‘).json()
            counts = totalData[‘data‘][‘counts‘]
            print(f"总抽奖次数:{counts}")