Top

NSD Python1 DAY03

  1. 案例1：模拟cp操作
  2. 案例2：生成随机密码

1 案例1：模拟cp操作

1.1 问题

创建cp.py文件，实现以下目标：

  1. 将/bin/ls“拷贝”到/root/目录下
  2. 不要修改原始文件

1.2 方案

获取用户原文件名和新文件名，打开原文件，打开新文件，从打开的原文件中读取数据，写入到打开的新文件中，关闭两个文件

cp代码的过程中，需要注意的部分在于：

如果一个文件过大，你将无法直接读取数据到内存，此时，使用while循环语句，分次读取数据，每次读4096字节，读取数据为空时，结束循环，将数据写入到目标文件

1.3 步骤

实现此案例需要按照如下步骤进行。

步骤一：编写脚本

  1.  [root@localhost day03]# vim cp.py 
  2. #!/usr/bin/env python3
  3.  
  4. f1 = open('/bin/ls', 'rb')
  5. f2 = open('/root/ls', 'wb')
  6.  
  7. data = f1.read()
  8. f2.write(data)
  9.  
  10. f1.close()
  11. f2.close()

或将上面的代码改为以下写法：

循环读取文件中数据，避免读取数据过大

  1. [root@localhost day03]# vim cp2.py 
  2. #!/usr/bin/env python3
  3.  
  4. src_fname = '/bin/ls'
  5. dst_fname = '/tmp/ls'
  6.  
  7. src_fobj = open(src_fname, 'rb')
  8. dst_fobj = open(dst_fname, 'wb')
  9.  
  10. while True:
  11.     data = src_fobj.read(4096)   # 每次读4096字节
  12.     if data == b'':              # 读不到数据意味着读写完毕，中断循环
  13.         break
  14.     dst_fobj.write(data)         # 将数据写到目标文件
  15.  
  16. src_fobj.close()
  17. dst_fobj.close()

或将上面的代码改为以下写法：

With打开文件读取数据或写入数据后，文件会直接关闭

  1. [root@localhost day03]# vim cp3.py 
  2. #!/usr/bin/env python3
  3.  
  4. src_fname = '/bin/ls'
  5. dst_fname = '/root/ls'
  6.  
  7. with open(src_fname, 'rb') as src_fobj:
  8.     with open(dst_fname, 'wb') as dst_fobj:
  9.         while True:
  10.             data = src_fobj.read(4096)
  11.             if not data:
  12.                 break
  13.             dst_fobj.write(data)

或将上面的代码改为以下写法：

sys.argv方法表示空列表，执行脚本时输入命令： python3 cp_func.py /bin/ls /root/ls，表示sys.argv=[cp_func.py，‘/bin/ls’，‘/root/ls’]，所以，调用copy函数时，列表切片方式获取实参为（‘/bin/ls’，‘/root/ls’）

  1. [root@localhost day03]# vim cp_func.py 
  2. #!/usr/bin/env python3
  3.  
  4. import sys
  5.  
  6. def copy(src_fname, dst_fname):
  7.     src_fobj = open(src_fname, 'rb')
  8.     dst_fobj = open(dst_fname, 'wb')
  9.  
  10.     while True:
  11.         data = src_fobj.read(4096)
  12.         if not data:
  13.             break
  14.         dst_fobj.write(data)
  15.  
  16.     src_fobj.close()
  17.     dst_fobj.close()
  18.  
  19. copy(sys.argv[1], sys.argv[2])

步骤二：测试脚本执行

  1. [root@localhost day03]# python3 cp.py
  2. [root@localhost day03]# cd /root
  3. [root@localhost ~]# ls
  4. core  ls
  5.  
  6. [root@localhost day03]# python3 cp2.py
  7. [root@localhost day03]# cd /root
  8. [root@localhost ~]# ls
  9. core  ls
  10.  
  11. [root@localhost day03]# python3 cp3.py
  12. [root@localhost day03]# cd /root
  13. [root@localhost ~]# ls
  14. core  ls
  15.  
  16. [root@localhost day03]# python3 cp_func.py /bin/ls /root/ls
  17. [root@localhost day03]# cd /root
  18. [root@localhost ~]# ls
  19. core  ls

2 案例2：生成随机密码

2.1 问题

创建randpass.py脚本，要求如下：

  1. 编写一个能生成8位随机密码的程序
  2. 使用random的choice函数随机取出字符
  3. 改进程序，用户可以自己决定生成多少位的密码

2.2 方案

导入random模块，通过random静态对象调用choice()方法，从自定义字符串all_chs中获取随机项，将获取到的随机字符ch与原result值进行拼接，将最终字符串结果返回给函数，for循环每循环一次获取一个随机字符，密码位数由循环次数决定，循环次数由传递参数值决定。

此程序需要注意的部分在于：

1.导入String模块，其中ascii_letters是生成所有字母，从a-z和A-Z，digits是生成所有数字0-9

2.将整个生成随机密码的代码封装进gen_pass()函数中，当模块文件直接执行时，调用函数即可输出结果

3.参数传递问题：调用函数无实参时，函数调用默认参数，有实参时，函数调用实际参数

2.3 步骤

实现此案例需要按照如下步骤进行。

步骤一：编写脚本

  1. [root@localhost day03]# vim randpass.py
  2. #!/usr/bin/env python3
  3. import random
  4. import string
  5.  
  6. all_chs = string.digits + string.ascii_letters
  7.  
  8. def gen_pass(n=8):
  9.     result = ''
  10.  
  11.     for i in range(n):
  12.         ch = random.choice(all_chs)
  13.         result += ch
  14.     return result
  15.  
  16. if __name__ == '__main__':
  17.     print(gen_pass())
  18.     print(gen_pass(4))

或将上面的代码改为以下写法：

利用列表推导式更简洁输出数据

  1. [root@localhost day03]# vim randpass2.py
  2. #!/usr/bin/env python3
  3.  
  4. from random import choice
  5. from string import ascii_letters, digits
  6.  
  7. all_chs = ascii_letters + digits
  8.  
  9.  
  10. def randpass(n=8):
  11.     result = [choice(all_chs) for i in range(n)]
  12.     return ''.join(result)  # 将列表的字符拼接起来
  13.  
  14.  
  15. if __name__ == '__main__':
  16.     print(randpass())
  17.     print(randpass(4))

步骤二：测试脚本执行

  1. [root@localhost day03]# python3 randpass.py
  2. 82wi2gOP
  3. XzMi
  4. [root@localhost day03]# python3 randpass.py
  5. 5wMoDEgC
  6. BDpc
  7. [root@localhost day03]# python3 randpass.py
  8. Ige2VGod
  9. Az0z
  10. [root@localhost day03]# python3 randpass2.py 
  11. eajAocMH
  12. edW1