模块

python中创建了很多py文件，每个py文件我们官方说法就是 模块。
在模块可以编写一些代码。调用的时候只能在 当前模块中调用。那么能不能跨文件进行调用呢？
肯定是可以。
可以不同的代码放在不同的模块中，可以同过引用模块进行调用。

创建新项目
image.png
image.png
可以选择 新窗口 打开
image.png

创建模块

在项目上创建一个python文件
image.png
文件命名为 utils 用来存放一些自己定义的常用工具。
image.png
在文件中编写工具方法，将获取手机号的函数复制进来。
utils.py

  1. # 导入python中的随机数生成模块
  2. import random
  3. def get_phone():
  4.     """
  5.     生成一个随机的11手机号码， 手机号码中 前2位 ['13','14','15','17','18']
  7.     :return:
  8.     """
  9.     # 定义一个空的字符串 表示手机号
  10.     phone=""
  11.     # 定义手机号的支持的前两位
  12.     pre_phone = ['13','14','15','17','18']
  13.     # 随机从列表中选择一个 需要使用到随机函数
  14.     p1 = random.choice(pre_phone) # 从列表中随机选择一个元素
  15.     # print(p1)
  16.     # 将随机获取的前两位放到手机号中
  17.     phone=phone+p1
  18.     # 生成9位随机的数字 （0，9） 循环9次
  19.     for i in range(9):
  20.         # 生成1位随机数
  21.         n = random.randint(0,9)
  22.         # print(n)
  23.         # 将随机数字添加到手机号 n 位数字，数字不能直接跟字符串拼接 str(n) 将数字转换位字符串
  24.         phone = phone+str(n)
  26.     # 循环完成之后将电话号码返回
  27.     return phone
  30. p = get_phone()
  31. print(f"生成随机电话 {p}")

image.png
在项目的根目录下创建一个 testdemo.py
image.png

调用模块中的方法 import

下面要在 testdemo.py 文件中调用 utils.py 中写的代码

  1. # 导入 utils 模块
  2. import utils
  5. # 调用 utils 文件的get_phone() 函数
  6. phone = utils.get_phone()
  7. print(f'调用utils文件中获取电话号码的功能, {phone}')

image.png

testdemo.py 中的代码
image.png

  1. import utils

这一句 会将 utlis.py 文件中的内容加载到 testdemo.py 文件中。
image.png
也就是说，在执行 testdemo.py 文件的时候，因为 testdemo.py 中有 import utils 导入utils.py 文件，那么执行会先执行 utils.py 文件中的代码。因为utils.py 文件中有打印，所以在执行 testdemo.py 的时候会 打印两句。
image.png

模块中的调试

在代码编辑器中输入 main 会自动提示语句
image.png

utils.py

  1. # 导入python中的随机数生成模块
  2. import random
  3. def get_phone():
  4.     """
  5.     生成一个随机的11手机号码， 手机号码中 前2位 ['13','14','15','17','18']
  7.     :return:
  8.     """
  9.     # 定义一个空的字符串 表示手机号
  10.     phone=""
  11.     # 定义手机号的支持的前两位
  12.     pre_phone = ['13','14','15','17','18']
  13.     # 随机从列表中选择一个 需要使用到随机函数
  14.     p1 = random.choice(pre_phone) # 从列表中随机选择一个元素
  15.     # print(p1)
  16.     # 将随机获取的前两位放到手机号中
  17.     phone=phone+p1
  18.     # 生成9位随机的数字 （0，9） 循环9次
  19.     for i in range(9):
  20.         # 生成1位随机数
  21.         n = random.randint(0,9)
  22.         # print(n)
  23.         # 将随机数字添加到手机号 n 位数字，数字不能直接跟字符串拼接 str(n) 将数字转换位字符串
  24.         phone = phone+str(n)
  26.     # 循环完成之后将电话号码返回
  27.     return phone
  30. if __name__ == '__main__':
  32.     p = get_phone()
  33.     print(f"生成随机电话 {p}")

testdemo.py

  1. # 导入 utils 模块
  2. import utils
  6. if __name__ == '__main__':
  8.     # 调用 utils 文件的get_phone() 函数
  9.     phone = utils.get_phone()
  10.     print(f'调用utils文件中获取电话号码的功能, {phone}')

