員工信息表 實現增刪改查操作

【項目要求】

可進行模糊查詢，語法至少支持下面3種:

　　select name,age from staff_table where age > 22
　　select * from staff_table where dept = "IT"
　　select * from staff_table where enroll_date like "2013%%"

可修改員工信息，語法如下:
　　update staff_table set dept="Market"  where dept = "IT"

1. 支持至少三種方法的select查詢，並在最后顯示查詢到的條數。
2. 創建新員工記錄，以phone為唯一鍵，staff_id自增
3. 輸入員工id可刪除指定員工信息記錄
4. 可以使用update命令修改指定員工信息。

 

方法一：利用數據庫實現！

具體代碼如下：

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column,Integer,CHAR,VARCHAR,DATE
from sqlalchemy.orm import sessionmaker

#建立鏈接
engine = create_engine("mysql+pymysql://root:761379545@127.0.0.1:3306/test_db?charset=utf8", max_overflow=5)
Session = sessionmaker(bind=engine)
session = Session()
Base = declarative_base()
class Emp(Base):
    __tablename__ = 'staff_table'
    id = Column(Integer,primary_key=True,autoincrement=True)
    name = Column(VARCHAR(20),nullable=False)
    age = Column(Integer,nullable=True)
    phone = Column(CHAR(11),unique=True)
    dept = Column(VARCHAR(255))
    enroll_date = Column(CHAR(20))
# #創建表格
# Base.metadata.create_all(engine)

def search_user():
    msg ='''
可進行模糊查詢，語法至少支持下面3種:
　　select name,age from staff_table where age > 22
　　select * from staff_table where dept = "IT"
　　select * from staff_table where enroll_date like('2013%%')
    '''
    while 1:
        print(msg)
        cmd = input('請輸入要進行操作的語句【返回B】：\n>>>').strip()
        if not cmd:continue
        if cmd.lower() == 'b':
            break
        try:
            data = engine.execute(cmd)
            res= data.fetchall()
            if not res:
                print('沒有滿足條件的員工！')
            else:
                for row in res:
                    print(row)
        except Exception:
            print('請輸入正確的查詢語句！')
            continue

def add_user():
    '''
    新增員工信息
    '''
    while 1:
        user_name = input('請輸入員工名【返回B】：').strip()
        if user_name.lower() == 'b':
            break
        user_age = input('請輸入年齡：').strip()
        user_age_int = int(user_age)
        user_phone = input('請寫入手機號：').strip()
        user_dept = input('所屬部門').strip()
        user_enroll_date = input('入職年月日【年-月-日】：').strip()
        try:
            session.add(Emp(name=user_name,age=user_age_int,phone=user_phone,dept=user_dept,enroll_date=user_enroll_date))
            session.commit()
            print('員工添加成功！')
        except Exception:
            print('員工添加失敗！')
            continue

def modify_user():
    msg = '''
可修改員工信息，語法如下:
　　update staff_table set dept="Market"  where dept = "IT"
    '''
    while 1:
        print(msg)
        cmd = input('請輸入要進行操作的語句【返回B】：\n>>>').strip()
        if not cmd: continue
        if cmd.lower() == 'b':
            break
        try:
            engine.execute(cmd)
            print('修改成功！')
        except Exception:
            print('請輸入正確的查詢語句！')
            continue

def delete_user():
    '''
    刪除員工信息
    '''
    while 1:
        del_id = input('請輸入您要刪除的員工ID【返回B】:').strip()
        if not del_id:continue
        if del_id.lower() == 'b':
            break
        if del_id.isdigit():
            del_id_int = int(del_id)
            try:
                session.query(Emp).filter(Emp.id==del_id_int).delete()
                session.commit()
                print('刪除成功！')
            except Exception:
                print('刪除失敗！')
                continue

#主程序
if __name__ == '__main__':
    print('歡迎進入員工信息操作界面！'.center(23,'*'))
    while 1:
        choice = input('請輸入想要進行的操作代碼：'
                       '\n\t【1】查詢員工信息'
                       '\n\t【2】新增員工信息'
                       '\n\t【3】修改員工信息'
                       '\n\t【4】刪除員工信息'
                       '\n\t【Q】退出操作'
                       '\n>>>').strip()
        if choice == '1':
            search_user()
        elif choice == '2':
            add_user()
        elif choice == '3':
            modify_user()
        elif choice == '4':
            delete_user()
        elif choice.lower() == 'q':
            print('退出成功，感謝您的使用！'.center(22,'*'))
            break
        else:
            print('您的操作有誤，請重新選擇！')

 

方法二：函數加文件讀寫實現（有待完善）！

