PythonTip >> 博文 >> python

使用python开发命令行程序的知识点

zihua 2014-01-20 23:01:43 点击: 775 | 收藏


#==========================

# 几个必须了解的模块: os, os.path和shutil

#==========================

写命令行程序, 经常要和操作系统和文件打交道,

关于os,os.path, shutil模块的介绍

http://www.cnblogs.com/lovemo1314/archive/2010/11/08/1871781.html

http://docs.python.org/library/os.html

http://docs.python.org/library/shutil.html#module-shutil

#==========================

# subprocess 模块

#==========================

python2.6后推荐使用subprocess模块来调用外部程序, 包括shell.

subprocess.call(*popenargs, **kwargs)

运行命令, 该函数将一直等待到子进程运行结束,并返回进程的returncode。

p=subprocess.Popen(*popenargs, **kwargs))

可以使用Popen来创建子进程,并可与子进程进行复杂的交互, 功能相当强大. 但该函数并不等子进程结束, 就会返回到父进程.

python31高级[创建进程]

http://www.cnblogs.com/itech/archive/2010/12/31/1922427.html

python中的subprocess.Popen的使用

http://apps.hi.baidu.com/share/detail/20257791

#==========================

# 其他文件常用操作

#==========================

判断文件是否存在, 可以使用:

os.path.isfile(flg2File)

判断是否是路径, 或则路径是否存在:

os.path.isdir(path)

判断路径是否存在, 可以使用:

os.path.exists(self.backupPath)

获取文件的大小, 可以使用:

os.path.getsize(newGzipFullName2)

拼接路径, 可以使用:

newGzipFullName2=os.path.join(self.dataPath, newGzipFile)

遍历path下所有的文件目录(包括子目录), 可以使用:

for root, dirs, files in os.walk(path):

os.walk()本质上是一个generator, 所以需要使用foreach来遍历, 每次循环, root是当前接受遍历的目录, dirs是root下的直接子目录列表, files是root下直接文件列表.

获取path本层目录的文件

def getShortFilesInFolder(self, path=""):

result=[]

for root, dirs, files in os.walk(path):

if root==path:                

result=files

break

return(result)

解压tar.gz, 可以使用tarfile模块,         

if tarfile.is_tarfile(fileFullName):

tar =tarfile.open(fileFullName)

tar.extractall(workingPath)

tar.close()

合并几个文件, 也可以很方便地使用tarfile模块完成.

tar =tarfile.open(fileFullName,"w")

tar.add(file1,"fileA") #"fileA"为在tarinfo中的显示名称, 如果缺省的话, 即为file1的全路径

tar.add(file2,"fileB")

tar.close()

压缩几个文件成tar.gz, 原则上也可以使用tarfile模块完成, 只不过需要指定open的格式为"w|gz",  即tarfile.open(fileFullName,"w|gz"). 但是这么做, gz中tar文件的显示名, 永远为tar文件的全路径. 如使用下面方法,gz中tar文件的显示名为tar文件短文件名.  

os.chdir(workingPath)

cmd ="tar -zcf newfile.tar.gz newfile*"

subprocess.call(cmd, shell=True)   

#==========================

# 在console打印两个方法的区别:

#==========================

1. print()自动会自动加上换行

print("abc")

print("def")

2. 如果不需要子自动换行, 需要使用sys.stdout.write()

import sys

sys.stdout.write('abc')

sys.stdout.write('def')

sys.stdout.write('\n') #new line

#==========================

# 在console上等待用户的输入

#==========================

命令行程序有时候需要用户输入一些东西:

guess = raw_input('guess a letter: ') #string

nb = input('Choose a number') #number

#==========================

# 命令行参数解析器

#==========================     

python optparse 模块使用 - python学习 - 博客园

http://zoomquiet.org/res/scrapbook/ZqFLOSS/data/20100903115917/    

或者, 参考下面这个简单的代码:

class TableFilesBatchRename:

def parseArguments(self):

cmdln_args=sys.argv[1:]

#print(cmdln_args)

#Sample is: bulkinit.py dataPath=D:\1 table=SDB_TB_EQP_GROUP backup=True

argKeyValues=dict([arg.split("=") for arg in cmdln_args])

"""

for arg in argKeyValues.iteritems():

print(arg)

"""

self.dataPath=argKeyValues["dataPath"]

self.backupPath=os.path.join(self.dataPath,"backup")

self.tableName=argKeyValues["table"]

if (argKeyValues.has_key("backup")):

self.backup=argKeyValues["backup"]

def printUsage(self):

usage="""

=======================================

One utility for edw project

=======================================        

bulkInit is to batchly rename all data files and flag files related to one table .

Usage:

python bulkInit.py dataPath=your_data_Path table=your_table backup=True|False

Example:

python bulkInit.py dataPath=/bac/kkk/working  table=table_a backup=True   

"""   

print(usage)

if __name__ == '__main__':

processor=TableFilesBatchRename()

parsed=False    

try:

processor.parseArguments()

parsed=True

except Exception as ex:

print("Argument parse failed.")

processor.printUsage()

if(parsed):               

processor.execute()

print("............done")

#==========================

# 如何处理文本文件:

#==========================

text_file = open("Output.txt", "w")

text_file.write("Purchase Amount: %s"%TotalAmount)

text_file.close()

在python 2.5之后 由于有了context manager, 就可以使用with语法, 在with语句结束时, 有系统来保证自动关闭文件句柄.

with open("Output.txt", "w") as text_file:

text_file.write("Purchase Amount: %s"%TotalAmount)

#==========================

# 如何sleep    

#==========================         

import time    

time.sleep(5)    

#==========================

# 如何使用dom方式访问XML,

#==========================

可以import xml.dom.minidom的parse和parseString()来解析XML, 可以使用getDOMImplementation()来用来修改XML文件

http://docs.python.org/library/xml.dom.minidom.html

#==========================

# 更高效XML库,

#==========================

lxml处理速度很快,关于xpath, 支持弱化版的xpath, 名为ElementPath.

libxml2是另一个高效包, 但比lxml难用多了, 推荐使用lxml.

http://www.iteye.com/topic/890812

#==========================

# 如何记录log

#==========================

使用logging记录log, 用法同log4j

http://docs.python.org/howto/logging.html#logging-basic-tutorial

import logging

logging.basicConfig(filename='example.log',level=logging.DEBUG)

logging.debug('This message should go to the log file')

logging.info('So should this')

logging.warning('And this, too')

原文链接:http://www.wumii.com/item/15J5dhygG

作者:zihua | 分类: python | 标签: python | 阅读: 775 | 发布于: 2014-01-20 23时 |