Мениджър на контекста за валидиране на данни

Опитвам се да обмисля добро решение за това и нищо не ми идва наум. Като упражнение се опитвам да създам контекстен мениджър, който ще обработва валидирането на данни, нещо като:

validation = lambda x: len(x) <= 10

with validator(validation):
    some_data = input("Please enter a name of 10 characters or less: ")

print(some_data)

# OUTPUT
>> Please enter a name of 10 characters or less: FooBarSpamEggs
>> Please enter a name of 10 characters of less: Adam
Adam

Първоначално мислех да направя това с unittest.mock.patch, но разбрах, че не мога да извикам оригиналната функция, докато е коригирана, напр.:

def patched(validation, *args):
    while True:
        p = __builtins__.input(args) # Doesn't work
        if validation(p):
            break
    return p

with unittest.mock.patch('builtins.input', patched):
    input("Some prompt here: ")
# fails on recursion error as patched calls itself

Тогава обмислих да напиша декоратор за валидиране на един ред, но това наистина е полезно само ако можете да направите нещо като:

@validate(lambda x: int(x) == 6)
p = input("How many sides does a d6 have? ")
# can't decorate a function call

Обаче съм пристрастен към тази идея за управление на контекста. За съжаление не знам дали контекстният мениджър има достъп до съдържанието си или е ограничен само до аргументите си. някакви мисли?

Като настрана, знам, че мога да направя тази функционалност във функция, например:

def validate_input(prompt, validation, msg_if_fail=None):
    while True:
        p = input(prompt)
        if validation(p):
            break
        if msg_if_fail is not None:
            print(msg_if_fail)
    return p

Но не е толкова красиво. Това, както казах, е повече упражнение, отколкото практически проблем.


python validation contextmanager
person Adam Smith    schedule 11.09.2014    source източник

Отговори (2)


arrow_upward
2
arrow_downward

Можете да използвате обикновен контекстен мениджър, за да обвиете unittest.mock.patch и да запазите препратка към оригиналната input функция, преди да я коригирате. След това можете да предадете оригиналния input към вашата patched функция:

import unittest.mock
import contextlib
from functools import partial

def patched(validation, orig_input, *args):
    while True:
        p = orig_input(*args)
        if validation(p):
            break
    return p

@contextlib.contextmanager
def validator(validate_func):
    func = partial(patched, validate_func, input)  # original input reference saved here
    patch = unittest.mock.patch('builtins.input', func)
    patch.start()
    try:
        yield 
    finally:
        patch.stop()

validation = lambda x: len(x) <= 10

След това можете да използвате contextmanager по следния начин:

with validator(validation):
    x = input("10 or less: ")

x = input("10 or less (unpatched): ")
print("done")

Примерен резултат:

10 or less: abcdefghijklmnop
10 or less: abcdefgdfgdgd
10 or less: abcdef
10 or less (unpatched): abcdefghijklmnop
done
person dano    schedule 11.09.2014
comment
Бих казал обаче, че това е малко пресилено! (Изглежда, че и вие сте наясно с това, но реших, че все пак ще го кажа.) - person dano; 12.09.2014
comment
Няма инженерство като свръхинженерство. - person Adam Smith; 12.09.2014

arrow_upward
1
arrow_downward

Декоратор, който приема параметри, може да го направи доста добре (http://www.artima.com/weblogs/viewpost.jsp?thread=240845):

max10 = lambda x: len(x) <= 10

def validator(test):
    def wrap(func):
        def wrapped(*args, **kwargs):
            result = func(*args, **kwargs)
            if test(result):
                return result
            return None
        return wrapped
    return wrap

Редовният начин за използване:

@validator(max10)
def valid_input(prompt="Please enter a name: "):
    return raw_input(prompt)

print valid_input()

Прилагането на декоратора ръчно изглежда по-близо до това, което искате:

valid_input = validator(max10)(raw_input)
print valid_input("Please enter a name: ")
person systemjack    schedule 12.09.2014