(python) 标准模块sys和os的使用

一、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 Files\Java\jdk1.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