import pymysql
import sys
# python mysql 建立庫、表增刪改查標準語句
print('----------------------------')
print('載入mysql模組完成')
con = pymysql.connect(host='localhost', user='root',
                      passwd='123456', charset='utf8')
# con = pymysql.connect(host='localhost', user='root',
# passwd='123456', db='test_db',  charset='utf8') # 直接連入db1庫
print('建立連線完成')
cur = con.cursor()
print('獲取游標完成')
cur.execute("create database test_db character set utf8;")
print('建立test_db庫完成')
cur.execute("use test_db;")
print('進入test_db庫完成')
cur.execute("create table test_tab(id int,name char(20))character set utf8;")
print('建立test_tab表完成')
count = 0
while True:
    sql_insert = 'insert into test_tab(id,name) values(%s,%s);'
    ID = input('請輸入id：')
    if not ID:
        break
    name = input('請輸入名字：')
    try:
        cur.execute(sql_insert, [int(ID), name])
        count += 1
        con.commit()
        print('新增第%d條記錄' % count, int(ID), name)
    except Exception as e:
        print('出錯回滾完成', e)
        con.rollback()
        cur.close()
        con.close()
        sys.exit()
del count
print('sql寫入語句執行完成')
sql_select = 'select * from test_tab;'
sql_delete = 'delete from test_tab where id=4;'
sql_update = 'update table set name ="大胖" where id=3;'
print('建立sql語句完成')
try:
    cur.execute(sql_select)
    print('sql查詢語句執行完成')
    data_one = cur.fetchone()
    print(data_one)
    print('已顯示第一條記錄')
    data_many = cur.fetchmany(2)
    print(data_many)
    print('已顯示後面兩條記錄')
    data_all = cur.fetchall()
    print(data_all)
    print('已顯示後面全部記錄')
    cur.execute(sql_delete)
    print('sql刪除語句執行完成')
    cur.execute(sql_update)
    print('sql更新語句執行完成')
    con.commit()
    print('sql寫入完成')
except Exception as e:
    print('出錯回滾完成', e)
    con.rollback()
cur.close()
print('關閉游標完成')
con.close()
print('關閉連線完成')
print('程式結束')
print('----------------------------')