从零开始撸python

Day3 ：超市管理系统2.0（mysql版）

实现第一阶段飞跃~，达到科班本科毕业水平，后端基本打通，接下来修整一周时间，学习前端内容hh

 

 

 

import market

def main():
    print("********************************************超市管理系统2.0******************************************")
    print("*************1.查看商品列表")
    print("*************2.根据编号查询商品")
    print("*************3.添加商品")
    print("*************4.根据编号删除商品")
    print("*************5.商品打折")
    print("*************6.查看所有订单")
    print("*************7.删除订单")
    print("*************8.订单统计")
    print("*************9.商品结算")
    print("*************10.退出")
    print("****************************************************************************************************")
    c=int(input("请选择："))
    if c==1:
        market.getAllProducts()
    elif c==2:
        market.getProduct()
    elif c==3:
        market.addProduct()
    elif c==4:
        market.delProduct()
    elif c==5:
        market.setDiscount()
    elif c==6:
        market.getAllOrders()
    elif c==7:
        market.delOrder()
    elif c==8:
        market.acountOrders()
    elif c==9:
        market.settle()
    else:
        print("****************************************************************************************************")
        

if __name__=='__main__':
    main()

import orm
import random

#查看商品列表

def getAllProducts():
    sql="SELECT * FROM products;"
    data=orm.getData(sql)
    print("序号        编号         名称      单价        折扣")
    for product in data:
        for x in product:
            print(x,end="      ")
        print()


#根据编号查询商品
def getProduct():
    num=input("请输入商品编号：")
    sql="SELECT * FROM products WHERE num="+num+";"
    data=orm.getData(sql)
    if data!=None:
        print("商品名称：",data[0][2]," 单价：",data[0][3]," 折扣：",data[0][4])
        return data[0][2],data[0][3],data[0][4]
    else:
        print("商品不存在！")
        return None



#添加商品
def addProduct():
    name=input("请输入商品名称：")
    num=random.randint(1000,9999)
    price=int(input("请输入商品价格："))
    sql="INSERT INTO products(num,name,price,discount) VALUES({},'{}',{},1)".format(num,name,price)
    r=orm.writeData(sql)
    print(r)



#删除商品
def delProduct():
    num=input("请输入待删除商品信息:")
    sql="DELETE FROM products WHERE num='{}'".format(num)
    r=orm.writeData(sql)
    if r==0:
        print("删除失败!")
    else:
        print("商品",num,"被删除！")


#订单统计(总销量、销售额)
def acountOrders():
    sql="SELECT * FROM orders;"
    data=orm.getData(sql)
    totalCount=0
    totalAmount=0
    for orders in data:
        totalCount+=orders[2]
        totalAmount+=orders[3]
    print("总销量",totalCount,"件！销售额",totalAmount,"元！")


#商品折扣（修改折扣信息）
def setDiscount():
    num=input("请输入需要修改折扣的商品编号：")
    discount=float(input("请输入要设置的折扣："))

    if 0.1<discount    <1:
            sql="UPDATE products SET discount={} WHERE num={};".format(discount,num)
            r=orm.writeData    (sql)
            if r==0:
                print("设置失败！")
            else:
                print("商品{}折扣设置成功".format    (num))
    else:
        print("折扣信息输入错误！")
    










#查看所有订单
def getAllOrders():
    sql="SELECT * FROM orders;"
    data=orm.getData(sql)
    print("序号        编号         数量      金额")
    for orders in data:
        for x in orders:
            print(x,end="      ")
        print()
    print("--------------------------------")
#删除订单
def delOrder():
    num=input("请输入待删除订单信息:")
    sql="DELETE FROM orders WHERE num='{}'".format(num)
    r=orm.writeData(sql)
    if r==0:
        print("删除失败!")
    else:
        print("订单",num,"被删除！")


#商品结算
def settle():
    # 输入商品编号和数量
    # 计算金额
    # 重复添加多个商品
    # 计算订单的总金额、数量
    # 在订单中添加订单
    orderCount=0
    orderAmount=0
    msg=0 #保存订单是否有效

    while 1==1:
        data=getProduct()
        # print(data)
        num=int(input("请输入商品数量："))
        if data!=None:
            msg=1
            price=data[1]
            discount=data[2]
            amount=float(price)*float(discount)*num
            orderCount+=num
            orderAmount+=amount
            print("当前添加{}件！金额{}元！".format(num,amount))
        c=int(input("继续添加请输入1，结算请输入2 :"))
        if c==1:
            continue
        elif c==2:
            print("=========================================================")
            break
        else:
            print("*****************不要闹！*****************")
            exit(0)

    print("*********当前您购买的总数量{}件！总金额{}元！".format(orderCount,orderAmount))    


#添加订单
    if msg==1:
        oid=str(random.randint(1000,9999))
        sql="INSERT INTO orders(num,count,amount) VALUES({},{},{});".format(oid,orderCount,orderAmount)
        orm.writeData(sql)
        print("-----------添加成功！") 

#数据库操作

import pymysql

host="localhost"
port=3306
user="root"
password="775825"
productdb="python"
charset="utf8"

#链接数据库
def getConection():
    db=pymysql.Connect(host=host,port=port,user=user,passwd=password,db=productdb,charset=charset)
    print("--------------------------------链接成功!")
    return db
#查询数据

def getData(sql):
    db=getConection()
    cursor=db.cursor()
    try:
        cursor.execute(sql)
        data=cursor.fetchall()
    except Exception as e:
        print("出现异常",e)
    finally:
        cursor.close()
        db.close()
    return data



#更新数据

def writeData(sql):
    db=getConection()
    cursor=db.cursor()
    r=0
    try:
        r=cursor.execute(sql)
        db.commit()
        print("------------------数据已更新{}条！".format(r))
    except Exception as e:
        print("出现异常",e)
    finally:
        cursor.close()
        db.close()
    return r         #返回受影响的行数