用python 实现生成双色球小程序

生成双色球小程序：
 #输入n，随机产生n条双色球号码，插入n条数据库
    #表结构: seq
        CREATE TABLE `seq` (
          `id` int(11) NOT NULL AUTO_INCREMENT,
          `red` varchar(100) NOT NULL,
          `blue` varchar(20) NOT NULL,
          `date` datetime DEFAULT NULL,
          PRIMARY KEY (`id`)
        ) ENGINE=InnoDB DEFAULT CHARSET=utf8;
##“双色球”每注投注号码由6个红色球号码和1个蓝色球号码组成。红色球号码从1--33中选择；蓝色球号码从1--16中选择。
#05 09 10 12 17 19 13

 import random,pymysql,time
 #处理球的号码，如果数字是1-9，则在前面加0，-> 09
 def process_num(num):
     if num in range(1,10):
         num = str(num)
         new_num = '' + num
     else:
         new_num = str(num)
     return  new_num
 def op_mysql(ball_list):
     conn = pymysql.connect(host='localhost',user='root',passwd='',port=3306,db='sakila',charset='utf8')
     cur = conn.cursor()
     sql = "insert into seq(red,blue,date) values(%s,%s,%s)"
     cur.executemany(sql,ball_list) #批量执行
     conn.commit()
     cur.close()
     conn.close()
 def generate_ball(count):
     red_ball = process_num(random.randint(1,34))
     blue_ball = process_num(random.randint(1,17))
     red_balls = [process_num(x) for x in range(1,34)] # 返回红球01-33的list
     blue_balls = [process_num(x) for x in range(1,17)] # 返回蓝球01-16的list
     ball_list = []
     for i in range(count):
         red_num = random.sample(red_balls, 6)  # 随机生成6位红球list
         blue_num = random.sample(blue_balls, 1)
         cur_time = time.strftime('%Y-%m-%d %H:%M:%S')#当前时间格式化
         ball = (','.join(red_num),blue_num, cur_time)# 将红，蓝球，时间加入元祖
         ball_list.append(ball) # 把每个结果加入list，[('25,18,23,28,24,19', '07', '2018-01-22 22:05:03'), ('16,23,27,07,31,21', '04', '2018-01-22 22:05:03')]
     op_mysql(ball_list)

 if __name__ =='__main__':
     count = int(input('请输入你想生成的双色球号码数量：').strip())
     generate_ball(count)