testsuites/vstf/vstf_scripts/vstf/common/input.py

   1 ##############################################################################
   2 # Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9
  10 import os
  11
  12
  13 def raw_choice(title):
  14     ret = {'Y': True, 'N': False}
  15     while True:
  16         os.system("clear")
  17         in_str = "\n%s:(y|n)  " % title
  18         uin = raw_input(in_str).title()
  19         if uin in ['Y', 'N']:
  20             break
  21     return ret[uin]