import string

def print_menu():
   print '1. Press 1 to (i)mport the namelist.txt file'
   print '2. Press 2 to (g)enerate usernames'
   print '3. Press 3 to (p)rint new usernames.txt file'
   print '4. Press 4 to E(x)it the Program'
numbers = {}
menu_choice = 0
print_menu()
while menu_choice != 4:
   menu_choice = input("Type in a number (1-4):")
   if menu_choice == 1:
      #Open the file "namelist.txt"
      in_file = open("/store/studentlist.txt","r")
      #variable "textfile" gets filled with info from namelist.txt
      textfile = in_file.readlines()
      #Close the file "namelist.txt"
      in_file.close()
      
   elif menu_choice == 2:     
      #The Loopy Loopy
      namenum = 0
      out_file = open("/store/usernames.txt","w")
      print namenum
      while namenum < len(textfile):
         #Manipulate the namelist into pieces we can handle
         mystring = textfile[namenum]
         newlist = mystring.split()
         lname = newlist[0]
         fname = newlist[1]
         fname = fname[:2]
         lname = lname[:4]
         #Concatenate and Write to File
         out_file.write("create:"+str(lname)+str(fname)+":"+str(lname)+str(fname)+"::505:"+str(newlist)+":/home/"+lname+fname+":/bin/bash:::::"+"\n")
         print lname+fname+" has been written to file." 
         namenum = namenum+1
         print namenum
      out_file.close()
      namenum = 0
   elif menu_choice == 3:
      #Gimme some output baby!
      in_file = open("/store/usernames.txt","r")
      funkytext = in_file.readlines()
      in_file.close()
      print funkytext
   elif menu_choice == 4:
      pass
   else:
      print_menu()
print "Goodbye!"