再次执行 testdemo.py
image.png

完善代码

utils.py

  1. # 导入python中的随机数生成模块
  2. import random
  3. import csv
  5. def get_phone():
  6.     """
  7.     生成一个随机的11手机号码， 手机号码中 前2位 ['13','14','15','17','18']
  9.     :return:
  10.     """
  11.     # 定义一个空的字符串 表示手机号
  12.     phone=""
  13.     # 定义手机号的支持的前两位
  14.     pre_phone = ['13','14','15','17','18']
  15.     # 随机从列表中选择一个 需要使用到随机函数
  16.     p1 = random.choice(pre_phone) # 从列表中随机选择一个元素
  17.     # print(p1)
  18.     # 将随机获取的前两位放到手机号中
  19.     phone=phone+p1
  20.     # 生成9位随机的数字 （0，9） 循环9次
  21.     for i in range(9):
  22.         # 生成1位随机数
  23.         n = random.randint(0,9)
  24.         # print(n)
  25.         # 将随机数字添加到手机号 n 位数字，数字不能直接跟字符串拼接 str(n) 将数字转换位字符串
  26.         phone = phone+str(n)
  28.     # 循环完成之后将电话号码返回
  29.     return phone
  31. def get_area():
  32.     """
  33.     生成随机的6位地区信息
  34.     :return: 6个数字字符串
  35.     """
  36.     # 从csv文件中读取所有的地区码，随机选择一个
  37.     # 定义空的list 用来存放地区码
  38.     area_code=[]
  39.     with open('gb2260.csv',mode='r',encoding='utf8') as file:
  40.         lines = csv.reader(file)
  41.         for line in lines:
  42.             # 每个地区码 都放在list 中
  43.             area_code.append(line)
  45.     # 随机从 所有地区码中选择一个
  46.     code = random.choice(area_code)
  47.     # code 值选择出来是 一个list 需要将list 转换位字符串
  48.     code_str = ''.join(code)  # ''.join(code) 将list 转换位字符串
  49.     # 将随机选择的code 返回
  50.     return code_str
  52. def get_birth():
  53.     """
  54.     生成随机的 出生年月日 1960-2022出生
  55.     :return:
  56.     """
  57.     # 年份
  58.     year = random.randint(1960,2022)
  59.     # 月份
  60.     month = random.randint(1,12)
  62.     # 日
  63.     # 考虑闰年，月份
  64.     # 如果年份是闰年，并且月份2月， 有29天
  65.     if year%4 == 0 and month == 2:
  66.         day = random.randint(1,29)
  67.     # 在 1,3,5,7,8,10,12 月中 有31天
  68.     elif month in [1,3,5,7,8,10,12]:
  69.         day = random.randint(1,31)
  70.     elif month in [4,6,9,11]:
  71.         day = random.randint(1,30)
  72.     # 如果以上都不是 就是 平年的2月份
  73.     else:
  74.         day = random.randint(1,28)
  76.     # 如果天为个位数 前面添加0
  77.     if day<10:
  78.         day = "0"+str(day)
  79.     # 月份前 01， 02
  80.     if month < 10:
  81.         month = "0" + str(month)
  83.     # 返回年月日
  84.     return f'{year}{month}{day}'
  86. def get_last():
  87.     """
  88.     最后4位
  90.     :return:
  91.     """
  92.     pre_3=""
  93.     for i in range(3):
  94.         n = random.randint(0,9)
  95.         pre_3 = pre_3+str(n)
  96.     last = random.choice(['0','1','2','3','4','5','6','7','8','9','X'])
  97.     # 返回
  98.     return f'{pre_3}{last}'
  101. if __name__ == '__main__':
  103.     p = get_phone()
  104.     print(f"生成随机电话 {p}")

