Pyhton定时生成模拟数据并存入Mysql

设计思路：

1.python 使用随机数模块，生成所需要的模拟数据。

2.将模拟数据定时存入mysql

3.设置定时器，定时执行

ps：数据库和表的创建在mysql的shell中以完成，模拟数据存入test表中，每次生成单条记录。

# -*- coding:utf-8 -*-

import MySQLdb
import schedule
import time
import datetime
import random
import string

class SaveToSql(object):
def __init__(self, conn):
self.conn = conn
# 生成随机数函数
def random_str(self, randomlength=random.randint(1,10)):
a = list(string.ascii_letters)
random.shuffle(a)
return ''.join(a[:randomlength])
# 存储函数
def save(self):
cursor = self.conn.cursor()

try:
sql = "insert into test(time, name, type, data) values('%s','%s', '%s', '%s')"% (datetime.datetime.now(),self.random_str(),self.random_str(),self.random_str())
cursor.execute(sql)

print('Insert the data: ', sql)
rs = cursor.rowcount
"""判断数据库表中数据所影响行数是否为1，
如果不是的话就进行异常抛出"""
if rs != 1:
raise Exception("Error of data inserting.")
self.conn.rollback()
self.conn.commit()
finally:
cursor.close()

if __name__ == "__main__":
conn = MySQLdb.connect(
host = '127.0.0.1',
port = 3306,
user = 'root',
passwd = '',
db = 'test',
charset = 'utf8'
)

def job():
save_data = SaveToSql(conn)
save_data.save()

try:
schedule.every(10).seconds.do(job)
except Exception as e:
print('Error: %s'% e)
#   finally:
#       conn.close()

while True:
schedule.run_pending()
time.sleep(1)