前提置要:想要寫一個方法,這個方法是插入數據到數據表的方法,只需要提供表名稱,字段名稱,還有插入的值,只要調用這個方法就可以自動幫助你插入數據
以下是不斷實踐優化出來
原本的插入數據庫中的代碼應該是這樣的:
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)