Python入门_自定义包的实战

 

trans/tools.py

import random
from datetime import datetime


def gen_trans_id(date = None):
    """
    根据所传入的时间得到一个唯一的交易流水ID
    :param date: 日期
    :return: 交易流水ID字符串
    """
    #如果没有传入时间
    if date is None:
        date = datetime.now()
    #增氧保证字符串的唯一
    #日期+时间+毫秒+随机数（6位随机数）
    vaule = "{0}{1}".format(date.strftime("%Y%m%d%H%M%S%f"), random.randint(100000,999999))
    #vaule = date.strftime("%Y%m%d%H%M%S%f" + str(random.randint(100000,999999)))
    return vaule

 

work/constants.py

##文件类型常量
FILE_TYPE_UNKNOWN = 10     # 未知文件类型
FILE_TYPE_IMG = 11         # 图片类型的文件
FILE_TYPE_DOC = 12         # word文档
FILE_TYPE_EXCEL = 13       # excel文档
FILE_TYPE_PPT = 14         # ppt文档

 

work/tools.py

import os.path
from work import constants


def get_file_type(file_name):
    """
    根据文件的名称来判断文件的类型
    :param file_name:str 文件名称
    :return:int 文件类型
    -1: 未知文件类型
    0: 图片类型的文件
    1: word文档
    2: excel文档
    3: ppt文档
    """
    # 默认文件是未知类型的
    result = constants.FILE_TYPE_UNKNOWN
    # 传进来的必须时一个文件的名字
    if not os.path.isfile(file_name):
        return result
    path_name, ext = os.path.splitext(file_name)
    # 将文件的后缀名统一成小写
    ext = ext.lower()
    # 图片类型的文件
    if ext in (".png", ".jpg", ".gif", ".bmp"):
        result = constants.FILE_TYPE_IMG
    # word文档
    elif ext in (".doc", ".docx"):
        result = constants.FILE_TYPE_DOC
    # Excel 文档
    elif ext in (".xls",".xlsx"):
        result = constants.FILE_TYPE_EXCEL
    # ppt文档
    elif ext in (".ppt", ".pptx"):
        result = constants.FILE_TYPE_PPT
    return result

 

test_module.py

from datetime import datetime
from trans import tools as trans_tools
from work import tools as work_tools


def test_trans_tool():
    """测试trans包下的tools模块"""
    id1 = trans_tools.gen_trans_id()
    print(id1)
    date = datetime(2021, 10, 2, 12, 30, 45)
    id2 = trans_tools.en_trans_id(date)
    print(id2)

def get_file_type():
    """测试work模块"""
    path_name = "C:\\Users\\Administrator\\Desktop\\videojet laser.ppt"
    rest = work_tools.get_file_type(path_name)
    print(rest)

if __name__ == "__main__":
    #test_trans_tool()
    get_file_type()