python: tự động in đại diện của mỗi thành phần trong một biểu thức

Question

tôi thấy mình viết khẳng định như thế này:

if f(x, y) != z: 
    print(repr(x)) 
    print(repr(y)) 
    print(repr(z)) 
    raise MyException('Expected: f(x, y) == z') 

Tôi đã tự hỏi nếu có một cách để viết một hàm có thể chấp nhận một biểu Python hợp lệ và một ngoại lệ lớp như một đầu vào, đánh giá biểu thức và nếu nó tìm thấy nó là sai, in ra biểu diễn của từng thuật ngữ mức thấp nhất trong biểu thức và tăng ngoại lệ đã cho?

# validate is the mystery function 
validate('f(x, y) == z', MyException) 

Answer 1

Dưới đây là một thực hiện:

import inspect, keyword, pprint, sys, tokenize 

def value_in_frame(name, frame): 
    try: 
     return frame.f_locals[name] 
    except KeyError: 
     try: 
      return frame.f_globals[name] 
     except KeyError: 
      raise ValueError("Couldn't find value for %s" % name) 

def validate(expr, exc_class=AssertionError): 
    """Evaluate `expr` in the caller's frame, raise `exc_class` if false.""" 
    frame = inspect.stack()[1][0] 
    val = eval(expr, frame.f_globals, frame.f_locals) 
    if not val: 
     rl = iter([expr]).next 
     for typ, tok, _, _, _ in tokenize.generate_tokens(rl): 
      if typ == tokenize.NAME and not keyword.iskeyword(tok): 
       try: 
        val = value_in_frame(tok, frame) 
       except ValueError: 
        val = '???' 
       else: 
        val = repr(val) 
       print " %s: %s" % (tok, val) 
     raise exc_class("Failed to validate: %s" % expr) 

if __name__ == '__main__': 
    a = b = 3 
    validate("a + b == 5") 

Answer 2

Điều này có thể thực hiện được. Bạn có thể sử dụng trình biên dịch Python (chi tiết khác nhau giữa các phiên bản, vì vậy đây chỉ là tổng quan) để biên dịch biểu thức đã cho thành một AST. Sau đó, bạn có thể biên dịch AST thành một đối tượng mã và đánh giá nó (hoặc chỉ cần gọi eval ở nơi đầu tiên, bất cứ điều gì). Sau đó, nếu giá trị là giả, hãy kiểm tra AST để xem cách biểu thức được xây dựng. In ra giá trị của mỗi mục trong biểu thức được truy cập theo tên theo AST.

Answer 3

Điều gì về xác nhận?

assert f(x, y) != z, 'Expected: f(%r, %r) == %r'%(x,y,z) 

Sửa

thêm% r in repr của - cảm ơn cho nhận xét.

Answer 4

Có thể có một cách để hack lên một cái gì đó để làm những gì bạn yêu cầu, nhưng các lựa chọn thay thế là, ít nhất, dễ dàng hơn nhiều:

1. Như bạn có bây giờ, tay mở rộng các giá trị của sự quan tâm. Điều này có khả năng đáng tin cậy hơn, đặc biệt là cho các biểu thức phức tạp, bởi vì nó dễ dàng hơn cho bạn để nói những gì thú vị hơn cho máy tính để suy luận nó. Cụ thể, bạn có thể muốn bao gồm giá trị không có trong biểu thức.

   if f(x, y) != z: 
       raise MyException("Unexpected: the foobar is glowing; " + 
           "f(%r, %r) != %r" % (x, y, z)) 
   

   Ở đây, bạn dường như muốn x và y nhưng không (kết quả của việc gọi) f (x, y), trong khi ở các trường hợp khác có thể được lộn, hoặc bạn có thể muốn cả ba. Một thuật toán sẽ có một thời gian khó xác định sở thích của bạn.

2. Bạn có thể bắt đầu pdb tại điểm mà bạn phát hiện lỗi, kiểm tra bất cứ điều gì bạn thích (thậm chí lên ngăn xếp cuộc gọi), và sau đó tiếp tục thực hiện "bình thường". Điều này chỉ có thể trong một số ngữ cảnh; ví dụ. có thể không dành cho ứng dụng web.

   if f(x, y) != z: 
       import pdb 
       pdb.set_trace() 
       raise MyException("Unexpected: ...") 
   

3. Mô-đun ghi nhật ký thay vì báo cáo in hoặc trong thông báo ngoại lệ.

Answer 5

Á hậu thử nghiệm nose có một trình cắm gọi là "chi tiết lỗi" cung cấp chính xác dịch vụ này: http://somethingaboutorange.com/mrl/projects/nose/0.11.1/plugins/failuredetail.html. Giải pháp này là tốt hơn so với những gì bạn đã yêu cầu, bởi vì biểu thức không phải là một chuỗi, đó là một xác nhận thực tế mà là introspected sau này để tìm mã nguồn để phân tích.

Một ví dụ là trong older docs:

Nói cách khác, nếu bạn có một bài kiểm tra như:

def test_integers(): 
    a = 2 
    assert a == 4, "assert 2 is 4" 

Bạn sẽ nhận được kết quả như sau:

File "/path/to/file.py", line XX, in test_integers: 
     assert a == 4, "assert 2 is 4" 
AssertionError: assert 2 is 4 
    >> assert 2 == 4, "assert 2 is 4" 

Answer 6

Bạn có thể làm điều đó theo chiều ngược lại:

expr = 'f(%r, %r) != %r' % (x,y,z) 

if eval(expr): 
    raise MyException(expr) 

Hoặc nói cách khác:

def validate(expr,myexception): 
    if eval(expr): 
     raise myexception(expr) 

Khá bẩn mặc dù :)