[operation] write CSV to database on May 28, 2022

import pymysql
import csv
from pymysql.cursors import Cursor

# utf-8-sig --->  Signed with byte order utf-8（utf-8 with BOM）
f = open('2018 Beijing points settlement data in .csv', 'r', encoding='utf-8-sig', newline='')
reader = csv.reader(f)

# 1.  Create connection 
conn = pymysql.connect(host='localhost', port=3306,
                       user='guest', password='guest.618',
                       database='hrs', charset='utf8mb4')

try:
    next(reader)
    # 2.  Get cursor object 
    with conn.cursor() as cursor:   # type: Cursor

        # 3.  Execute through cursor objects SQL sentence 
        cursor.execute(
            'create table if not exists tb_houses'
            '('
            'id int unsigned not null,'
            'name varchar(10) not null,'
            'birthday date not null,'
            'company varchar(50) not null,'
            'score decimal(5,2) not null,'
            'primary key (id)'
            ')engine=innodb comment " Beijing integral settlement data "'
        )
        params = []
        count = 0
        for row in reader:
            row[2] = row[2] + '-01'
            params.append(row)
            count += 1
            if count % 1000 == 0:
                # insert The batch , Write the data to the database in batches 
                cursor.executemany(
                    'insert into tb_houses (id, name, birthday, company, score) '
                    'values (%s, %s, %s, %s, %s)',
                    params
                )
                params = []
        cursor.executemany(
            'insert into tb_houses (id, name, birthday, company, score) '
            'values (%s, %s, %s, %s, %s)',
            params)
    # 4.  Manual submission ( Let the previous operation take effect )
    conn.commit()

except pymysql.MySQLError as err:
    # 4.  Manual rollback ( Undo the previous operation )
    conn.rollback()
    print(err)

finally:
    # 5.  Close the connection 
    conn.close()
    f.close()