os模块习题

os

1.使用python代码统计一个文件夹中所有文件的总大小

import os
def func(path):    
    size_sum = 0#文件总大小为0
    name_lst = os.listdir(path)#拿到这个文件夹下所有文件夹和文件
    for name in name_lst:
        path_abs = os.path.join(path,name)#将每个文件拼接为目录
        if os.path.isdir(path_abs):#如果是文件夹
            size = func(path_abs)#继续打开这个文件
            size_sum += size
        else:
            size_sum += os.path.getsize(path_abs)#是文件就取大小
    return size_sum

ret = func(path')
print(ret)

2.写一个函数,接收一个参数,如果是文件,就执行这个文件,如果是文件夹,就执行这个文件夹下的所有py文件

import os
def func(path):#接收文件路径
    if os.path.isfile(path) and path.endswith(".py"):#如果这个路径是文件且是以.py为结尾(是个py文件)
        os.system("python %s"% path)#执行这个文件
    elif os.path.isdir(path):#如果这个路径是文件夹
        for name in os.listdir(path):#取这个文件夹下的每个子目录
            abs_path = os.path.join(path,name)#将每个子目录拼接路径
            if abs_path.endswith(".py"):#如果这个子目录是以.py为结尾
                os.system("python %s"% ans_path)#执行这个文件
func(path)

3.写一个copy函数,接收两个参数,第一个参数是源文件的位置,第二个参数是目标位置,将源文件copy到目标位置,还需要判断一下这个文件之前是否存在

import os
def copy(path1,path2)
    filename = os.path.basename(path1)#拿出这个文件的名字
    if os.path.isfile(path1) and os.path.isdir(path2):#如果path1这个路径是文件而且path2这个路径是文件夹
        if os.path.exists(path):#如果这个文件名已存在
            print("已有同名文件")
        with open(path1,"rb") as f1,open(os.path.join(path2,filename),""wb"") as f2:#以读的方式打开path1,以写的方式打开path2
            content = f1.read()#读path1的文件
            f2.write(content)#写入path2
copy(path1,path2)

4.获取某个文件所在目录的上一级目录

path1 = os.path.dirname(path)
base_name = os.path.basename(path1)
print(base_name)

可以合并为

base_name = os.path.basename(os.path.dirname(path))
print(base_name)

5.写一个用户注册登录的程序,每一个用户的注册都要把用户名和密码用字典的格式写入文件userinfo,在登录的时候,再从文件中读取信息进行验证

import pickle
#注册
def register():
    user = input("username:")
    pwd = input("password:")
    dic = {"username":user,"password":pwd}
    with open("userinfo","ab") as f:
        pickle.dump(dic,f)

#登录
def login():
    flag = True
    user = input("username:")
    pwd = input("password:")
    with open("userinfo","rb") as f:
        while flag:
            try:
                dic = pickle.load(f)
                if user == dic["username"] and pwd == dic["password"]:
                    print("登录成功")
                    flag = False
                    break
            except EOFError:
                print("登录失败")
                break
register()
login()

6.发红包

import random
def red_packet(money,num):
    money = money*100
    ret = random.sample(range(1,money),num - 1)
    ret.sort()
    ret.insert(0,0)
    ret.append(money)
    for i in range(len(ret)-1):
        yield(ret[i+1]-ret[i])/100
res = red_packet(200,10)
for money in res:
    print(money)