Python模块习题

# coding=utf-8
"""
习题一：

1.1 用time模块获取当前的时间戳.
1.2 用datetime获取当前的日期，例如：2013-03-29
1.3 用datetime返回一个月前的日期：比如今天是2013-3-29 一个月前的话：2013-02-27
"""
from datetime import *
import time
import os
import random

print time.time()
print date.today()
delta = date(2013, 3, 29) - date(2013, 2, 27)
print date.today() - delta

"""
习题二:
 定义一个函数kuozhan(dirpwd)，用os模块的相关方法，返回一个列表，列表包括：dirpwd路径下所有文件不重复的扩展名，
如果有2个".py"的扩展名，则返回一个".py"。

"""

def kuozhan(dirpwd):
    if os.path.isdir(dirpwd):
        listdir = os.listdir(dirpwd)
        for k in listdir:
            if os.path.isfile(k):
                result.add(os.path.splitext(k)[1])
            else:
                kuozhan(k)
    else:
        result.add(os.path.splitext(dirpwd)[1])

result = set()
kuozhan(os.getcwd())
print result

"""

习题三：

定义一个函数xulie(dirname,info) 参数：dirname:路径名，info:需要序列化的数据，
功能：将info数据序列化存储到dirname路径下随机的文件里。
"""
def xulie(dirname, info):
    info.sort()
    tmp = str(info)
    if os.path.isfile(dirname):
        f = open(dirname, 'a')
        f.write(info)
        f.close()
    else:
        dirlist = os.listdir(dirname)
        filelist = filter(lambda k: os.path.splitext(k)[1] == '.txt', dirlist)
        ran = random.randint(0, len(filelist) - 1)
        print filelist[ran]
        f = open(filelist[ran], 'a')
        f.write(tmp)
        f.close()

xulie(os.getcwd(), [3, 2, 1])