import sqlite3
class Student(object):
def __init__(self,name='',age='',tel = ''):
self.name = name
self.age = age
self.tel = tel
class DBAction(object):
def __init__(self,dbName,tableName):
self.dbName = dbName
self.tableName= tableName
#屬性可以在勒種任意地方使用
self.connect = None
self.cuedor = None
def createDBAnaTable(self):
self.connect = sqlite3.connect('{}'.format(self.dbName))
self.cursor = self.connect.cursor()
self.cursor.execute('create table if not exists"{}"(name text,age text,tel text)'.format(self.tableName))
def commitAndClose(self):
self.connect.commit()
self.cursor.close()
self.connect.close()
def openDB(self):
self.connect = sqlite3.connect('{}'.format(self.dbName))
self.cursor = self.connect.cursor()
def addNewStudentToTable(self,student):
self.openDB()
self.cursor.execute('insert into "{}"(name,age,tel)VALUES ("{}","{}","{}")'.format(self.tableName,student.name,student.age,student.tel))
self.commitAndClose()
def deleteStudenttoTable(self,name):
self.openDB()
self.cursor.execute('delete from "{}" WHERE name ="{}"'.format(self.tableName,name))
self.commitAndClose()
# student = Student('王老吉',107,123)
myDB = DBAction('BDAction','tableAction')
myDB.createDBAnaTable()
while True:
value = input("""
請輸入操作:
1.增加學生資訊
2.删除學生資訊
3.修改學生資訊
4.檢視學生資訊
""")
value = int(value)
if value ==1:
name = input('請輸入姓名')
age = input('請輸入年齡')
tel = input('請輸入聯系方式')
student = Student(name,age,tel)
myDB.addNewStudentToTable(student)
if value == 2:
name = input('請輸入要删除的學生資訊')
myDB.deleteStudenttoTable(name)