#!/usr/bin/env python

import os
import sys


def addNis ( stem ):
  sysFile = '/etc/%s'     % stem
  tmpFile = '/etc/%s.tmp' % stem
  nisFile = '/etc/%s.nis' % stem

  fdSys    = open( sysFile )
  fdSysTmp = open( tmpFile, 'w' )
  
  for entry in fdSys.readlines():
    if entry[0] != '+':
      print entry[:-1]
      fdSysTmp.write( entry )
  
  if os.path.isfile( nisFile ):
    print 'Adding NIS entries.'
    fdSysNis = open( nisFile )
  
    for entry in fdSysNis.readlines():
      if entry[0] == '+':
        print entry[:-1]
        fdSysTmp.write( entry )
  
    fdSysNis.close()
  
  fdSysTmp.close()
  fdSys.close()

  if stem == 'shadow': os.chmod( tmpFile, 0 )

  os.rename( tmpFile, sysFile )

  return
    

if __name__ == '__main__':

   addNis( 'passwd' )
   addNis( 'shadow' )

   sys.exit( 0 )