在testdemo.py 中引用

  1. # 导入 utils 模块
  2. import utils
  4. def get_id():
  5.     """
  6.     生成身份证id
  7.     :return:
  8.     """
  9.     # 调用 utils 中定义函数
  10.     area_code = utils.get_area()
  11.     ymd_code = utils.get_birth()
  12.     last_code = utils.get_last()
  14.     return area_code+ymd_code+last_code
  17. if __name__ == '__main__':
  19.     # 调用 utils 文件的get_phone() 函数
  20.     phone = utils.get_phone()
  21.     print(f'调用utils文件中获取电话号码的功能, {phone}')
  23.     user_id = get_id()
  24.     print(f'生成的身份证id {user_id}')

运行testdemo.py代码
image.png

所有的代码
main.zip

导入模块的两种方式

使用import 导入

创建python 文件
demo.py

  2. def f1():
  3.     print('demo文件中的f1 函数')
  6. if __name__ == '__main__':
  7.     f1()

在main.py 文件引用 demo.py 中定义的f1函数。
main.py

  2. import demo
  4. if __name__ == '__main__':
  5.     demo.f1()

通过import 方式导入文件 使用 文件中定义的函数。
image.png

使用from…import…导入

也可以使用 from … import… 方式来进行导入。
demo.py

  2. def f1():
  3.     print('demo文件中的f1 函数')
  5. def f2():
  6.     print('demo文件中的f2 函数')
  9. if __name__ == '__main__':
  10.     f1()

在main.py 文件中导入

  2. # 使用 from 方式来导入 demo 文件中的 f2
  3. from demo import f2
  7. if __name__ == '__main__':
  8.     # 可以直接使用
  9.     f2()

使用 import 和 from … import 功能都是一样的。

from… import * 导入所有

main.py 中引入 demo.py 文件中所有的函数
main.py

  1. # * 导入demo 文件中所有的函数
  2. from demo import *
  4. if __name__ == '__main__':
  5.     f2()
  6.     f1()

使用 import 导入可以，使用from … import … 的方式也行。两种方法都可以使用。使用的时候根据自己的使用习惯即可。

比如生成随机数，你可以这样写

  1. import random
  3. print(random.randomint(1,10))

也可以使用from的方式

  1. from random import randint
  3. print(randint(1,10))

包

image.png
目前来说，项目文件都在根目录下，当编写的代码越来越多的时候，需要将代码按照不同的功能放在不同的文件夹。这些不同文件夹在Python中称为 包

创建Python包

image.png
开发过程中，一些常用的函数都会放在 common 包，所以为这个包 命名为 common
image.png
在common包中创建 file.py 文件，这个文件后面主要用来解析一些数据文件。
image.png
在files.py 文件中 编写代码。
common/files.py

  2. def parse_csv():
  3.     print("files中解析csv文件的函数")
  6. if __name__ == '__main__':
  7.     parse_csv()

引入包中的文件

引入包中的文件原理跟引入模块中函数原理一样，通过import 或者 from方式来引用。

import导入

在main.py 中导入

  1. # import导入。需要跟上包的路径  common.files  as 给路径起别名（别名可以随意）
  2. import common.files as f
  6. if __name__ == '__main__':
  7.     # 调用包中文件的函数
  8.     f.parse_csv()

from 导入

也可以通过from 的方式来导入包

  1. # from 包.文件的路径 导入
  2. from common.files import parse_csv
  5. if __name__ == '__main__':
  6.     # 调用包中文件的函数
  7.     parse_csv()

不同的包中引用

创建 一个 testcases 的包，在testcase 包中创建 test_demo.py 文件。
image.png
在test_demo.py 文件中 引用 common包下 files.py 文件中定义的函数。
testcases/test_demo.py

  2. import common.files as f
  3. from common.files import parse_csv
  6. if __name__ == '__main__':
  7.     f.parse_csv()
  8.     parse_csv()

运行都可以引用到
image.png

加强练习

定义解析csv文件
common/files.py

  1. import csv
  2. def parse_csv(filename):
  3.     """
  4.     通过传入一个csv文件，将csv文件中的内容解析出来存放到列表中返回。
  6.     :param filename: 传入的文件路径
  7.     :return:
  8.     """
  9.     # 定义空的列表存放csv文件中的数据
  10.     data = []
  11.     with open(filename,mode='r',encoding='utf8') as f:
  12.         lines = csv.reader(f)
  13.         for line in lines:
  14.             # 将读取出来的数据放到data中
  15.             data.append(line)
  16.     # 数据读取完成之后
  17.     return data

