Как я могу убедиться, что пользователь вводит строку, содержащую только числа, операторы и букву Q?

Question

Кажется, я не могу найти, как ограничить пользователя использованием реальных чисел и буквы Q в качестве переменной. (начало обучения Python этим утром)

astr = 0 
while True: 
     astringfunctionMB = raw_input('please enter MB equation in terms of Q e.i 95-.5*Q : ') 
     if (float(astringfunctionMB) or "Q" in astringfunctionMB): 
       print "Please try again" 
       #else if 
     else: 
      break 
      astr = astringfunctionMB 

Answer 1

Существует несколько способов сделать это; Я бы либо использовать regular expression:

import re 

... 

if re.match(r'^[\dQ+\-*/.]$', astringfunctionMB): 
    ... 

(см demo для тестирования) или создать set допустимых символов для проверки против:

import string 

VALID_CHARS = set(string.digits + 'Q+-*/.') 

... 

if astringfunctionMB and all(char in VALID_CHARS for char in astringfunctionMB): 
    ... 

Заметим, однако, что это не обязательно означает, что вход пользователя является допустимым выражением, так как он содержит только допустимые символы. Для более надежной проверки вам нужно будет написать что-то, чтобы проанализировать их ввод в соответствующую форму и посмотреть, можно ли запускать их на том, что они набирают.

Answer 2

Предполагая, что вам нужно выражение питона в aStringFunctionMB вы можете оценить его в контролируемом пространстве имен исключений и поймать:

try: 
eval(aStringFunctionMB, {}, {"Q": 1}) 
astr = aStringFunctionMB 
except (NameError, SyntaxError): 
print "Please try again"