summaryrefslogtreecommitdiffstats
path: root/vstf/vstf/common/input.py
blob: 14ec8552d05cc33728cc9ee5f64df7d8062691b7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
##############################################################################
# Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################

import os


def raw_choice(title):
    ret = {'Y': True, 'N': False}
    while True:
        os.system("clear")
        in_str = "\n%s:(y|n)  " % title
        uin = raw_input(in_str).title()
        if uin in ['Y', 'N']:
            break
    return ret[uin]