#!/usr/bin/env python3
# coding: utf-8
# author: hyh
import pymysql
class CreateTable:
def __init__(self, tab_num, host, port, user, passwd, db):
self.tab_num = tab_num
self.host = host
self.port = port
self.user = user
self.passwd = passwd
self.db = db
def connect_database(self):
"""连接数据库"""
con = pymysql.connect(host=self.host, port=self.port, user=self.user, passwd=self.passwd, db=self.db)
return con
def crate_table(self):
"""创建表"""
db = self.connect_database()
cursor = db.cursor()
for i in range(self.tab_num):
sql = 'CREATE TABLE table_name_{i}(id int(5),name VARCHAR(16))'.format(i=i)
cursor.execute(sql)
db.commit()
db.close()