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)