import os

import string

import random

import pymysql

def main():

    config = {

    'host': '192.168.3.222',

    'port': 3306,

    'user': 'root',

    'password': 'root',

    'db': 'test',

    'charset': 'utf8mb4',

    'cursorclass': pymysql.cursors.DictCursor,

    }

    conn = pymysql.connect(**config)    #建立数据库连接

    rowId = 0

    if os.path.exists('./activateCode.txt'):

        os.remove('./activateCode.txt')

        print("重新生成激活码文件！")

    chars = string.ascii_letters + string.digits    #a-zA-Z0-9

    codeNumber = int(input("请输入需要生成的激活码数量："))

    codeLength = int(input("请输入需要生成的激活码长度："))

    if codeNumber == '':

        codeNumber = 10

    if codeLength == '':

        codeLength = 8

    for i in range(codeNumber):

        code = random.choices(list(chars),k=codeLength)

        with conn.cursor() as cursor:

            sql = "insert into CODES (id,number) values(%s,%s)"

            cursor.execute(sql,(str(rowId),''.join(code)))  #执行sql语句

            conn.commit()   #提交数据库

            rowId += 1

    conn.close()    #关闭连接

if __name__ == "__main__":

    main()