python主程序如何调用子程序

原创

mob649e815da088 2023-10-12 11:53:57 ©著作权

文章标签 子程序模块化 Python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e815da088的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python主程序如何调用子程序

引言

在软件开发中，程序的模块化是一个重要的概念。模块化允许我们将程序拆分成更小的部分，每个部分负责特定的功能。Python作为一种高级编程语言，提供了灵活的方式来实现程序模块化。

在本文中，我们将探讨Python主程序如何调用子程序的方法，并解决一个实际问题。我们将通过一个示例来说明如何实现这个过程。

实际问题

假设我们正在开发一个学生管理系统，其中包含以下功能：

添加学生信息
显示学生信息
删除学生信息

我们希望将这些功能拆分成不同的子程序，并由主程序调用。

子程序示例

首先，我们需要创建三个子程序来实现上述功能。我们可以将每个子程序保存在一个单独的文件中，然后在主程序中引用它们。

添加学生信息子程序

# add_student.py

def add_student(name, age):
    # 将学生信息添加到数据库或文件中的代码
    print(f"添加学生 {name} ({age}岁) 成功")

显示学生信息子程序

# display_students.py

def display_students():
    # 从数据库或文件中获取学生信息的代码
    students = [("张三", 18), ("李四", 20), ("王五", 19)]
    
    # 打印学生信息
    for name, age in students:
        print(f"学生 {name} ({age}岁)")

删除学生信息子程序

# delete_student.py

def delete_student(name):
    # 从数据库或文件中删除学生信息的代码
    print(f"删除学生 {name} 成功")

主程序调用子程序

现在，我们可以开始编写主程序，并调用上述的子程序。

# main.py

from add_student import add_student
from display_students import display_students
from delete_student import delete_student

def main():
    while True:
        print("请选择操作:")
        print("1. 添加学生信息")
        print("2. 显示学生信息")
        print("3. 删除学生信息")
        print("4. 退出")

        choice = input("请输入选项: ")

        if choice == "1":
            name = input("请输入学生姓名: ")
            age = input("请输入学生年龄: ")
            add_student(name, age)
        elif choice == "2":
            display_students()
        elif choice == "3":
            name = input("请输入要删除学生的姓名: ")
            delete_student(name)
        elif choice == "4":
            break
        else:
            print("无效选项，请重新输入")

if __name__ == "__main__":
    main()

在上面的示例中，我们使用from语句导入了每个子程序的函数。然后，我们在主程序中使用这些函数来调用相应的子程序。

主程序通过一个无限循环来提供菜单选项，根据用户的选择调用不同的子程序。当用户选择退出时，循环终止。

关系图

erDiagram
    MAINPROGRAM }|..| ADDSTUDENT : imports
    MAINPROGRAM }|..| DISPLAYSTUDENTS : imports
    MAINPROGRAM }|..| DELETESTUDENT : imports
    MAINPROGRAM }|--| ADDSTUDENT : calls
    MAINPROGRAM }|--| DISPLAYSTUDENTS : calls
    MAINPROGRAM }|--| DELETESTUDENT : calls

上述关系图显示了主程序与三个子程序之间的依赖关系。