python 定义一个插入数据(可以插入到每个表中)通用的方法
前提置要:想要写一个方法,这个方法是插入数据到数据表的方法,只需要提供表名称,字段名称,还有插入的值,只要调用这个方法就可以自动帮助你插入数据
以下是不断实践优化出来
原本的插入数据库中的代码应该是这样的:
insert into ty_test_info(name, age) VALUES ("wq",23)
#*args的用法:当传入的参数个数未知,且不需要知道参数名称时。
在写代码之前就在想,有没有一种参数(传入的时候并不知道参数名称),传入参数的时候把所有表的字段名称写上,然后那边可以返回一个元组,替代上面(name,age)的,还真的就有一个*key,这种参数可以传入多个,然后返回的是一个元组
def test(a,b,c=3,*args): print(a) print(b) print(c) print(args) test(11,22,33,44,55)
输出的值为
11
22
33
(44,55)
**kargs的用法:当传入的参数个数未知,但需要知道参数的名称时(立马想到了字典,即键值对)
def test(**kwargs): print(kwargs) keys = kwargs.keys() value = kwargs.values() print(keys) print(value) test(a=1,b=2,c=3,d=4) # 输出值分别为 # {'a': 1, 'b': 2, 'c': 3, 'd': 4} # dict_keys(['a', 'b', 'c', 'd']) # dict_values([1, 2, 3, 4])
因为后面需要把要插入的数据写出来(“wq”,23),那么怎么样能够显示成这样呢.
kwargs.values()把所有的value值读取出来,先放到list里,然后把list转换成tupple这样就可以了
下面是这个方法:
def insertData(db,cursor,tablename,*key,**kwargs):# *key返回的是元组(),**返回的是字典 values = [] for value in kwargs.values(): values.append(value) print(tuple(values)) sql = 'insert into {} {}'.format(tablename,key).replace("'","")+' VALUES {}'.format(tuple(values)) print(sql) try: cursor.execute(sql) db.commit() print("成功添加数据") print("插入数据的ID:",cursor.lastrowid) except Exception as e: print(e) # 发生错误时候回滚 db.rollback()
insertData(db,cur,"ty_test_info","name","age",name = "xw",age = 18)