[Mysql] Pymysql INSERT 쿼리 생성기

def input_sql(data, input_table_name, pk=None):
    """
    data:               Input Data          DataFrame
    input_table_name:   Table name          Str
    pk:                 Primary Key Name    List
    """
    colnames = list(data.columns)
    INPUT_NAME = ''
    for col in colnames: 
        INPUT_NAME = INPUT_NAME + col.upper() +','
    INPUT_NAME = INPUT_NAME[:-1]
    VALUES_INPUT = ''
    for col in colnames: 
        VALUES_INPUT = VALUES_INPUT +'"%s",'
    VALUES_INPUT = VALUES_INPUT[:-1]
 
    sql = f"INSERT INTO {input_table_name} ({INPUT_NAME}) VALUES ({VALUES_INPUT})"
    if pk==None:
        sql = f"INSERT INTO {input_table_name} ({INPUT_NAME}) VALUES ({VALUES_INPUT})"
    else:
        UPDATE_VALUE = ''
        for key in pk: 
            UPDATE_VALUE = UPDATE_VALUE +key+"="+'"%s",'
        UPDATE_VALUE=UPDATE_VALUE[:-1]
        sql = f"INSERT INTO {input_table_name} ({INPUT_NAME}) VALUES ({VALUES_INPUT}) ON DUPLICATE  KEY UPDATE {UPDATE_VALUE}"
    return sql

 

### Pymysql 함수 

 

cursor.execute(sql)


cursor.executemany(sql, list(set))
** Format Type 주의