9–1. 文件过滤. 显示一个文件的所有行, 忽略以井号( # )开头的行. 这个字符被用做Python , Perl, Tcl, 等大多脚本文件的注释符号.附加题: 处理不是第一个字符开头的注释.
1 filename = input("输入文件名：") 2 with open(filename) as f: 3 for i in f: 4 if i.startswith('#'): 5 continue 6 else: 7 print(i, end='')
9–2. 文件访问. 提示输入数字 N 和文件 F, 然后显示文件 F 的前 N 行.
1 num = int(input("输入数字：")) 2 filename = input("输入文件名：") 3 with open(filename) as f: 4 for i, j in enumerate(f): 5 if i == (num): 6 break 7 else: 8 print(j, end='')
9–3. 文件信息. 提示输入一个文件名, 然后显示这个文本文件的总行数.
1 filename = input('输入文件名：') 2 with open(filename) as f: 3 allline = f.readlines() 4 print(len(allline))
9–4. 文件访问.
写一个逐页显示文本文件的程序. 提示输入一个文件名, 每次显示文本文件的 25 行, 暂停并向用户提示”按任意键继续.”, 按键后继续执行.
1 filename = input("输入文件名：") 2 num = 25 3 with open(filename) as f: 4 for i, j in enumerate(f, 1): 5 if i == num: 6 space = input("输入任意字符回车：") 7 num += 25 8 else: 9 print(j, end='')
9-5 考试成绩，改进你的考试成绩问题（练习5-3和6-4），要求能从多个文件中读入考试成绩。文件的数据格式由你自己决定。
1 filename = input("输入文件名：") 2 with open(filename) as f: 3 for i in f: 4 i = i.strip() 5 if int(i) < 60: 6 print("F") 7 elif int(i) < 70: 8 print("D") 9 elif int(i) < 80: 10 print("C") 11 elif int(i) < 90: 12 print("B") 13 elif int(i) <= 100: 14 print("A") 15 else: 16 print("%s不在范围" % i)
9–6. 文件比较. 写一个比较两个文本文件的程序. 如果不同, 给出第一个不同处的行号和列号.
1 def file(filename1, filename2): 2 with open(filename1) as f1: 3 f1alllines = f1.readlines() 4 5 with open(filename2) as f2: 6 f2alllines = f2.readlines() 7 8 minlen1 = min(len(f1alllines), len(f2alllines)) 9 for i in range(minlen1): 10 if f1alllines[i] != f2alllines[i]: 11 12 minlen2 = min(len(f1alllines[i]), len(f2alllines)) 13 for j in range(minlen2): 14 if f1alllines[i][j] != f2alllines[i][j]: 15 return '不同行号: %s' % (i+1), '不同列号: %s' % (j+1) 16 return '文件相同' 17 18 if __name__ == '__main__': 19 filename1 = input("输入文件名1：") 20 filename2 = input("输入文件名2：") 21 print(file(filename1, filename2))
9–7. 解析文件. Win32 用户: 创建一个用来解析 Windows .ini 文件的程序. POSIX 用户:创建一个解析 /etc/serves 文件的程序. 其它平台用户: 写一个解析特定结构的系统配置文件的程序.
1 windows = {} 2 with open(r'C:\Windows\win.ini') as f: 3 # print(f.readlines()) 4 for line in f: 5 if line.startswith(';'): 6 continue 7 if line.startswith('['): 8 iterm = [] 9 name = line[1: line.rfind(']')] 10 windows.setdefault(name, iterm) 11 continue 12 if '=' in line: 13 windows[name].append(line.strip()) 14 print(windows)
9–8. 模块研究. 提取模块的属性资料. 提示用户输入一个模块名(或者从命令行接受输入).然后使用 dir() 和其它内建函数提取模块的属性, 显示它们的名字, 类型, 值.
1 m = input('输入模块名: ') 2 module = __import__(m) 3 ml = dir(module) 4 # print(ml) 5 for i in ml: 6 print('name: ', i) 7 print('type: ', type(getattr(module,i))) 8 print('value: ', getattr(module,i)) 9 print('')
9–9. Python 文档字符串.
进入 Python 标准库所在的目录. 检查每个 .py 文件看是否有__doc__ 字符串, 如果有, 对其格式进行适当的整理归类. 你的程序执行完毕后, 应该会生成一个漂亮的清单. 里边列出哪些模块有文档字符串, 以及文档字符串的内容. 清单最后附上那些没有文档字符串模块的名字.附加题: 提取标准库中各模块内全部类(class)和函数的文档.
1 import os 2 3 pymodules = {} 4 path = input("输入路径：") 5 pyfiles = [f for f in os.listdir(os.path.abspath(path)) if f.endswith('.py')] 6 7 for f in pyfiles: 8 module = f[:-3] 9 pymodules.setdefault(module, '') 10 pyfile = os.path.join(path, f) 11 file = open(pyfile) 12 doc = False 13 for line in file: 14 if line.strip().startswith('"""') and line.strip().endswith('"""'): 15 pymodules[module] += line 16 file.close() 17 break 18 elif line.strip().startswith('"""') or line.strip().startswith('r"""') and len(line) > 3: 19 doc = True 20 pymodules[module] += line 21 continue 22 elif doc: 23 if line == '"""': 24 pymodules[module] += line 25 file.close() 26 doc = False 27 break 28 else: 29 pymodules[module] += line 30 else: 31 continue 32 file.close() 33 34 hasdoc = [] 35 nodoc = [] 36 for module in pymodules: 37 if pymodules[module]: 38 hasdoc.append(module) 39 else: 40 nodoc.append(module) 41 42 print('没有文档模块:') 43 for key in nodoc: 44 print(key) 45 46 print("") 47 48 print("有文档模块:") 49 for key in hasdoc: 50 print("%s：%s" % key, pymodules[key])
9-10.家庭理财。创建一个家庭理财程序。你的程序需要处理储蓄 、支票、金融市场 ，定期存款等多
种账户。为每种账户提供一个菜单操作界面 ，要有存款、取款、借、贷等操作 。另外还要提 一个取消操作选项 。用户退出这个程序时相关数据应该保存到文件里去 （出于备份的目的， 序执行过程中也要备份）。
1 # 函数思路不够用，只好用上类 2 import os 3 import json 4 import sys 5 6 class Fm: 7 8 def deposit(self): 9 """存款""" 10 money = input("存款：") 11 if money == 'q': 12 self.saving() 13 else: 14 self.user_dict['余额'] += float(money) 15 with open('%s.json' % self.ID, 'w') as f: 16 json.dump(self.user_dict, f) 17 print("存款成功") 18 self.saving() 19 20 def withdrawal(self): 21 """取款""" 22 while True: 23 money = input("取款：") 24 if money == 'q': 25 self.saving() 26 else: 27 if float(money) <= self.user_dict['余额']: 28 self.user_dict['余额'] -= float(money) 29 with open('%s.json' % self.ID, 'w') as f: 30 json.dump(self.user_dict, f) 31 print("取款成功") 32 self.saving() 33 else: 34 print("取款金额超出余额，重新输入") 35 36 def loan(self): 37 """借款""" 38 money = input("存款：") 39 if money == 'q': 40 self.saving() 41 else: 42 self.user_dict['借款'] += float(money) 43 with open('%s.json' % self.ID, 'w') as f: 44 json.dump(self.user_dict, f) 45 print("借款成功") 46 self.saving() 47 48 def credit(self): 49 """贷款""" 50 money = input("贷款：") 51 if money == 'q': 52 self.saving() 53 else: 54 self.user_dict['贷款'] += float(money) 55 with open('%s.json' % self.ID, 'w') as f: 56 json.dump(self.user_dict, f) 57 print("贷款成功") 58 self.saving() 59 60 def saving(self): 61 """储蓄主菜单""" 62 showmenu = (""" 63 [1]存款 64 [2]取款 65 [3]借款 66 [4]贷款 67 [5]返回 68 [6]退出 69 70 请输入编号：""") 71 num = input(showmenu) 72 if num == '6': 73 sys.exit() 74 elif num in '12345': 75 if num == '1': self.deposit() 76 if num == '2': self.withdrawal() 77 if num == '3': self.loan() 78 if num == '4': self.credit() 79 if num == '5': self.main() 80 else: 81 print("输入有误") 82 83 def main(self): 84 """主菜单""" 85 showmenu = (""" 86 [1]储蓄 87 [2]支票(未写) 88 [3]金融市场(未写) 89 [4]定期存款(未写) 90 [5]退出 91 92 请输入编号：""") 93 num = input(showmenu) 94 if num == '5': 95 sys.exit() 96 elif num in '1234': 97 if num == '1': self.saving() 98 else: 99 print("输入有误") 100 101 def registered(self): 102 """注册""" 103 while True: 104 ID = input('ID：') 105 filename = '%s.json' % ID 106 if filename in os.listdir(os.getcwd()): 107 print("用户已存在") 108 else: 109 passwd = input('密码：') 110 name = input('姓名：') 111 registered_user = {} 112 registered_user['密码'] = passwd 113 registered_user['姓名'] = name 114 registered_user['余额'] = 0 115 registered_user['借款'] = 0 116 registered_user['贷款'] = 0 117 filename = open('%s.json' % ID, 'w') 118 json.dump(registered_user, filename) 119 filename.close() 120 print("注册成功") 121 self.user() 122 123 def login(self): 124 """登录""" 125 self.ID = input('ID：') 126 filename = '%s.json' % self.ID 127 if filename in os.listdir(os.getcwd()): 128 with open(filename) as user_file: 129 self.user_dict = json.load(user_file) 130 passwd = input("密码：") 131 if self.user_dict['密码'] == passwd: 132 print("登录成功") 133 print("---------------\n" 134 "姓名：{姓名}\n余额：{余额:.2f}\n借款：{借款:.2f}\n贷款：{贷款:.2f}\n" 135 "---------------".format(**self.user_dict)) 136 self.main() 137 else: 138 print("密码错误") 139 else: 140 print("用户不存在") 141 142 def user(self): 143 """账号""" 144 showmenu = (""" 145 [1]注册 146 [2]登录 147 [3]退出 148 149 请输入编号：""") 150 num = input(showmenu) 151 while True: 152 if num == '3': 153 break 154 elif num in '12': 155 if num == '1': self.registered() 156 if num == '2': 157 if self.login(): 158 self.main() 159 else: 160 print("输入有误") 161 162 if __name__ == '__main__': 163 fm = Fm() 164 fm.user()
9–11. Web 站点地址.
a) 编写一个 URL 书签管理程序. 使用基于文本的菜单, 用户可以添加, 修改或者删除书签数据项. 书签数据项中包含站点的名称, URL 地址, 以及一行简单说明(可选). 另外提供检索功能,可以根据检索关键字在站点名称和 URL 两部分查找可能的匹配. 程序退出时把数据保存到一个磁盘文件中去; 再次执行时候加载保存的数据.
b)改进 a) 的解决方案, 把书签输出到一个合法且语法正确的 HTML 文件(.html 或 htm )中,这样用户就可以使用浏览器查看自己的书签清单. 另外提供创建”文件夹”功能, 对相关的书签进行分组管理.
附加题: 请阅读 Python 的 re 模块了解有关正则表达式的资料, 使用正则表达式对用户输入的 URL 进行验证.
1 import sys 2 import re 3 4 regex = re.compile(r"^(((http|https)://)" # http、https 5 r"?([a-z]{2,}))\." # www 6 r"([a-z0-9]+)\." # 7 r"([a-z]{3}|([a-z]{3}\.[a-z]{2,4}))$") #com、com.cn 8 9 filename = "bookmark.txt" 10 def add(): 11 """添加""" 12 urllist = [] 13 while True: 14 url = input("输入地址[b:返回]：").strip() 15 if url == 'b': 16 main() 17 if regex.match(url): 18 urlname = input("输入名称：").strip() 19 description = input("简要说明：").strip() 20 urllist.append(urlname) 21 urllist.append(url) 22 urllist.append(description) 23 with open(filename, 'a+') as f: 24 f.write(str(urllist) + '\n') 25 print('添加成功') 26 else: 27 print("地址不规范") 28 29 def modify(): 30 """修改""" 31 urllist = [] 32 f = open(filename) 33 for i, j in enumerate(f): 34 urllist.append(eval(j)) 35 print(i, j, end='') 36 f.close() 37 38 while True: 39 num = input("修改编号[b:返回]：") 40 if num == 'b': 41 main() 42 if int(num) <= len(urllist): 43 url = input("输入地址：").strip() 44 if regex.match(url): 45 urlname = input("输入名称：").strip() 46 description = input("简要说明：").strip() 47 urllist[int(num)][0] = urlname 48 urllist[int(num)][1] = url 49 urllist[int(num)][2] = description 50 print("修改成功") 51 with open(filename, 'w') as urlfile: 52 for urlline in urllist: 53 urlfile.write(str(urlline) + '\n') 54 else: 55 print("地址不规范") 56 else: 57 print('输入有误') 58 59 60 def delete(): 61 """删除""" 62 urllist = [] 63 f = open(filename) 64 for i, j in enumerate(f): 65 urllist.append(eval(j)) 66 print(i, j, end='') 67 f.close() 68 69 while True: 70 num = input("\n删除编号[b:返回]：") 71 if num == 'b': 72 main() 73 if int(num) <= len(urllist): 74 del urllist[int(num)] 75 print('删除成功') 76 with open(filename, 'w') as urlfile: 77 for urlline in urllist: 78 urlfile.write(str(urlline) + '\n') 79 else: 80 print("输入有误") 81 82 def find(): 83 """查找""" 84 urllist = [] 85 while True: 86 name = input("查找内容[b:返回]：") 87 if name == 'b': 88 main() 89 with open(filename) as f: 90 for urlline in f: 91 urllist.append(eval(urlline)) 92 if name in str(urllist): 93 for urlline in urllist: 94 if name in urlline: 95 print('%s\n名称：%s\n地址：%s\n说明：%s\n%s' % 96 ('-'*20, urlline[0], urlline[1], urlline[2], '-'*20)) 97 else: 98 print('书签不存在') 99 100 def main(): 101 """主菜单""" 102 showmenu = (""" 103 [1]添加书签 104 [2]修改书签 105 [3]删除书签 106 [4]查找书签 107 [5]退出程序 108 109 请输入编号：""") 110 num = input(showmenu) 111 if num == '5': 112 sys.exit() 113 elif num in '1234': 114 if num == '1': add() 115 if num == '2': modify() 116 if num == '3': delete() 117 if num == '4': find() 118 else: 119 print("输入有误") 120 121 if __name__ == '__main__': 122 main()
9-12 用户名和密码。回顾练习7-5，修改代码使之可以支持“上次登录时间”。请参阅time模块中的文档了解如何记录用户上次登录的时间。另外提供一个系统管理员，他可以导出所有用户的用户名，密码（如需要可以加密），以及上次登录时间。
a）数据应保存在磁盘中，使用冒号：分隔，一次写入一行，例如“Joe：boohoo：953176591.145，文件中数据的行数应该等于你系统上的用户数。
b）进一步改进你的程序，不再一次写入一行，而使用pickle模块保存整个数据对象。请参阅pickle模块的文档了解如何序列化/扁平化对象，以及如何读写保存的对象。一般来说，这个解决方案的代码行数要比a）少；
c）使用shelve模块替换pickle模块，由于可以省去一些维护代码，这个解决方案的代码比b）的更少
1 from datetime import datetime 2 import hashlib, time, sys 3 import pickle as p 4 import shelve as s 5 6 db = {} 7 8 def newuser(): 9 """注册""" 10 value = [] 11 prompt = 'login desired: ' 12 while True: 13 name = input(prompt).lower() 14 if not name.isalnum() and '' in name: 15 print('name format error') 16 continue 17 else: 18 if name in db: 19 prompt = 'name taken, try another: ' 20 continue 21 else: 22 break 23 pwd = input('login password desired：') 24 m = hashlib.md5() 25 m.update(pwd.encode(encoding='utf-8')) 26 value.append(m.hexdigest()) 27 value.append(datetime.now()) 28 value.append(time.time()) 29 db[name] = value 30 print('new user is %s, register time is %s' % (name, db[name][1])) 31 32 def olduser(): 33 """登录""" 34 name = input('Username：').lower() 35 pwd = input('Password：') 36 m = hashlib.md5() 37 m.update(pwd.encode(encoding='utf-8')) 38 passwd = db.get(name) 39 if passwd[0] == m.hexdigest(): 40 newtimestamp = time.time() 41 newtime = datetime.now() 42 if newtimestamp - db[name][2] < 14400: 43 print('You already logged in at %s：' % db[name][1]) 44 else: 45 passwd[1] = newtime 46 print('Welcome back %s, login time is %s' % (name, passwd[1])) 47 else: 48 print('Login incorrect') 49 50 def removeuser(): 51 """删除""" 52 print(db) 53 name = input('Input a user name to remove：').lower() 54 if name in db: 55 db.pop(name) 56 print("Done") 57 else: 58 print('Input error') 59 60 def getuser(): 61 """查询""" 62 while True: 63 name = input('login name desired：').lower() 64 if not name.isalnum() and '' in name: 65 print('name format error') 66 continue 67 else: 68 if name not in db: 69 print('User name is not in db') 70 answer = input('register a new user? y/n').lower() 71 if 'y' == answer: 72 newuser() 73 break 74 elif 'n' == answer: 75 break 76 else: 77 print('user name is already in db') 78 olduser() 79 break 80 81 def textfile(): 82 """text""" 83 print(db) 84 f = open('account.txt', 'w') 85 f.write(str(db)) 86 f.close() 87 88 def picklefile(): 89 """pickle""" 90 accountfile = 'pickle.data' 91 f = open(accountfile, 'wb') 92 p.dump(db, f) 93 f.close() 94 95 f = open(accountfile, 'rb') 96 accountdb = p.load(f) 97 print(accountdb) 98 99 def shelvefile(): 100 """shelve""" 101 accountfile = 'shelve.data' 102 accountdb = s.open(accountfile, 'c') 103 accountdb['data'] = db 104 accountdb.close() 105 106 accountdb = s.open(accountfile, 'r') 107 print(accountdb['data']) 108 109 def adminlogin(): 110 """管理员""" 111 while True: 112 name = input('Username：').lower() 113 if not name.isalnum() and '' in name: 114 print('name format error') 115 continue 116 else: 117 pwd = input('Password：') 118 if name == 'root' and pwd == 'root': 119 print('Welcom admin') 120 if len(db) == 0: 121 print('There is nothing you can do') 122 showmenu() 123 else: 124 while True: 125 answer = input('Output all account? y/n').lower() 126 if 'y' == answer: 127 prompt = "(T)ext\n(P)ickle\n(S)helve\n\nEnter choice：" 128 choice = input(prompt).lower() 129 if choice in 'tps': 130 if choice == 't': textfile() 131 if choice == 'p': picklefile() 132 if choice == 's': shelvefile() 133 elif 'n' == answer: 134 print('Bye') 135 showmenu() 136 sys.exit() 137 else: 138 print('User name or password is wrong, input again') 139 140 141 def showmenu(): 142 """功能""" 143 prompt = """ 144 (N)ew User Login 145 (E)xisting User Login 146 (G)et user 147 (R)emove a existing user 148 (A)dmin Login 149 (Q)uit 150 151 Enter choice: """ 152 153 done = False 154 while not done: 155 chosen = False 156 while not chosen: 157 try: 158 choice = input(prompt).strip()[0].lower() 159 except (EOFError, KeyboardInterrupt): 160 choice = 'q' 161 print('\nYou picked: [%s]' % choice) 162 if choice not in 'negraq': 163 print('Invalid option, try again') 164 else: 165 chosen = True 166 167 if choice == 'q': done = True 168 if choice == 'n': newuser() 169 if choice == 'e': olduser() 170 if choice == 'g': getuser() 171 if choice == 'r': removeuser() 172 if choice == 'a': adminlogin() 173 174 if __name__ == '__main__': 175 showmenu()
9-14 记录结果。修改你的计算器程序（练习5-6）使之接受命令行参数。例如$ calc.py 1 + 2 只输出计算结果。另外，把每个表达式和它的结果写入到一个磁盘文件中，当使用下面的命令时 $ calc.py print 会把记录的内容显示到屏幕上，然后重置文件。这里是样例展示：
$ calc.py 1 + 2
3
$ calc.py 3 ^ 3
27
$ calc.py print
1 + 2
3
3 ^ 3
27
$ calc.py print
$
1 import sys, os 2 3 if sys.argv[1] == 'print': 4 if os.path.exists(r'test.txt'): 5 f = open(r'test.txt', 'r') 6 for line in f: 7 print(line) 8 f.close() 9 else: 10 print('No file yet') 11 f = open(r'text.txt', 'w') 12 f.close() 13 else: 14 print(sys.argv[1], sys.argv[2], sys.argv[3]) 15 a, b = sys.argv[1], sys.argv[3] 16 operation = sys.argv[2] 17 expression = sys.argv[1] + '' + sys.argv[2] + '' + sys.argv[3] + os.linesep 18 f = open(r'test.txt', 'a+') 19 f.write(expression) 20 if '+' == operation: 21 print(float(a) + float(b)) 22 result = str(float(a) + float(b)) + os.linesep 23 f.write(result) 24 elif '-' == operation: 25 print(float(a) - float(b)) 26 result = str(float(a) - float(b)) + os.linesep 27 f.write(result) 28 elif '**' == operation: 29 print(float(a) ** float(b)) 30 result = str(float(a) ** float(b)) + os.linesep 31 f.write(result) 32 elif '/' == operation: 33 print(float(a) / float(b)) 34 result = str(float(a) / float(b)) + os.linesep 35 f.write(result) 36 elif '%' == operation: 37 print(float(a) % float(b)) 38 result = str(float(a) % float(b)) + os.linesep 39 f.write(result) 40 elif '*' == operation: 41 print(float(a) - float(b)) 42 result = str(float(a) * float(b)) + os.linesep 43 f.write(result) 44 f.close()
9–15. 复制文件. 提示输入两个文件名(或者使用命令行参数). 把第一个文件的内容复制到第二个文件中去.
1 filename1 = input("输入文件名1：") 2 filename2 = input("输入文件名2：") 3 with open(filename1) as f: 4 filelines = f.readlines() 5 6 with open(filename2, 'w') as f: 7 for fileline in filelines: 8 f.write(fileline)
9–16. 文本处理.
人们输入的文字常常超过屏幕的最大宽度. 编写一个程序, 在一个文本文件中查找长度大于 80 个字符的文本行. 从最接近 80 个字符的单词断行, 把剩余文件插入到下一行处.程序执行完毕后, 应该没有超过 80 个字符的文本行了.
1 filename = input("输入文件名：") 2 f1 = open(filename) 3 f = f1.readlines() 4 f1.close() 5 6 content = [] 7 for line in f: 8 if len(line) > 80: 9 num = list(line) 10 count = len(num) // 80 11 for i in range(count+1): 12 content.append("".join(num[:79]) + '\n') 13 num = num[79:] 14 content.append("".join(num)) 15 else: 16 content.append(line) 17 18 with open(filename, 'w') as f: 19 for fileline in content: 20 f.write(fileline) 21 print("Done")
9-17.文本处理 。创建一个原始的文本文件编辑器 。你的程序应该是菜单驱动的 ，有如下这些选项
1） 创建文件 （提示输入文件名和任意行的文本输入） ：
2 ） 显示文件 （把文件的内容显示到屏幕）：
3） 编辑文件 （提示输入要修改的行 ，然后让用户进行修改） ：
4 ） 保存文件：
5 ） 退出。
1 import sys 2 import os 3 4 def create(): 5 """创建文件""" 6 filename = input('输入文件名[b:返回]：') 7 if filename == 'b': 8 main() 9 while True: 10 text = input('输入内容[b:返回]：') 11 if text == 'b': 12 main() 13 with open(filename, 'a') as f: 14 f.write(text + '\n') 15 print("保存成功") 16 17 def display(): 18 """显示文件""" 19 while True: 20 filename = input('输入文件名[b:返回]：') 21 if filename == 'b': 22 main() 23 with open(filename) as f: 24 for num, line in enumerate(f, 1): 25 print("第%d行:" % num, line, end='') 26 print('读取完毕') 27 28 def edit(): 29 """编辑文件""" 30 filename = input('输入文件名[b:返回]：') 31 if filename == 'b': 32 main() 33 filelist = [] 34 f = open(filename) 35 for num, line in enumerate(f, 1): 36 filelist.append(line) 37 print("第%d行:" % num, line, end='') 38 f.close() 39 print('读取完毕') 40 41 while True: 42 num = input("输入修改的行编号[b:返回]：") 43 if num == 'b': 44 main() 45 if int(num)-1 <= len(filelist): 46 text = input("输入内容：") 47 filelist[int(num)-1] = text + '\n' 48 with open(filename, 'w') as f: 49 for line in filelist: 50 f.write(line) 51 else: 52 print("输入有误") 53 54 def save(): 55 """保存文件""" 56 while True: 57 for_mat = input("查看文件格式(如: .txt)[b:返回]：") 58 if for_mat == 'b': 59 main() 60 if for_mat not in str(os.listdir(os.getcwd())): 61 print("格式不存在") 62 else: 63 for foldername, subfolders, filename in os.walk(os.getcwd()): 64 print("已保存%s文件:\n%s" % (for_mat, '-' * 20)) 65 for file in filename: 66 if file.endswith(for_mat): 67 print(file) 68 print("%s" % '-' * 20) 69 70 def main(): 71 showmenu = (""" 72 [1]创建文件 73 [2]显示文件 74 [3]编辑文件 75 [4]保存文件 76 [5]退出 77 78 请输入编号：""") 79 while True: 80 num = input(showmenu) 81 if num == '5': 82 sys.exit() 83 elif num in '1234': 84 if num == '1': create() 85 if num == '2': display() 86 if num == '3': edit() 87 if num == '4': save() 88 else: 89 print("输入有误") 90 91 if __name__ == '__main__': 92 main()
9–18. 搜索文件. 提示输入一个字节值(0 – 255)和一个文件名. 显示该字符在文件中出现的次数.
1 num = int(input("输入数字(0-255)：")) 2 filename = input("输入文件名：") 3 with open(filename) as f: 4 print(sum(line.count(chr(num)) for line in f))
9–20. 压缩文件.
写一小段代码, 压缩/解压缩 gzip 或 bzip 格式的文件. 可以使用命令行下的 gzip 或 bzip2 以及 GUI 程序 PowerArchiver , StuffIt , 或 WinZip 来确认你的 Python支持这两个库.
1 import gzip 2 3 text = open(r'test.txt', 'rb') 4 gzipfile = gzip.open(r'test.txt.gz', 'wb') 5 gzipfile.writelines(text) 6 text.close() 7 gzipfile.close() 8 9 gzipfile = gzip.open(r'test.txt.gz', 'rb') 10 text = open(r'test1.txt', 'wb') 11 content = gzipfile.read() 12 text.write(content) 13 gzipfile.close() 14 text.close()
9–21. ZIP 归档文件.
创建一个程序, 可以往 ZIP 归档文件加入文件, 或从中提取文件,有可能的话, 加入创建ZIP 归档文件的功能.
1 import zipfile, sys 2 3 def archive_add(): 4 """归档、添加""" 5 filename = input("压缩文件名[b:返回]：") 6 if filename == 'b': 7 main() 8 sys.exit() 9 while True: 10 file = input("文件名[b:返回]：") 11 if file == 'b': 12 archive_add() 13 filezip = zipfile.ZipFile(filename, 'a') 14 filezip.write(file) 15 filezip.close() 16 print('Done') 17 18 def extract(): 19 """提取""" 20 filename = input("压缩文件名[b:返回]：") 21 if filename == 'b': 22 main() 23 sys.exit() 24 filezip = zipfile.ZipFile(filename) 25 print(filezip.namelist()) 26 filezip.close() 27 while True: 28 file = input("文件名[b:返回]：") 29 if file == 'b': 30 extract() 31 filezip = zipfile.ZipFile(filename) 32 filezip.extract(file) 33 filezip.close() 34 print('Done') 35 36 def main(): 37 menu = """ 38 [1]归档、添加 39 [2]提取 40 [3]退出 41 42 请输入编号：""" 43 44 while True: 45 num = input(menu) 46 if num == '3': 47 break 48 elif num in '12': 49 if num == '1': archive_add() 50 if num == '2': extract() 51 else: 52 print('输入有误') 53 54 if __name__ == '__main__': 55 main()
9–22. ZIP 归档文件.
unzip -l 命令显示出的 ZIP 归档文件很无趣. 创建一个 Python脚本 lszip.py , 使它可以显示额外信息: 压缩文件大小, 每个文件的压缩比率(通过比较压缩前后文件大小), 以及完成的 time.ctime() 时间戳, 而不是只有日期和 HH:MM .
提示: 归档文件的 date_time 属性并不完整, 无法提供给 time.mktime() 使用….这由你自己决定.
1 import zipfile, time, os 2 3 filename = input("Zip file name：") 4 print('Zip file size：%d bytes' % (os.stat(filename)).st_size) 5 z = zipfile.ZipFile(filename) 6 print('filename\t\t\tdatetime\t\t\tsize\tcompress size\trate') 7 for info in z.infolist(): 8 t = time.ctime(time.mktime(tuple(list(info.date_time) + [0, 0, 0]))) 9 print('%8s\t%s\t%4d\t\t%4d\t\t%.2f%%' % (info.filename, t, info.file_size, info.compress_size, 10 float(info.compress_size / info.file_size * 100) )) 11 z.close()
9–23. TAR 归档文件.
为 TAR 归档文件建立类似上个问题的程序. 这两种文件的不同之处在于 ZIP 文件通常是压缩的, 而 TAR 文件不是, 只是在 gzip 和 bzip2 的支持下才能完成压缩工作. 加入任意一种压缩格式支持.附加题: 同时支持 gzip 和 bzip2 .
1 import tarfile, os 2 3 def compressfile(tarname, filename1, filename2): 4 t = tarfile.open(tarname, 'w:gz') #w.bz2 5 t.add(filename1) 6 t.add(filename2) 7 t.close() 8 9 def extractfile(tarname): 10 t = tarfile.open(tarname, 'r') 11 t.extractall(os.getcwd()) 12 t.close() 13 14 if __name__ == '__main__': 15 compressfile(r'abc.tar.gz', r'a.txt', r'b.txt')#r'abc.tar.bz2' 16 extractfile(r'abc.tar.gz')#r'abc.tar.bz2'
9–24. 归档文件转换.
参考前两个问题的解决方案, 写一个程序, 在 ZIP (.zip) 和TAR/gzip (.tgz/.tar.gz) 或 TAR/bzip2 (.tbz/.tar.bz2) 归档文件间移动文件. 文件可能是已经存在的, 必要时请创建文件.
1 import zipfile, tarfile, os 2 3 def movefile(compressfile1, compressfile2, file): 4 5 if compressfile1.endswith('.zip') and compressfile2.endswith(('.tar.gz', '.tgz', '.tbz', '.tar.bz2')): 6 z = zipfile.ZipFile(compressfile1, 'a') 7 if file not in z.namelist(): 8 f = open(file, 'w') 9 f.close() 10 z.write(file) 11 z.extract(file) 12 else: 13 z.extract(file) 14 z.close() 15 t = tarfile.open(compressfile2) 16 ls = t.getnames() 17 if file not in ls: 18 t.extractall() 19 t.close() 20 mode = 'w:gz' if compressfile2.endswith(('tar.gz', 'tgz')) else 'w:bz2' 21 t = tarfile.open(compressfile2, mode) 22 for name in ls + [file]: 23 t.add(name) 24 t.close() 25 os.remove(file) 26 t.close() 27 28 elif compressfile1.endswith(('.tar.gz', '.tgz', '.tbz', '.tar.bz2')) and compressfile2.endswith('.zip'): 29 t = tarfile.open(compressfile1) 30 if file not in t.getnames(): 31 f = open(file, 'w') 32 f.close() 33 else: 34 t.extract(file) 35 t.close() 36 z = zipfile.ZipFile(compressfile2, 'a') 37 if file not in z.namelist(): 38 z.write(file) 39 z.close() 40 os.remove(file) 41 42 if __name__ == '__main__': 43 compressfile1 = input('压缩文件名1：') 44 compressfile2 = input('压缩文件名2：') 45 file = input('文件名：') 46 movefile(compressfile1, compressfile2, file)
9–25. 通用解压程序.
创建一个程序, 接受任意数目的归档文件以及一个目标目录做为参数.归档文件格式可以是 .zip, .tgz, .tar.gz, .gz, .bz2, .tar.bz2, .tbz 中的一种或几种. 程序会把第一个归档文件解压后放入目标目录, 把其它归档文件解压后放入以对应文件名命名的目录下(不包括扩展名). 例如输入的文件名为 header.txt.gz 和 data.tgz ， 目录为 incoming ,header.txt 会被解压到 incoming 而 data.tgz 中的文件会被放入 incoming/data .
1 import zipfile, tarfile, gzip, bz2, os 2 3 def extract(dir1, dir2): 4 5 filelist = os.listdir(dir1) 6 if filelist[0].endswith(('.tar.gz', '.tgz', '.tbz', '.tar.bz2')): 7 t = tarfile.open(os.path.join(dir1, filelist[0])) 8 t.extractall(dir2) 9 t.close() 10 11 elif filelist[0].endswith('.gz'): 12 g = gzip.open(os.path.join(dir1, filelist[0]), 'rb') 13 ug = open(os.path.join(dir2, os.path.splitext(filelist[0])[0]), 'wb') 14 data = g.read() 15 ug.write(data) 16 ug.close() 17 g.close() 18 19 elif filelist[0].endswith('.bz2'): 20 b = bz2.BZ2File(os.path.join(dir1, filelist[0])) 21 ub = open(os.path.join(dir2, os.path.splitext(filelist[0])[0]), 'wb') 22 data = b.read() 23 ub.write(data) 24 ub.close() 25 b.close() 26 27 elif filelist[0].endswith('.zip'): 28 z = zipfile.ZipFile(os.path.join(dir1, filelist[0])) 29 z.extractall(dir2) 30 z.close() 31 32 filelist.remove(filelist[0]) 33 34 for file in filelist: 35 36 dirname = os.path.splitext(file)[0] 37 38 if dirname in os.listdir(dir2): 39 dirname = os.path.join(dir2, dirname, str(filelist.index(file))) 40 else: 41 dirname = os.path.join(dir2, dirname) 42 os.makedirs(dirname) 43 44 if file.endswith(('.tar.gz', '.tgz', '.tbz', '.tar.bz2')): 45 t = tarfile.open(os.path.join(dir1, file)) 46 t.extractall(dirname) 47 t.close() 48 49 elif file.endswith('.gz'): 50 g = gzip.open(os.path.join(dir1, file), 'rb') 51 ug = open(os.path.join(dirname, os.path.splitext(file)[0]), 'wb') 52 data = g.read() 53 ug.write(data) 54 ug.close() 55 g.close() 56 57 elif file.endswith('.bz2'): 58 b = bz2.BZ2File(os.path.join(dir1, file)) 59 ub = open(os.path.join(dirname, os.path.splitext(file)[0]), 'wb') 60 data = b.read() 61 ub.write(data) 62 ub.close() 63 b.close() 64 65 elif file.endswith('.zip'): 66 z = zipfile.ZipFile(os.path.join(dir1, file)) 67 z.extractall(dirname) 68 z.close() 69 70 if __name__ == '__main__': 71 dir1 = os.path.abspath(input('Path1: ')) 72 dir2 = os.path.abspath(input('Path2: ')) 73 extract(dir1, dir2)
