连接数据库+注册->登录->抽奖（有关联关系的接口）

注册账号信息需要写入数据库，登录和抽奖时从数据库获取数据

一、连接数据库

my_sql.py:

import pymysql
class MyDb:
    def __init__(self,host,password,user,db,port=3306,charset='utf8',autocommit=True):
        try:
            self.coon = pymysql.connect(host=host,password=password,
                                        user=user,db=db,port=port,
                                        charset=charset,autocommit=autocommit)
        except Exception as e:
            print('数据库连接失败！错误信息是%s'%e)
            raise Exception('数据库连接失败！错误信息是%s'%e)
            #主动抛出一个异常
        else:
            self.cur = self.coon.cursor(pymysql.cursors.DictCursor)

    def select_all(self,sql):
        try:
            self.cur.execute(sql)
        except Exception as e:
            res = 'sql语句错误，%s\n，错误的sql是【%s】'%(e,sql)
            print(res)
        else:
            res = self.cur.fetchall()
        return res
    def select_one(self,sql):
        try:
            self.cur.execute(sql)
        except Exception as e:
            res = 'sql语句错误，%s\n，错误的sql是【%s】'%(e,sql)
            print(res)
        else:
            res = self.cur.fetchone()
        return res
    def other_sql(self,sql):
        try:
            self.cur.execute(sql)
        except Exception as e:
            res = 'sql语句错误，%s\n，错误的sql是【%s】'%(e,sql)
            print(res)
            return res
    def __del__(self):
        print('拜拜啦')
        self.cur.close()
        self.coon.close()

mysql = MyDb(host='ip',user='main',password='spz123456',db='main')

# res = mysql.select_all('select * from app_myuser;')
# print(res)

import redis
r = redis.Redis(host='ip',password='HK139bc&*',decode_responses=True)

二、注册->登录->抽奖

reg_login_choice.py

from my_sql import mysql,r
import unittest
import requests

class Choujiang(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.base_url = 'http://ip/'
        cls.username = 'test_niuhy'
        cls.passwd = 'aA123456'

    def register(self):
        #注册
        url = self.base_url+'api/user/user_reg'

        data = {'username':self.username,'pwd':self.passwd,'cpwd':self.passwd}
        # sql = 'update app_myuser set is_admin = 1 where username = "%s";'%self.username
        res = requests.post(url,data).json()
        print('这是注册的结果，',res)
        self.assertEqual(res.get('error_code'),1000,msg='注册没有成功 ，实际结果是%s ' %res)

    def login(self):
        #登陆
        url = self.base_url + 'api/user/login'
        data = {'username':self.username,'passwd':self.passwd}
        res = requests.post(url,data).json()
        self.assertIn('sign',str(res),msg='登陆没有成功，实际结果是 %s'%res)
        userid = res.get('login_info').get('userId')
        sign = res.get('login_info').get('sign')
        print('登陆的结果',res)
        return userid,sign

    def choice(self,userid,sign):
        #抽奖
        url = self.base_url + 'api/product/choice'
        data = {'userid':userid,'sign':sign}
        res = requests.get(url,params=data).json()
        print('抽奖的结果',res)
        self.assertIn('product_name',str(res),msg='抽奖失败 实际结果是 %s'%res)

    def test_main(self):
        self.register()
        userid,sign = self.login()
        self.choice(userid,sign)

    @classmethod
    def tearDownClass(cls):
        sql = 'delete from app_myuser where username="%s";'%cls.username
        mysql.other_sql(sql)
        redis_key = 'choujiang:%s'%cls.username
        r.set(redis_key,0) #把抽奖次数设为0
        print('这是收尾工作')

unittest.main()

# 1、注册、 把你的用户改成管理员   app_myuser  is_admin字段代表是否为管理员
# 2、登陆、
# 3、添加商品  添加商品这里面是有文件的  app_product表