一、sys模块
包含了系统的相关的功能。我们来学习sys.argv,它包含命令行参数。
例子:定义了一个add函数,用来实现两个整数的相加。
#! coding=utf-8 # usersys.py import sys def add(a,b): print a+b print sys.argv if len(sys.argv)<2: print "argv is lower 2" else: if sys.argv[1].startswith("-") and (len(sys.argv)==2): option=sys.argv[1][2:] if option=="help": print """this is my add,two parama a and b""" elif option=="version": print "v1.0" else: print "no this option" elif len(sys.argv)==3: try: a=int(sys.argv[1]) b=int(sys.argv[2]) add(a,b) except Exception ,e: print e else: print "parama is biger 3"
在cmd命令行下,运行sysuser.py。有以下几种情况:
1、输入命令:python sysuser.py
命令行参数只有1个,此时sys.argv=["sysuser.py"]. 打印“argv is lower 2”
2、输入命令:python sysuser.py --help (或输入命令:python sysuser.py --version)
命令行参数有2个,此时sys.argv=["sysuser.py","--help"]. 打印"""this is my add,two parama a and b"""
3、输入命令:python sysuser.py 1 2
命令行参数有3个,此时sys.argv=["sysuser.py","1","2"]. 这时候做加法运算
4、其他输入情况,将进行相应处理。
我们发现sys.argv是一个列表,并且把用户输入的每个参数都以字符串的形式存储为每一个元素。所以在处理数据的时候,需要进行数据类型转换
二、os模块
这个模块包含普遍的操作系统功能。
1、os.remove()删除一个文件
import os path="E:\study\test.txt" os.remove(path)
2、os.name获取当前使用的平台,如果是windows,则返回:“nt”,如果是linux/unix,则返回:“posix”
import os print os.name
3、os.getcwd()获取当前工作的目录路径
import os print os.getcwd()
4、os.getenv()获取环境变量
import os print os.getenv("JAVA_HOME")#获取环境变量JAVA_HOME
结果:D:Program FilesJavajdk1.7.0_10
5、os.path.isfile() 和os.path.isdir()分别用来检测所给路径是文件还是目录
import os print os.path.isdir("E:\study\")
结果:True
6、os.path.exists() 检测所给路径是否存在
import os print os.path.exists("E:\study\")
7、os.path.split()用来将一个目录分割成路径和文件名
import os print os.path.split("E:\study\1.txt")
结果:('E:\study', '1.txt')
8、os.sep 获取操作系统的文件目录分割符,如Windows为
9 、os.linesep获取行终止符
import os print "ahhaha"+os.linesep+"jsjssj" print "ahhaha"+"jsjssj"
结果:
ahhaha
jsjssj
ahhahajsjssj
例子:两种遍历某目录下的文件
#! coding=utf-8
import os
import time
def getListDir(basepath):
listDir = os.listdir(basepath)
for listIndex in listDir:
listIndexPath=basepath+"//"+listIndex
if os.path.isdir(listIndexPath):
getListDir(listIndexPath)#递归遍历
else:
print (listIndexPath)
def useOsWalk(basepath):
for folder,subfolder,files in os.walk(basepath):#使用os.walk函数,该函数将迭代的返回起始目录,起始目录下的子文件夹和文件,其格式为一个元祖。
for file in files:
print (str(folder.replace("\","//"))+"//"+file)
if __name__=="__main__":
basepath = "E://test"
print("遍历文件第1种方式")
getListDir(basepath)
print("遍历文件第2种方式")
useOsWalk(basepath)
输出结果:
遍历文件第1种方式
E://test//0.0011305996553642712.txt
E://test//1.0803627858234796e-06.txt
E://test//study//0.002160185390254047.txt
E://test//study//0.003150878064854178.txt
E://test//study//0.004092954414092252.txt
E://test//study//yy//0.005170076111558261.txt
遍历文件第2种方式
E://test//0.0011305996553642712.txt
E://test//1.0803627858234796e-06.txt
E://test//study//0.002160185390254047.txt
E://test//study//0.003150878064854178.txt
E://test//study//0.004092954414092252.txt
E://test//study//yy//0.005170076111558261.txt