python 自动创建vsftp

[root@js-93 vsftp_install]# cat install_vsftp.py 
#!/usr/bin/env python3
#coding:utf-8
#install vsftp service

import os,sys,subprocess
import getpass
def install_vsftp():
    check_open = os.system('rpm -qa|grep openssh-server >/dev/null 2>&1')
    check_vsftp = os.system('rpm -qa|grep vsftp >/dev/null 2>&1')
    if check_open != 0 :
        check_open_second = os.system('yum install openssh-server -y >/dev/null 2>&1')
        if check_open_second != 0:
            print('Error: install openssh-server faild ,please check your system')
        else:
            print('openssh-server install successful')
    else:
        print('openssh-server already installed')
    if check_vsftp != 0:
        check_vsftp_second = os.system('yum install vsftpd -y >/dev/null 2>&1')
        if check_vsftp_second != 0:
            print('Error: install vsftp faild, please check your system')
        else:
            print('vsftpd install successful!')
    else:
        print('vsftp already successful')

def create_vsftp_conf():
    conf_dict = {
    'ftp_username':'nobody',
    'anonymous_enable':'YES',
    'local_enable':'YES',
    'write_enable':'YES',
    'local_umask':'022',
    'dirmessage_enable':'YES',
    'xferlog_enable':'YES',
    'connect_from_port_20':'YES',
    'xferlog_std_format':'YES',
    'chroot_local_user':'NO',
    'chroot_list_enable':'YES',
    'chroot_list_file':'/etc/vsftpd/chroot_list',
    'listen':'YES',
    'pam_service_name':'vsftpd',
    'userlist_enable':'YES',
    'tcp_wrappers':'YES',
    'anon_world_readable_only':'YES',
    'pasv_enable':'YES',
    'pasv_min_port':'10000',
    'pasv_max_port':'10010'}
    with open('/etc/vsftpd/vsftpd.conf','w') as conf_obj: 
        for key,val in conf_dict.items():
            line = key+"="+val
            conf_obj.write(line)
            conf_obj.write('\n')


def create_ftp_user(user,pwd,user_passwd):
    res = subprocess.Popen(['ls -d %s' % pwd] ,shell=True,stdin=subprocess.PIPE,stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout_val = res.stdout.read()
    stderr_val = res.stderr.read()
    if not stdout_val:
        print('Warning: "%s" 目录不存在! '% pwd)
        return
    else:
        if len(user) == 0 or len(user_passwd) == 0:
            print('用户和密码不能为空')
            return
        else:
            create_ftp_user = os.system('useradd -s /sbin/nologin -d %s %s' %(pwd,user))
            create_passwd = os.system('echo %s|passwd --stdin %s' %(user_passwd,user) )
            with open('/etc/vsftpd/chroot_list','a+') as chroot_list_obj:
                chroot_list_obj.write(user)
                chroot_list_obj.write('\n')
                os.system('/etc/init.d/vsftpd restart')

def create_vsftp_user(user,pwd,user_passwd):
    res = subprocess.Popen(['ls -d %s' % pwd] ,shell=True,stdin=subprocess.PIPE,stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout_val = res.stdout.read()
    if not stdout_val:
        print('Warning: "%s" 目录不存在! '% pwd)
        return
    else:
        if len(user) == 0 or len(user_passwd) == 0:
            print('用户和密码不能为空')
            return
        else:
            create_vsftp_user = os.system('useradd  -d %s %s' %(pwd,user))
            create_passwd = os.system('echo %s|passwd --stdin %s' %(user_passwd,user) )
            with open('/etc/vsftpd/chroot_list','a+') as chroot_list_obj:
                chroot_list_obj.write(user)
                chroot_list_obj.write('\n')
                os.system('/etc/init.d/vsftpd restart')

def del_user(user):
    res = subprocess.Popen(['id %s' % user] ,shell=True,stdin=subprocess.PIPE,stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout_val = res.stdout.read()
    if len(stdout_val) == 0:
        print('您删除的用户不存在')
    else:
        print('''
删除用户时，将会删除该用户的Home目录，请确认是否删除
删除：yes
不删除：NO''')
        inp = input("请输入您的选择：").strip()
        if inp == 'yes':
            print("即将删除用户 %s" % user)
            os.system('userdel -r %s' % user)
        else:
            print("撤销删除用户")
            return

if __name__ == '__main__':
    while True:
        print('''
+++++++++++++++++++++++++
请选择：                +
1.新建vsftp服务器       +    
2.创建ftp用户           +
3.创建vsftp用户         +
4.删除用户              +
5.退出                  +
+++++++++++++++++++++++++''')
        your_choice = input("请选择编号：").strip()
        if your_choice == '1':
            install_vsftp()
            create_vsftp_conf()
            continue
        if your_choice == '2':
            user = input('请输入用户名：').strip()
            pwd = input('请输入 Home 目录：').strip()
            user_passwd = getpass.getpass('请输入用户密码:')
            create_ftp_user(user,pwd,user_passwd)
            continue
        if your_choice == '3':
            user = input('请输入用户名：').strip()
            pwd = input('请输入 Home 目录：').strip()
            user_passwd = getpass.getpass('请输入用户密码:')
            create_vsftp_user(user,pwd,user_passwd)
            continue
        if your_choice == '4':
            user = input('请输入将要删除的用户名：').strip()
            del_user(user)
        if your_choice == '5':
            print('欢迎下次使用')
            break