一.項(xiàng)目背景

一個(gè)目錄下有若干個(gè)Csv/Excel文件，現(xiàn)在需求是根據(jù)這些Csv文件的表頭，生成對
應(yīng)的SQL建表語句。

二.實(shí)現(xiàn)過程

1.獲取文件
》》遍歷路徑下文件
》》獲取Excel文件
》》返回文件路徑

#獲取文件路徑
def get_path(file_path):
    #存儲(chǔ)路徑信息
    path_list=[]
    #遍歷路徑下文件
    for file in os.listdir(file_path):
        #獲取文件名稱
        file_name=os.path.splitext(file)[0]
        #判斷是否為Excel文件
        if file.endswith('.xlsx'):
            #添加文件路徑
            path_list.append((os.path.join(file_path,file),file_name))
    #返回文件路徑
    return path_list

2.構(gòu)建Mysql類
》》初始化Mysql連接
》》運(yùn)行SQL語句

#MySQL類
class Mysql_connect():
    def __init__(self, host, user, password, database):
        #初始化連接信息
        self.conn = pymysql.connect(host=host,
                                    user=user,
                                    password=password,
                                    database=database
                                    )

    #運(yùn)行sql語句
    def run_sql(self, sql):
        #獲取游標(biāo)
        cursor = self.conn.cursor()
        #執(zhí)行SQL語句
        cursor.execute(sql)
        #提交事務(wù)
        self.conn.commit()
        #關(guān)閉連接
        self.conn.close()

3.獲取sql語句
》》讀取文件名稱
》》獲取表頭和類型
》》生成SQL語句

#獲取sql語句
def get_sql(table_name,df,length=255):
    '''
    table_name：表名稱
    df:DataFrame數(shù)據(jù)結(jié)構(gòu)
    length：默認(rèn)是255
    '''
    
    #存儲(chǔ)sql語句列表
    create_list=[]
    #遍歷每一列
    for col in df.columns.tolist():
        #判斷數(shù)據(jù)類型
        if df[col].dtypes=='float':
            #構(gòu)建sql語句
            sql_col=col+' '+'float'
            #添加到列表
            create_list.append(sql_col)
        #判斷數(shù)據(jù)類型
        elif df[col].dtypes=='int64' or df[col].dtypes=='int32' or df[col].dtypes=='int':
            #構(gòu)建sql語句
            sql_col=col+' '+'int'
            #添加到列表
            create_list.append(sql_col)
         #判斷數(shù)據(jù)類型
        elif df[col].dtypes=='datetime64[ns]':
             #構(gòu)建sql語句
            sql_col=col+' '+'datetime'
            #添加到列表
            create_list.append(sql_col)
        else:
            #構(gòu)建sql語句
            sql_col=col+' '+f'varchar({length})'
            #添加到列表
            create_list.append(sql_col)
    #添加主鍵，替換列表中不必要內(nèi)容
    table_str='id int(64) primary key not null auto_increment,'+str(create_list).replace('[','').replace(']','').replace("'","")
    #設(shè)置引擎
    sql_create=f'''create table if not exists {table_name}({table_str})'''+' engine=innodb default charset=utf8'
    #返回sql語句
    return sql_create

4.生成數(shù)據(jù)庫表格
》》運(yùn)行SQL語句生成表格

#遍歷該目錄下文件
for files in get_path('./'):
    #獲取文件名稱
    name=files[1]
    #讀取表格
    df=pd.read_excel(files[0])
    #獲取sql語句
    sql=get_sql(name,df)
    #連接MySQL
    conn=Mysql_connect('127.0.0.1','root','253611','test')
    #輸出sql語句
    print(sql+'n')
    #運(yùn)行sql語句
    conn.run_sql(sql)