• 【解决方案】数据随机生成脚本


    因需要生成数据,验证数据库相关问题,且此需求可能以后也会经常遇到,因此分享出来~

    在基础的云ECS,通过python3执行即可

    1. # encoding=utf-8
    2. import pymysql
    3. import random
    4. import time
    5. # create table student_source(id int NOT NULL PRIMARY KEY AUTO_INCREMENT,name varchar(20),subject varchar(20),score int(5))
    6. class RandomDataInsert:
    7. # 初始化类
    8. def __init__(self, ty):
    9. self.type = ty
    10. self.coon = None
    11. self.sqlList = []
    12. pass
    13. # 连接数据库
    14. def connectDB(self, ip, port, dbname, user=None, pwd=None, ):
    15. if self.type == 'mysql':
    16. self.coon = pymysql.connect(
    17. host=ip,
    18. user=user,
    19. passwd=pwd,
    20. port=port,
    21. db=dbname,
    22. charset='utf8'
    23. )
    24. return True
    25. # 关闭连接
    26. def disconnectDB(self):
    27. self.coon.close()
    28. # 添加sql语句
    29. def appendSql(self, sql):
    30. if sql is not None:
    31. self.sqlList.append(sql)
    32. # 执行SQL
    33. def executeInsertAll(self):
    34. try:
    35. cur = self.coon.cursor()
    36. for sql in self.sqlList:
    37. print("Going to execute: ", sql)
    38. cur.execute(sql)
    39. self.coon.commit()
    40. cur.close()
    41. except Exception as e:
    42. print('发生异常', e)
    43. self.coon.rollback()
    44. # 生成数据SQL小工厂
    45. def defineSql(ope, count):
    46. for i in range(count):
    47. subject = random.choice(['Chinese', 'Math', 'English'])
    48. score = random.randint(0, 100)
    49. name = ''.join(random.sample(
    50. ['z', 'y', 'x', 'w', 'v', 'u', 't', 's', 'r', 'q', 'p', 'o', 'n', 'm', 'l', 'k', 'j', 'i', 'h', 'g', 'f',
    51. 'e', 'd', 'c', 'b', 'a'], 2))
    52. sql = "insert into student values(null,'" + name + "','" + subject + "'," + str(score) + ");"
    53. ope.appendSql(sql)
    54. # 主函数
    55. if __name__ == '__main__':
    56. print('Random DB Insert begin!!')
    57. sqlList = RandomDataInsert('mysql')
    58. sqlList.connectDB('实例名称', 3306, '数据库', '账户', '密码')
    59. print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())))
    60. defineSql(sqlList, 20)
    61. print("defineSql,sqlList:{}", len(sqlList.sqlList))
    62. sqlList.executeInsertAll()
    63. sqlList.disconnectDB()

  • 相关阅读:
    springboot写一个简单的接口样例
    设计模式:桥接模式
    python中random模块求随机数
    计算有效声压
    JavaWeb开发之——数据库设计(20)
    初识设计模式 - 解释器模式
    shell脚本中export命令未生效,原因详解
    JavaScript中的Error错误对象与自定义错误类型
    【JPA】批量操作数据时耗时长的问题解决方案
    软件测试培训之写给要学习自动化测试的同学的建议
  • 原文地址:https://blog.csdn.net/victorzzzz/article/details/128000557