files.py 文件中 传入csv文件来测试 上面编写的函数是否正常。

  1. import csv
  2. def parse_csv(filename):
  3.     """
  4.     通过传入一个csv文件，将csv文件中的内容解析出来存放到列表中返回。
  6.     :param filename: 传入的文件路径
  7.     :return:
  8.     """
  9.     # 定义空的列表存放csv文件中的数据
  10.     data = []
  11.     with open(filename,mode='r',encoding='utf8') as f:
  12.         lines = csv.reader(f)
  13.         for line in lines:
  14.             # 将读取出来的数据放到data中
  15.             data.append(line)
  16.     # 数据读取完成之后
  17.     return data
  20. if __name__ == '__main__':
  21.     # 检查一下 parse_csv 函数是否正常运行  使用绝对路径 可以解析出这个文件的内容。
  22.     # data = parse_csv(r'C:\Users\zengy\PycharmProjects\pythonProject10\gb2260.csv')
  23.     import os
  24.     # 获取当前文件所在的目录  获取目录名
  25.     dir = os.path.dirname(__file__)
  26.     print(dir)
  27.     # 找到项目的根目录   获取目录名
  28.     project_root = os.path.dirname(dir)
  29.     print('项目根目录',project_root)
  30.     # csv文件路径  join 方法可以将两个路径拼接到一起
  31.     csvfile = os.path.join(project_root,'gb2260.csv')
  32.     print(f'csv 文件路径: {csvfile}')
  33.     data = parse_csv(csvfile)
  34.     print(data)

image.png
封装获取项目根目录下的方式。

common/files.py

  1. import csv
  2. import os
  3. def get_project_root():
  4.     """
  5.     在解析文件的时候，需要拼接文件的路径，这个函数 返回项目的根目录
  7.     :return:
  8.     """
  9.     # 获取当前文件所在的目录
  10.     dir = os.path.dirname(__file__) # 找common 目录
  11.     print(dir)   # C:\Users\zengy\PycharmProjects\pythonProject10\common
  12.     # 找到项目的根目录
  13.     project_root = os.path.dirname(dir)  # common的上级目录 当前项目的根目录
  14.     print(project_root)  # C:\Users\zengy\PycharmProjects\pythonProject10
  15.     return project_root
  18. def parse_csv(filename):
  19.     """
  20.     通过传入一个csv文件，将csv文件中的内容解析出来存放到列表中返回。
  22.     :param filename: 传入的文件路径
  23.     :return:
  24.     """
  25.     # 定义空的列表存放csv文件中的数据
  26.     data = []
  27.     with open(filename,mode='r',encoding='utf8') as f:
  28.         lines = csv.reader(f)
  29.         for line in lines:
  30.             # 将读取出来的数据放到data中
  31.             data.append(line)
  32.     # 数据读取完成之后
  33.     return data
  36. if __name__ == '__main__':
  37.     # 检查一下 parse_csv 函数是否正常运行  使用绝对路径 可以解析出这个文件的内容。
  38.     # data = parse_csv(r'C:\Users\zengy\PycharmProjects\pythonProject10\gb2260.csv')
  39.     # 获取到项目的根目录
  40.     project_root = get_project_root()
  41.     # csv文件路径  join 方法可以将两个路径拼接到一起
  42.     csvfile = os.path.join(project_root,'gb2260.csv')
  43.     print(f'csv 文件路径: {csvfile}')
  44.     data = parse_csv(csvfile)
  45.     print(data)

现在在test_demo.py 文件解析 gb2260.csv 文件
image.png
通过获取项目的根目录来拼接路径 进行操作。

testcases/test_demo.py

  1. import os
  2. from common.files import parse_csv,get_project_root
  5. if __name__ == '__main__':
  6.     # 拼接csv路径
  7.     # 1. 拿到项目的根目录
  8.     root = get_project_root()
  9.     # 2. 拼接csv文件
  10.     csvfile = os.path.join(root,'gb2260.csv')
  11.     # 解析文件路径
  12.     data = parse_csv(csvfile)
  13.     print(data)

