python常用代码积累

一、文件操作

1、判断一个目录是否存在，若不存在则创建

if not os.path.isdir(new_path):

os.makedirs(new_path)

2、新建一个文件

f=open("filename",'w')

f.close()

首先会判断这个文件是否存在，若不存在则新建，。也可以在open和close直接f.write（）函数写文件。

3、比较两个文件是否相同

import filecmp

filecmp.cmp(r'文件1'，r'文件2')

如果两个文件相同。会出输出true。否则输出false

二、系统级操作

1、获取当前运行脚本所在目录

import sys

sys.path[0]

2、获取当前工作路径

import os

os.getcwd()

3、 获取系统时间

import time   
    time.strftime('%Y-%m-%d',time.localtime(time.time()))
    输出格式类似2010-07-19

三、数据库操作

1、连接数据库

import MySQLdb

conn=MySQLdb.connect(host=host1,user=user1,passwd=pass1,port=port1,db=dbname)

cur=conn.cursor()

cur.close(）

conn.close()

cursor是用来执行命令的方法，close断开连接，可在connect和close直接进行数据库的操作

2、数据库操作

conn.select_db('python')

count=cur.execute('select * from test')

result=cur.fetchone()

results=cur.fetchmany(5)

results=cur.fetchall()

conn.commit()

select_db()选择数据库

execute()执行单条sql语句，返回值为受影响的行数

fetchone() 返回一条结果行

fetchmany(size)接收size条返回结果行.如果size的值大于返回的结果行的数量,则会返回cursor.arraysize条数据.

fetchall() 返回所有结果值

3、数据库连接失败抛出异常

try：    conn=MySQLdb.connect(host=host1,user=user1,passwd=pass1,port=port1,db=dbname)

except MySQLdb.Error,e:

print "Mysql Error %d:%s" % (e.args[0],e.args[1])

四、其他功能型操作

1、将元组的值转化成字符串，并输出到文件，用到string下的join函数

例：  import string

seq=("apple","pair","orange")

f =open("/data/cheriegong/python/test.data",'w')

f.write(' '.join(seq))

f.close()

使用string.join(list) ，会报错：TypeError: sequence item 0: expected string, int found
        li = range(10)

print '\n'.join(li)

因为join函数处理的list，元素必须是string。它不进行任何的强制类型转换。对于其他类型需要强制转换为字符串。

下面为正确的调用：

li = range(10)

print '\n'.join(str(item) for item in li)

2、将string转换成list

可以用split函数，但是很麻烦。可以使用eval函数比较方便。例：eval('[1, 2, "a"]') 可以得到list [1, 2, "a"]。大部分用str()函数生成的string，都能用eval()还原成原来的对象。

3、获取命令行输入的参数

首先，必须 from optparse import OptionParser 类，创建一个 OptionParser 对象：

parser = OptionParser()

然后，使用 add_option 来定义命令行参数：

parser.add_option("-f", "--file", action="store", type="string", dest="filename")

-f是短参数；--file是长参数，可选；type数据类型，默认为string；dest用来保存参数值得变量，如果没有指定将用参数名来对option对象的值进行存取，这里即为file。action可用值有store，store_true，stote_false等。 store_true 和 store_false，用于处理带命令行参数后面不带值的情况。

例：-v,-q 等命令行参数：

parser.add_option("-v", action="store_true", dest="verbose")

parser.add_option("-q", action="store_false", dest="verbose")

这样的话，当解析到 ‘-v’，options.verbose 将被赋予 True 值，反之，解析到 ‘-q’，会被赋予 False 值。

最后，一旦你已经定义好了所有的命令行参数，调用 parse_args() 来解析程序的命令行：

(options, args) = parser.parse_args()

parse_args() 返回的两个值：options，它是一个对象（optpars.Values），保存所有命令行参数值。只要知道命令行参数名，如 file，就可以访问其对应的值：options.file 。

例：

parser.add_option("-f", "--file", action="store", type="string", dest="filename")

args = ["-f", "foo.txt"]

(options, args) = parser.parse_args(args)

print options.filename

最后将会打印出 “foo.txt”。

4、打印0~9 十个数字

print "\n".join(str(i) for i in range(10))

也可以用for循环实现

for i in range(10):

print i;

5、import指定目录下的模块

将文件所在路径加入sys.path，sys.path是一个list。import sys;

sys.path.append("/home/a/")

import b

为了避免由于之前存在一个同名模块导致加载失败，使用以下方法：

import sys;

if not "/home/a/" in sys.path:

sys.path.append("/home/a/")

if not 'b' in sys.modules:

b = __import__('b')

else:

eval('import b')

b = eval('reload(b)')

此外，在相同目录下.py文件，可以直接被import。