class Student(object):
    def __init__(self, name, gender, tel):
        # 姓名、性别、手机号
        self.name = name
        self.gender = gender
        self.tel = tel

    def __str__(self):
        return f'{self.name}, {self.gender}, {self.tel}'

from student import *
import pymysql
#数据库名称
DBhost = 'localhost'
DBuser = 'root'
DBpassword = '*******'#数据库系统
DBname = 'dbtest'
class StudentManager(object):
def __init__(self):
# 存储学员数据 -- 列表
self.student_list = ()
#存储学员数据--数据库
try:
self.db = pymysql.connect(host=DBhost, user=DBuser, password=DBpassword, database=DBname)
print("数据库连接成功！")
self.cur = self.db.cursor()  # 声明一个游标
# self.cur.execute('DROP TABLE IF EXISTS studentmassage')
# # 用户输入姓名、性别、手机号
# sql = "CREATE table studentmassage(Name char(20) NOT NULL,sex char(30),idcard char(20))"  # 表头标签 双引号！
# self.cur.execute(sql)  # execute执行sql语句
except pymysql.Error as a:
print("数据库连接失败！" + str(a))
# 一. 程序入口函数
def run(self):
# 1. 加载文件里面的学员数据
self.load_student()
while True:
# 2. 显示功能菜单
self.show_menu()
# 3. 用户输入目标功能序号
menu_num = int(input('请输入您需要的功能序号：'))
# 4. 根据用户输入的序号执行不同的功能 -- 如果用户输入1，执行添加
if menu_num == 1:
# 添加学员
self.add_student()
elif menu_num == 2:
# 删除学员
self.del_student()
elif menu_num == 3:
# 修改学员信息
self.modify_student()
elif menu_num == 4:
# 查询学员信息
self.search_student()
elif menu_num == 5:
# 显示所有学员信息
self.show_student()
elif menu_num == 6:
# 保存学员信息
self.save_student()
elif menu_num == 7:
# 退出系统 -- 退出循环
break
# 二. 系统功能函数
# 2.1 显示功能菜单 -- 打印序号的功能对应关系 -- 静态
@staticmethod
def show_menu():
print('请选择如下功能：')
print('1:添加学员')
print('2:删除学员')
print('3:修改学员信息')
print('4:查询学员信息')
print('5:显示所有学员信息')
print('6:保存学员信息')
print('7:退出系统')
# 2.2 添加学员
def add_student(self):
# 1. 用户输入姓名、性别、手机号
name = input('请输入您的姓名：')
gender = input('请输入您的性别：')
tel = input('请输入您的手机号：')
# 2. 创建学员对象 -- 类 ？类在student文件里面  先导入student模块，再创建对象
student = Student(name, gender, tel)
sql1 = "insert into studentmassage(Name,sex,idcard)value(%s,%s,%s)"
# 3. 将该对象添加到学员数据库
value = (student.name,student.gender,student.tel)#添加信息
#print(value)
self.cur.execute(sql1,value)#执行语句
self.db.commit()#数据提交
# 2.3 删除学员
def del_student(self):
# 1. 用户输入目标学员姓名
del_name = input('请输入要删除的学员姓名：')
sql3 = "select Name from studentmassage" #按名字查询数据库
self.cur.execute(sql3)
results1 = self.cur.fetchall()#返回的名字的元组
# print(results1)
u=0#标志变量
for ty in results1:#循环遍历
name = ty[0]
if name == del_name:#判断名字是否存在
u=1
sql2 = "delete from studentmassage where Name =%s"
self.cur.execute(sql2, del_name)#删除名字的操作
self.db.commit()
if u==0:
print("查无此人！")
# 2.4 修改学员信息
def modify_student(self):
# 1. 用户输入目标学员姓名
modify_name = input('请输入要修改的学员姓名：')
sql3 = "select Name from studentmassage"
self.cur.execute(sql3)
results1 = self.cur.fetchall()
u = 0
for ty in results1:
name = ty[0]
if name == modify_name:
u = 1
sql2 = "delete from studentmassage where Name =%s"
# newname = input("请输入新名字:")
# newsex = input("请输入性别:")
# newtel = input("请输入手机号:")
self.cur.execute(sql2,modify_name)#删除姓名
self.db.commit()
self.add_student()#执行新添加
break
if u == 0:
print("查无此人！")
# 2. 遍历列表数据，如果学员存在修改姓名性别手机号，否则提示学员不存在
# for i in self.student_list:
#     if modify_name == i.name:
#         i.name = input('姓名：')
#         i.gender = input('性别：')
#         i.tel = input('手机号：')
#         print(f'修改学员信息成功，姓名{i.name}, 性别{i.gender}, 手机号{i.tel}')
#         break
# else:
#     print('查无此人！')
# 2.5 查询学员信息
def search_student(self):
# 1. 用户输入目标学员姓名
search_name = input('请输入您要搜索的学员姓名：')
sqlselect = "select * from studentmassage"
self.cur.execute(sqlselect)
results1 = self.cur.fetchall()  # 接收查询的信息,是一个元组
#print(results1)
# 2. 遍历列表。如果学员存在打印学员信息，否则提示学员不存在
h=1
for i in results1:
name = i[0]
gread = i[1]
stuid = i[2]
if search_name == name:
print(f'姓名是{name}, 性别是{gread}, 手机号是{stuid}')
h = 0
if h==1:
print('查无此人！')
# 2.6 显示所有学员信息
def show_student(self):
sql4 = "select Name,sex,idcard from studentmassage"
self.cur.execute(sql4)
results2 = self.cur.fetchall()
print(results2)
# 2.7 保存学员信息
def save_student(self):
pass
# 2.8 加载学员信息
def load_student(self):
pass

# 1. 导入管理系统模块
from managerSystem import *
# 2. 启动管理系统
# 保证是当前文件运行才启动管理系统：if --创建对象并调用run方法
if __name__ == '__main__':
student_manager = StudentManager()
student_manager.run()