image.png

多层目录引用文件

image.png

testcases/test_csv/test_file.py

  1. import os
  2. from common.files import parse_csv,get_project_root
  5. if __name__ == '__main__':
  6.     # 拼接csv路径
  7.     # 1. 拿到项目的根目录
  8.     root = get_project_root()
  9.     # 2. 拼接csv文件
  10.     csvfile = os.path.join(root,'gb2260.csv')
  11.     # 解析文件路径
  12.     data = parse_csv(csvfile)
  13.     print(data)

在这个文件，通过调用封装好的 获取项目根目录路径的方法 在任何地方调用，都可以取到项目的根目录，再进行路径拼接。就可以调用里面的方法。

项目数据文件解析

在做一个项目的时候，需要对单接口做csv数据参数化，每个接口需要准备对应的csv文件。如果有很多csv文件，需要将csv文件放在对应目录中。
在项目中 创建 testdata 目录，这个目录用来存放测试数据。
创建目录
image.png
目录名称为 testdata
image.png
在testdata 目录中放入 csv文件
image.png

解析testdata 目录中的csv文件
testcases/test_demo.py

  1. import os
  2. from common.files import parse_csv,get_project_root
  5. if __name__ == '__main__':
  6.     # 拼接csv路径
  7.     # 1. 拿到项目的根目录
  8.     # root = get_project_root()
  9.     # # 2. 拼接csv文件
  10.     # csvfile = os.path.join(root,'gb2260.csv')
  11.     # # 解析文件路径
  12.     # data = parse_csv(csvfile)
  13.     # print(data)
  14.     # 1. 获取系统的根目录
  15.     root = get_project_root()
  16.     # 2. 拿到文件路径
  17.     topic_csv = os.path.join(root,'testdata/testtopic.csv')
  18.     # 解析文件
  19.     data = parse_csv(topic_csv)
  20.     print(data)

image.png

附件

所有对应的代码
demo.zip

作业

在项目的common包下创建 utils.py 文件 实现生成手机号码 和身份证id的函数
image.png

  2. def generate_id():
  3.     """
  4.     生成一个随机的身份证号 并返回
  6.     :return:
  7.     """
  10. def generate_phone():
  11.     """
  12.     生成一个随机的 11位手机号码
  13.     :return:
  14.     """

答案

  1. def generate_phone():
  2.     """
  3.     生成一个随机的 11位手机号码
  4.     :return:
  5.     """
  6.     #确定前两位手机
  7.     pre_2 =["13","14","15","16","17","18"]
  8.     # 随机选择一个元素
  9.     p2 = random.choice(pre_2)
  10.     # 生成9位随机数字
  11.     l9 = ""
  12.     for i in range(9):
  13.         # 每次循环生成一个随机数字
  14.         n = random.randint(0,9)
  15.         l9 = l9+str(n)  # 将数字转换为字符串之后进行拼接
  16.     return p2+l9
  18. if __name__ == '__main__':
  19.     phone = generate_phone()
  20.     print(phone)
  1. import random
  2. from common.files import get_project_root,parse_csv
  3. import os
  4. def generate_id():
  5.     """
  6.     生成一个随机的身份证号 并返回
  8.     :return:
  9.     """
  10.     # 拿到项目根目录
  11.     root = get_project_root()
  12.     # 拼接csv文件
  13.     file_csv = os.path.join(root,'gb2260.csv')
  14.     # 通过解析csv文件获取所有的地区编码
  15.     all_code = parse_csv(file_csv)
  16.     print(all_code)
  17.     # 随机选择一个列表
  18.     code = random.choice(all_code)
  19.     print(code)
  20.     # 列表转换为字符串
  21.     p6 = ''.join(code)
  22.     print(p6)
  23.     # 设置出生年月日
  24.     year = random.randint(1960,2020)
  25.     month = random.randint(1,12)
  26.     # 闰年 2月份 29天
  27.     if year%4==0 and month == 2:
  28.         day=random.randint(1,29)
  29.     # 31天
  30.     elif month in [1,3,5,7,8,10,12]:
  31.         day = random.randint(1,31)
  32.     elif month in [4,6,9,11]:
  33.         day = random.randint(1,30)
  34.     # 平年的2月 28天
  35.     else:
  36.         day = random.randint(1,28)
  37.     # month ，day 小于10 前面补0
  38.     if month<10:
  39.         #转换为字符串
  40.         month = f'0{month}'
  41.     if day <10:
  42.         day = f'0{day}'  
  43.     # 合并年月日 转换为字符串
  44.     ymd = f'{year}{month}{day}'
  45.     print(ymd)
  46.     # 获取最后4位 最后一位X
  47.     l3 = ""
  48.     for i in range(3):
  49.         n = random.randint(0,9)
  50.         # 数字转换字符串 拼接
  51.         l3 = l3+str(n)
  52.     l1 = list(range(10))
  53.     l1.append("X") # [0,1,2,3,4,5,6,7,8,9,"X"]
  54.     print(l1)
  55.     l = random.choice(l1) # 最后1位
  56.     l4 = l3 + str(l) # 最后4位
  58.     return f'{p6}{ymd}{l4}'
  61. if __name__ == '__main__':
  62.     id=generate_id()
  63.     print(id)

