#!/usr/bin/python #!/usr/bin/env python # -*- coding: utf-8 -*- # @Time : 2018/6/15 22:46 # @Author : Kwan # @File : insert_db.py # @Software: PyCharm import sqlite3 import random import datetime # conn = sqlite3.connect('local.db') # # c = conn.cursor() # # c.execute("insert into system_cfg values(?,?,?)",(2,'test2',1)) # # conn.commit() # # conn.close() def make_date(): # def make_card_number(bits): # counter = bits # number_list = [] # while counter: # number_list.append(str(random.randrange(0, 10))) # counter -= 1 # return number_list number_list = [str(x) for x in range(0, 10)] card_number = '' card_number = card_number.join(tuple(random.choices(number_list,k=10))) # card_number = '' # card_number = card_number.join(tuple(make_card_number(10))) car_type = random.choice(('m','p')) localtime = datetime.date.isoformat(datetime.datetime.now()) data_title = ['card_number','car_type','localtime'] data_dict = dict.fromkeys(data_title) data_dict['card_number'] = card_number data_dict['car_type'] = car_type data_dict['localtime'] = localtime return data_dict def insert_data(list): conn = sqlite3.connect('test.db') c = conn.cursor() try: id = max(c.execute("select max(id) from monthly_card").fetchall()) max_id = int(id[0]) for data in list: card_number = data['card_number'] db_car_number = c.execute("select card_number from monthly_card").fetchall() if card_number in db_car_number: break else: car_type = data['car_type'] localtime = data['localtime'] max_id += 1 c.execute("insert into monthly_card values (?,?,?,?)", (max_id, card_number, car_type, localtime)) except sqlite3.OperationalError: sql = '''create table if not exists monthly_card (id int primary key not null, card_number text not null, car_type text not null, valid text not null);''' c.execute(sql) id = 0 for data in list: try: card_number = data['card_number'] db_car_number = c.execute("select card_number from monthly_card").fetchall() if card_number in db_car_number: break else: car_type = data['car_type'] localtime = data['localtime'] id += 1 c.execute("insert into monthly_card values (?,?,?,?)", (id, card_number, car_type, localtime)) except sqlite3.OperationalError: card_number = data['car_number'] car_type = data['car_type'] localtime = data['localtime'] id += 1 c.execute("insert into monthly_card values (?,?,?,?)", (id, card_number, car_type, localtime)) # print(type(id_1)) # id = id_1.fetchall() conn.commit() conn.close() # def select_data(): # conn = sqlite3.connect('test.db') # c = conn.cursor() # db_car_number = c.execute("select card_number") # test1 = make_date() # my_list = [] # my_list.append(test1) # insert_data(my_list) # print('OK') n = 500 my_list = [] while n: my_list.append(make_date()) n-=1 insert_data(my_list) print('OK')
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。