#!/usr/bin/env python
# -*- coding:utf-8 -*-
#====#====#====#====
#Author:
#CreatDate:
#Version:
#====#====#====#====
import pymysql
class DBClass:
#获取数据库连接对象
def con_db(self,hostname1="QZX",username1="root",password1="123456",dbname1="atm"):#默认参数
db= pymysql.connect(hostname1,username1,password1,dbname1,charset="utf8")#连接数据库
return db
#查询方法
def search_db(self,sql):
db= DBClass().con_db()#得到数据库连接对象
cu= db.cursor()#获取游标
cu.execute(sql)#通过游标执行sql
db.commit()#提交
# datas = cu.fetchall()#查询所有
datas= cu.fetchone()#查询一条
# datas = cu.fetchmany(3)#查询多条
db.close()#关闭数据库连接
return datas#返回查询结果
#增删改方法
def fix_db(self,sql):
db= DBClass().con_db()# 得到数据库连接对象
cu= db.cursor()# 获取游标
cu.execute(sql)# 通过游标执行sql
db.commit()# 提交
db.close()# 关闭数据库连接
if __name__== '__main__':
#初始化数据库
#删除表userinfo
print("开始初始化数据库:")
DBClass().fix_db("drop table userinfo")
print("表删除成功!")
#创建表userinfo
DBClass().fix_db("create table userinfo(id int primary key not null auto_increment,username varchar(20) not null,password varchar(20),phone varchar(20),count float not null)")
print("表创建成功!")
print("数据库初始化完成!")