最终文件

  1. import random
  2. from common.files import get_project_root,parse_csv
  3. import os
  4. def generate_id():
  5.     """
  6.     生成一个随机的身份证号 并返回
  8.     :return:
  9.     """
  10.     # 拿到项目根目录
  11.     root = get_project_root()
  12.     # 拼接csv文件
  13.     file_csv = os.path.join(root,'gb2260.csv')
  14.     # 通过解析csv文件获取所有的地区编码
  15.     all_code = parse_csv(file_csv)
  16.     # print(all_code)
  17.     # 随机选择一个列表
  18.     code = random.choice(all_code)
  19.     # print(code)
  20.     # 列表转换为字符串
  21.     p6 = ''.join(code)
  22.     #print(p6)
  23.     # 设置出生年月日
  24.     year = random.randint(1960,2020)
  25.     month = random.randint(1,12)
  26.     # 闰年 2月份 29天
  27.     if year%4==0 and month == 2:
  28.         day=random.randint(1,29)
  29.     # 31天
  30.     elif month in [1,3,5,7,8,10,12]:
  31.         day = random.randint(1,31)
  32.     elif month in [4,6,9,11]:
  33.         day = random.randint(1,30)
  34.     # 平年的2月 28天
  35.     else:
  36.         day = random.randint(1,28)
  37.     # month ，day 小于10 前面补0
  38.     if month<10:
  39.         month = f'0{month}'
  40.     if day <10:
  41.         day = f'0{day}'
  42.     # 合并年月日 转换为字符串
  43.     ymd = f'{year}{month}{day}'
  44.     # print(ymd)
  45.     # 获取最后4位 最后一位X
  46.     l3 = ""
  47.     for i in range(3):
  48.         n = random.randint(0,9)
  49.         # 数字转换字符串 拼接
  50.         l3 = l3+str(n)
  51.     l1 = list(range(10)) #生成[0,1,2,3,4,5,6,7,8,9] 列表
  52.     l1.append("X") # [0,1,2,3,4,5,6,7,8,9,"X"]
  53.     # print(l1)
  54.     l = random.choice(l1) #  随机从列表中选择一个
  55.     l4 = l3 + str(l) # 最后4位
  57.     return f'{p6}{ymd}{l4}'
  62. def generate_phone():
  63.     """
  64.     生成一个随机的 11位手机号码
  65.     :return:
  66.     """
  67.     #确定前两位手机
  68.     pre_2 =["13","14","15","16","17","18"]
  69.     # 随机选择一个元素
  70.     p2 = random.choice(pre_2)
  71.     # 生成9位随机数字
  72.     l9 = ""
  73.     for i in range(9):
  74.         # 每次循环生成一个随机数字
  75.         n = random.randint(0,9)
  76.         l9 = l9+str(n)  # 将数字转换为字符串之后进行拼接
  77.     return p2+l9
  79. if __name__ == '__main__':
  80.     id = generate_id()
  81.     print(f'最终生成的id：{id}')