import os
def inquiry_system():
    welcome ='''
    -----------員工信息查詢系統------------
    操作說明：
            增加ADD:1
            刪除DELETE:2
            修改CHANGE:3
            查詢SEARCH:4
            退出QUIT:5
    ---------------------------------------
    '''
    print(welcome)

def add():
    # 添加語法： name,age,phone,dept,enroll-date
    add_staff =input('請輸入要添加的員工信息：(添加格式： name,age,phone,dept(大寫）,enroll-date)\n')
    list_data =add_staff.split(',')
    list_all =[]
    f = open('staff_table.txt','r+',encoding='utf8')
    for line in f:
        list_all.append(line.strip().split(',')[3])
    if list_data[2] in list_all:
        print('用戶已存在！')
        f.close()
    else:
        for line in f:
            f.write(line)
        staff_id = str(len(list_all) + 1)
        list_data.insert(0, str(staff_id))
        f.write('\n')
        f.write(','.join(list_data))
        f.close()
        print('添加成功')

def delete():
    del_staff_id =input('請輸入要刪除的員工ID：')
    f = open('staff_table.txt','r',encoding='utf8')
    f1 = open('new_staff_table.txt','w',encoding='utf8')
    for line in f:
        each_line = line.split(',')
        if int(each_line[0]) != int(del_staff_id):
            f1.write(line)
        #若想實現刪除后員工ID順序排列使用下三行代碼！
        # elif int(each_line[0]) > int(del_staff_id):
        #     each_line[0] =str(int(each_line[0])-1)
        #     f1.write(','.join(each_line))
        else:
            continue
    f.close()
    f1.close()
    os.remove('staff_table.txt') #刪除指定目錄
    os.rename('new_staff_table.txt', 'staff_table.txt') #重命名

def change():
    # 修改請輸入（注意空格和沒有引號）：UPDATE staff_table SET dept = IT where dept = MARKET
    data = input("請輸入您要修改的信息：(修改格式：UPDATE staff_table SET dept = IT where dept = MARKET)\n")
    old = data.split(' ')[5]
    new = data.split(' ')[9]
    f = open('staff_table.txt', 'r', encoding='utf-8')
    f1 = open('new_staff_table.txt', 'w', encoding="utf-8")
    for line in f:
        if old in line:
            line = line.replace(old, new)
        f1.write(line)
    f.close()
    f1.close()
    os.remove('staff_table.txt')
    os.rename('new_staff_table.txt', 'staff_table.txt')
    print('修改成功')

def search():
    formation='''
    查詢方式1：（按年齡查詢）select * from staff_table where age >= 22
    查詢方式2：（按職業查詢）select * from staff_table where dept = "IT"
    查詢方式3：（按年份查詢）select * from staff_table where enroll_date like "2013"
    '''
    print(formation)

    while True:
        search_way = input('請輸入您的查詢方式(退出Q)：')
        res = []
        count = 0
        if search_way.lower() =='q':
            exit()
        if search_way =='1':
            age_search = input('請輸入要查詢的最小年齡：')
            with open('staff_table.txt','r',encoding='utf8') as f:
                for line in f:
                    each_line =line.split(',')
                    if int(each_line[2]) >=int(age_search):
                        res.append(line)
                        count += 1
            print("查詢的結果為：")
            for per in res:
                print(per)
            print('共有%s個員工滿足查詢信息條件!'%count)
            break
        if search_way =='2':
            dept_search = input('請輸入要查詢的職業：')
            with open('staff_table.txt', 'r', encoding='utf8') as f:
                for line in f:
                    each_line = line.split(',')
                    res.append(each_line)
                result =filter(lambda l:l[4]==dept_search.upper(),res)
                for i in result:
                    count+=1
                    print(i)
                print('共有%s個員工滿足查詢信息條件!'%count)
                break
        if search_way =='3':
            enroll_date = input('請輸入要查詢的年份：')
            with open('staff_table.txt', 'r', encoding='utf8') as f:
                for line in f:
                    each_line = line.split(',')
                    res.append(each_line)
                result = filter(lambda l:enroll_date in l[5],res)
                for i in result:
                    count += 1
                    print(i)
                print('共有%s個員工滿足查詢信息條件!' %count)
                break
        else:
            print('請重新輸入！')

my_choice = {
    '1':add,
    '2':delete,
    '3':change,
    '4':search,
    '5':'quit'
}
while True:
    inquiry_system()
    choice = input('請輸入操作序號：')
    if choice not in my_choice:
        print("輸入錯誤，請重新輸入！")
        continue
    if int(choice) == 5:
        exit()
    else:
        my_choice[choice]()