I started off by stating the version number, date it was finished, and that I created it.

# Version Eins Null = May 4, 2018.
# Created by William "Vulcan" McDonald

The first block is where I take my imports. I import the sunprocess package for OS control, os for path, and time for sleep.

# Import Subprocess package.
import subprocess
from supbrocess import call
from subprocess import Popen
from subprocess import PIPE as pipe

# Import OS Package
import os
from os import path

# Import Time Package
import time
from time import sleep

Now I need to let the user know that they need a user list, and inform them that the list of users will be overwritten. This way, they know to include all users in the list of users.

# Let user know that they need a user list.
print('For me to work, you need to make the list of users you would like to add.')
input('Press Enter if yo understand...')

# Inform user that the list of users will be overwritten.
print('Make sure this list has ALL the users you want access,')
print('because it will OVERWRITE all the current users.')
input('Press Enter if you understand...')

I will then start the loop that locates the file.

while True:
  try:

I use a try except block in case the user inputs the file incorrectly. If they enter it in incorrectly, I give them a change to enter it in again. The next part is to retrieve the user list, read the lines in, and then break from the loop.

    # Retrieve user list
    userlistFile = raw_input('Where is your user list?')
    f = open(userlistFile, 'r')
    print('Reading lines...')
    users = f.readlines()
    f.close()
    print('Lines read in.')
    break

If I receive a FileNotFoundError, I enter an exception where it notifies the user and then rounds the loop

    # If file does not exist, notify user and try again.
  escept FileNotFoundError:
    print('File ' + userlistFile + ' does not exist.')

Now that I know where the user's userlist is, I can begin the installation, but first I need to define the starting variables.

# Define starting Variables
filestring = ""
ftpstring = ''
useruser = ''
username = ''
           
config = '/etc/vsftpd.conf'
           
head = "Modified by FORGE"
headnotinline = True

I start an infinite loop that will install VSFTPD if it is not already installed.

while True:

I next determine if the vsftpd config file exists. If it does exist, I check to see if I have edited it before.

# Does config file exist:
  if (path.isfile(config)):
    print('Reading config file lines in...')
    f = open(config, 'r')
    lines = f.readlines()
    f.close()
           
    # Is head not in the file?
    for line in lines:
      if (head in line):
        headnotinline = False

Now that I know whether or not the head is in the file, I start an if statement to use this information accordingly.

    If head is not in the file, copy my file in.
    if (headnotinline):
      call(['cp', file, file + '.bak'])
      call(['cp', 'myconf.conf', file])
           
      # Copy over your user list file.
      call(['cp', userlistFile, '/etc/vsftpd.userlist'])
    break

If the config file does not exist, install vsfptd.

  else:
    call(['apt-get', 'install', 'vsftpd', '-y'])
    sleep(5)

I now need to open the Firewall.

call(['ufw', 'enable'])
call(['ufw', 'status'])
           
call(['ufw', 'allow', '20/tcp'])
call(['ufw', 'allow', '21/tcp'])
call(['ufw', 'allow', '990/tcp'])
call(['ufw', 'allow', '40000:50000/tcp'])
           
call(['ufw', 'status'])

I now start the creation of the users.

for user in users:
  user = user.strip('\n')
  print('Starting ' + user + '.')

I initialize the variables based upon the user.

  print('Setting ftpstring')
  ftpstring = '/home/' + user + '/ftp/'
  print('ftpstring has been set.')
           
  print('Setting filestring.')
  filestring = ftpstring + 'files/'
  print('filestring has been set.')
           
print('Setting useruser.')
useruser = user + ':' + users
print('useruser has been set.')
           

I then need to determine if the user already exists on the server.

  print("Determining user's existence.")
  username = Popen(['grep', '-c', '^' + user, '/etc/passwd'], stdout=pipe)
  username = str(username.stdout.read())
  username = username[:-3]
  username = username.strip('b')
  username = username.strip("'")

If the user doesn't exist, create the user and add the user's group just in case it doesn't when user is created.

  if (username = '0'):
    print('User does not exist.')
           
    print('Creating user.')
    call(['adduser', '--gecos', '-', '--disabled-password', user])
    print('User has been created.')
           
    print("Adding user's group.")
    call(['groupadd', '-f', user])
    print("Added user's group.")

If the user already exists, inform the user and move on.

  else:
    print(user + ' already exists.')

Change the user's password to the default as a precaution.

  print('Changing password to default: ' + user + '.')
  passwordIN = Popen(['echo', user + '\n' + user], stdout=pipe)
  passwordOut = Popen(['passwd', user], stdin=passwordIN.stdout)
  passwordIN.stdout.close()
  print('Password set.')

Then I need to change the permissions and set up the directory for each of the users.

  print('Setting up permissions and directory.')
  call(['mkdir', '-p', filestring])
  call(['chown', 'nobody:nogroup', ftpstring])
  call(['chmod', 'a-w', ftpstring])
  call(['chown', useruser, filestring])
  print('Permissions and directory set up.')

I finish by setting up the test text file and print a new line to differentiate between the debug screen for users.

  print('Setting up test text file for user.')
  sub1 = Popen(['echo', 'test'], stdout=pipe)
  sub2 = Popen(['tee', '-a', filestring + 'test.txt'], stdin=sub1.stdout)
  sub1.stdout.close()
  print('Test text file has been set up.')
  print("")