#!/usr/bin/env python # -*- encoding: utf-8 -*- # @Date : 2015-03-05 16:51:15 # @Author : NSSimacer # @Email : wuxiaoqiang1020@gmail.com # @Version : 1.0 import MySQLdb import string import random def connect(): ''' 创建数据库连接 ''' try: conn = MySQLdb.connect( host='127.0.0.1', user='user', # 你的数据库用户名 passwd='passwd', # 你的数据库密码 db='db', # 你的数据库名 port=3306) except MySQLdb.Error, e: print 'MySQL Error %d: %s' % (e.args[0], e.args[1]) return conn def save_coupon_to_db(coupon_list): ''' 把激活码/优惠券保存到数据库 ''' try: conn = connect() cursor = conn.cursor() cursor.execute('drop database if exists py_db') cursor.execute('create database py_db') conn.select_db('py_db') cursor.execute('drop table if exists t_coupon') cursor.execute('create table t_coupon(\ id int(3) primary key,\ coupon varchar(8))') values = [] for i in xrange(len(coupon_list)): values.append((i + 1, coupon_list[i])) cursor.executemany('insert into t_coupon values(%s, %s)', values) print 'Finish!' conn.commit() cursor.close() conn.close() except MySQLdb.Error, e: print 'MySQL Error %d: %s' % (e.args[0], e.args[1]) def load_coupon(): ''' 从数据库中载入激活码/优惠券 ''' try: conn = connect() cursor = conn.cursor() print 'Loading data …' count = cursor.execute('select * from t_coupon') result = cursor.fetchall() for r in result: print r[1] print 'There are %s rows.' % count conn.commit() cursor.close() conn.close() except MySQLdb.Error, e: print 'MySQL Error %d: %s' % (e.args[0], e.args[1]) def generate_coupon(coupon_num, coupon_length): ''' 产生激活码/优惠券 ''' coupon_list = [] coupon_character_list = list(string.uppercase + string.digits) for i in xrange(coupon_num): coupon_str = '' for j in xrange(coupon_length): coupon_str += random.choice(coupon_character_list) coupon_list.append(coupon_str) return coupon_list if __name__ == '__main__': coupon_num = 200 coupon_length = 8 save_coupon_to_db(generate_coupon(coupon_num, coupon_length)) load_coupon()