importosimportpwdfromlogilab.common.pytestimportPyTesterdefgetlogin():"""avoid usinng os.getlogin() because of strange tty / stdin problems (man 3 getlogin) Another solution would be to use $LOGNAME, $USER or $USERNAME """returnpwd.getpwuid(os.getuid())[0]defupdate_parser(parser):login=getlogin()parser.add_option('-r','--rebuild-database',dest='rebuild_db',default=False,action="store_true",help="remove tmpdb and rebuilds the test database")parser.add_option('-u','--dbuser',dest='dbuser',action='store',default=login,help="database user")parser.add_option('-w','--dbpassword',dest='dbpassword',action='store',default=login,help="database name")parser.add_option('-n','--dbname',dest='dbname',action='store',default=None,help="database name")parser.add_option('--euser',dest='euser',action='store',default=login,help="esuer name")parser.add_option('--epassword',dest='epassword',action='store',default=login,help="euser's password' name")returnparserclassCustomPyTester(PyTester):def__init__(self,cvg,options):super(CustomPyTester,self).__init__(cvg,options)ifoptions.rebuild_db:os.unlink('tmpdb')os.unlink('tmpdb-template')