自定义包的实现

项目的结构图示

trans/tools

from datetime import datetime import random def gen_trans_id(date=None): """ 根据所传入的时间得到一个唯一的交易流水ID :param date: 日期 :return: 交易流水的ID字符串 """ # 如果没有传入时间,则使用系统当前的时间 if date is None: date = datetime.now() # 怎样保证字符串的唯一 # 日期+时间+毫秒+随机数(6位随机数) #return'{0}{1}'.format(date.strftime('%Y%m%d%H%M%S%f'),random.randint(10000,999999)) return date.strftime('%Y%m%d%H%M%S%f') + str(random.randint(10000,999999))

work/tools

import os.path 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

constants

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

test_module

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

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(2015, 10, 1, 15, 1, 30) id2 = trans_tools.gen_trans_id(date) print(id2) def test_work_tools(): """测试work模块""" file_name = 'C:\\Users\\徐达\\Pictures\\Saved Pictures\\151165695.jpg' rest = work_tools.get_file_type(file_name) print(rest) if __name__ == '__main__': test_trans_tool() test_work_tools()