python3 ubuntu 自动挂载u盘

Gin_Q

1. #!/usr/bin/env python3
   
2. #coding=utf-8
   
3. 
   
4. import os
   
5. import re
   
6. import shutil
   
7. 
   
8. def mount_u():
   
9.     '''
   
10.     # mount U Disk to /mnt
    
11.     '''
    
12.     os.system('echo "\033[32mMount U Disk To /mnt\033[0m"')
    
13.     if not os.path.isdir(r'/mnt'): os.mkdir(r'/mnt')
    
14.     #check u disk is mount to mnt
    
15.     if os.system('mountpoint -q /mnt') == 0:
    
16.         print('/mnt is exist, umount /mnt')
    
17.         os.system('umount /mnt')
    
18.     fdisk = os.popen('fdisk -l')
    
19.     mode = re.compile(r'/dev/sd[b-z]1')
    
20.     u_path = re.search(mode, fdisk.read())
    
21.     if u_path:
    
22.         print('U Disk: {}'.format(u_path.group()))
    
23.         if os.system('mount ' + u_path.group() + ' /mnt') == 0:
    
24.             print('U Disk mounted successfully')
    
25.         else:
    
26.             print('U Disk mounting failed')
    
27.     else:
    
28.         print('U Disk: {}'.format('NO Find!'))
    
29. 
    
30. if __name__ == '__main__':
    
31.     mount_u()
    

复制代码