python 接口测试查询数据库

# -*- coding: utf-8 -*-

import MySQLdb
import random,re,requests
import Login.postlogin
from Login.postlogin import login

# def add_friend():
''' 注册200个账号加两百个好友'''
for i in range(200):
    print i
    ''' 生成手机号'''
    # 第二位数字
    second = [3, 5, 8][random.randint(0, 2)]
    # 第三位数字
    third = {
        3: random.randint(0, 9),
        # 4: [5, 7, 9][random.randint(0, 2)],
        5: [i for i in range(10) if i != 4][random.randint(0, 8)],
        # 7: [i for i in range(10) if i not in [4, 9]][random.randint(0, 7)],
        8: random.randint(0, 9),
    }[second]
    # 最后八位数字
    suffix = "".join(random.choice("0123456789") for i in range(8))
    # 拼接手机号
    phone = "1{}{}{}".format(second, third, suffix)
    print(phone)

    ''' 获取手机验证码'''
    body= {
        'phone':"",
        'app_type': "an