1. 程式人生 > 其它 >python 定義一個插入資料(可以插入到每個表中)通用的方法

python 定義一個插入資料(可以插入到每個表中)通用的方法

技術標籤: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的用法:當傳入的引數個數未知,但需要知道引數的名稱時(立馬想到了字典,即鍵值對)

'''
遇到問題沒人解答?小編建立了一個Python學習交流QQ群:778463939
尋找有志同道合的小夥伴,互幫互助,群裡還有不錯的視訊學習教程和PDF電子書!
'''
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)