# -*- coding: UTF-8 -*-
 
'''
@summary: 驗(yàn)證器
該模塊提供了一個(gè)裝飾器用于驗(yàn)證參數(shù)是否合法，使用方法為：
 
from validator import validParam, nullOk, multiType
 
@validParam(i=int)
def foo(i):
  return i+1
 
編寫(xiě)驗(yàn)證器：
 
1. 僅驗(yàn)證類型：
@validParam(type, ...)
例如：
檢查第一個(gè)位置的參數(shù)是否為int類型：
@validParam(int)
檢查名為x的參數(shù)是否為int類型：
@validParam(x=int)
 
驗(yàn)證多個(gè)參數(shù)：
@validParam(int, int)
指定參數(shù)名驗(yàn)證：
@validParam(int, s=str)
 
針對(duì)*和**參數(shù)編寫(xiě)的驗(yàn)證器將驗(yàn)證這些參數(shù)實(shí)際包含的每個(gè)元素：
@validParam(varargs=int)
def foo(*varargs): pass
 
@validParam(kws=int)
def foo7(s, **kws): pass
 
2. 帶有條件的驗(yàn)證：
@validParam((type, condition), ...)
其中，condition是一個(gè)表達(dá)式字符串，使用x引用待驗(yàn)證的對(duì)象；
根據(jù)bool(表達(dá)式的值)判斷是否通過(guò)驗(yàn)證，若計(jì)算表達(dá)式時(shí)拋出異常，視為失敗。
例如：
驗(yàn)證一個(gè)10到20之間的整數(shù)：
@validParam(i=(int, '10<x<20'))
驗(yàn)證一個(gè)長(zhǎng)度小于20的字符串：
@validParam(s=(str, 'len(x)<20'))
驗(yàn)證一個(gè)年齡小于20的學(xué)生：
@validParam(stu=(Student, 'x.age<20'))
 
另外，如果類型是字符串，condition還可以使用斜杠開(kāi)頭和結(jié)尾表示正則表達(dá)式匹配。
驗(yàn)證一個(gè)由數(shù)字組成的字符串：
@validParam(s=(str, '/^\d*$/'))
 
3. 以上驗(yàn)證方式默認(rèn)為當(dāng)值是None時(shí)驗(yàn)證失敗。如果None是合法的參數(shù)，可以使用nullOk()。
nullOk()接受一個(gè)驗(yàn)證條件作為參數(shù)。
例如：
@validParam(i=nullOk(int))
@validParam(i=nullOk((int, '10<x<20')))
也可以簡(jiǎn)寫(xiě)為：
@validParam(i=nullOk(int, '10<x<20'))
 
4. 如果參數(shù)有多個(gè)合法的類型，可以使用multiType()。
multiType()可接受多個(gè)參數(shù)，每個(gè)參數(shù)都是一個(gè)驗(yàn)證條件。
例如：
@validParam(s=multiType(int, str))
@validParam(s=multiType((int, 'x>20'), nullOk(str, '/^\d+$/')))
 
5. 如果有更復(fù)雜的驗(yàn)證需求，還可以編寫(xiě)一個(gè)函數(shù)作為驗(yàn)證函數(shù)傳入。
這個(gè)函數(shù)接收待驗(yàn)證的對(duì)象作為參數(shù)，根據(jù)bool(返回值)判斷是否通過(guò)驗(yàn)證，拋出異常視為失敗。
例如：
def validFunction(x):
  return isinstance(x, int) and x>0
@validParam(i=validFunction)
def foo(i): pass
 
這個(gè)驗(yàn)證函數(shù)等價(jià)于：
@validParam(i=(int, 'x>0'))
def foo(i): pass
 
 
@author: HUXI
@since: 2011-3-22
@change: 
'''
 
import inspect
import re
 
class ValidateException(Exception): pass
 
 
def validParam(*varargs, **keywords):
  '''驗(yàn)證參數(shù)的裝飾器。'''
   
  varargs = map(_toStardardCondition, varargs)
  keywords = dict((k, _toStardardCondition(keywords[k]))
          for k in keywords)
   
  def generator(func):
    args, varargname, kwname = inspect.getargspec(func)[:3]
    dctValidator = _getcallargs(args, varargname, kwname,
                  varargs, keywords)
     
    def wrapper(*callvarargs, **callkeywords):
      dctCallArgs = _getcallargs(args, varargname, kwname,
                    callvarargs, callkeywords)
       
      k, item = None, None
      try:
        for k in dctValidator:
          if k == varargname:
            for item in dctCallArgs[k]:
              assert dctValidator[k](item)
          elif k == kwname:
            for item in dctCallArgs[k].values():
              assert dctValidator[k](item)
          else:
            item = dctCallArgs[k]
            assert dctValidator[k](item)
      except:
        raise ValidateException,\
            ('%s() parameter validation fails, param: %s, value: %s(%s)'
            % (func.func_name, k, item, item.__class__.__name__))
       
      return func(*callvarargs, **callkeywords)
     
    wrapper = _wrapps(wrapper, func)
    return wrapper
   
  return generator
 
 
def _toStardardCondition(condition):
  '''將各種格式的檢查條件轉(zhuǎn)換為檢查函數(shù)'''
   
  if inspect.isclass(condition):
    return lambda x: isinstance(x, condition)
   
  if isinstance(condition, (tuple, list)):
    cls, condition = condition[:2]
    if condition is None:
      return _toStardardCondition(cls)
     
    if cls in (str, unicode) and condition[0] == condition[-1] == '/':
      return lambda x: (isinstance(x, cls)
               and re.match(condition[1:-1], x) is not None)
     
    return lambda x: isinstance(x, cls) and eval(condition)
   
  return condition
 
 
def nullOk(cls, condition=None):
  '''這個(gè)函數(shù)指定的檢查條件可以接受None值'''
   
  return lambda x: x is None or _toStardardCondition((cls, condition))(x)
 
 
def multiType(*conditions):
  '''這個(gè)函數(shù)指定的檢查條件只需要有一個(gè)通過(guò)'''
   
  lstValidator = map(_toStardardCondition, conditions)
  def validate(x):
    for v in lstValidator:
      if v(x):
        return True
  return validate
 
 
def _getcallargs(args, varargname, kwname, varargs, keywords):
  '''獲取調(diào)用時(shí)的各參數(shù)名-值的字典'''
   
  dctArgs = {}
  varargs = tuple(varargs)
  keywords = dict(keywords)
   
  argcount = len(args)
  varcount = len(varargs)
  callvarargs = None
   
  if argcount <= varcount:
    for n, argname in enumerate(args):
      dctArgs[argname] = varargs[n]
     
    callvarargs = varargs[-(varcount-argcount):]
   
  else:
    for n, var in enumerate(varargs):
      dctArgs[args[n]] = var
     
    for argname in args[-(argcount-varcount):]:
      if argname in keywords:
        dctArgs[argname] = keywords.pop(argname)
     
    callvarargs = ()
   
  if varargname is not None:
    dctArgs[varargname] = callvarargs
   
  if kwname is not None:
    dctArgs[kwname] = keywords
   
  dctArgs.update(keywords)
  return dctArgs
 
 
def _wrapps(wrapper, wrapped):
  '''復(fù)制元數(shù)據(jù)'''
   
  for attr in ('__module__', '__name__', '__doc__'):
    setattr(wrapper, attr, getattr(wrapped, attr))
  for attr in ('__dict__',):
    getattr(wrapper, attr).update(getattr(wrapped, attr, {}))
   
  return wrapper
 
 
#===============================================================================
# 測(cè)試
#===============================================================================
 
 
def _unittest(func, *cases):
  for case in cases:
    _functest(func, *case)
   
 
def _functest(func, isCkPass, *args, **kws):
  if isCkPass:
    func(*args, **kws)
  else:
    try:
      func(*args, **kws)
      assert False
    except ValidateException:
      pass
 
def _test1_simple():
  #檢查第一個(gè)位置的參數(shù)是否為int類型：
  @validParam(int)
  def foo1(i): pass
  _unittest(foo1, 
       (True, 1), 
       (False, 's'), 
       (False, None))
 
  #檢查名為x的參數(shù)是否為int類型：
  @validParam(x=int)
  def foo2(s, x): pass
  _unittest(foo2, 
       (True, 1, 2), 
       (False, 's', 's'))
   
  #驗(yàn)證多個(gè)參數(shù)：
  @validParam(int, int)
  def foo3(s, x): pass
  _unittest(foo3, 
       (True, 1, 2), 
       (False, 's', 2))
   
  #指定參數(shù)名驗(yàn)證：
  @validParam(int, s=str)
  def foo4(i, s): pass
  _unittest(foo4, 
       (True, 1, 'a'), 
       (False, 's', 1))
   
  #針對(duì)*和**參數(shù)編寫(xiě)的驗(yàn)證器將驗(yàn)證這些參數(shù)包含的每個(gè)元素：
  @validParam(varargs=int)
  def foo5(*varargs): pass
  _unittest(foo5,
       (True, 1, 2, 3, 4, 5),
       (False, 'a', 1))
   
  @validParam(kws=int)
  def foo6(**kws): pass
  _functest(foo6, True, a=1, b=2)
  _functest(foo6, False, a='a', b=2)
   
  @validParam(kws=int)
  def foo7(s, **kws): pass
  _functest(foo7, True, s='a', a=1, b=2)
 
 
def _test2_condition():
  #驗(yàn)證一個(gè)10到20之間的整數(shù)：
  @validParam(i=(int, '10<x<20'))
  def foo1(x, i): pass
  _unittest(foo1, 
       (True, 1, 11), 
       (False, 1, 'a'), 
       (False, 1, 1))
   
  #驗(yàn)證一個(gè)長(zhǎng)度小于20的字符串：
  @validParam(s=(str, 'len(x)<20'))
  def foo2(a, s): pass
  _unittest(foo2, 
       (True, 1, 'a'), 
       (False, 1, 1), 
       (False, 1, 'a'*20))
   
  #驗(yàn)證一個(gè)年齡小于20的學(xué)生：
  class Student(object):
    def __init__(self, age): self.age=age
   
  @validParam(stu=(Student, 'x.age<20'))
  def foo3(stu): pass
  _unittest(foo3, 
       (True, Student(18)), 
       (False, 1), 
       (False, Student(20)))
   
  #驗(yàn)證一個(gè)由數(shù)字組成的字符串：
  @validParam(s=(str, r'/^\d*$/'))
  def foo4(s): pass
  _unittest(foo4, 
       (True, '1234'), 
       (False, 1), 
       (False, 'a1234'))
 
 
def _test3_nullok():
  @validParam(i=nullOk(int))
  def foo1(i): pass
  _unittest(foo1, 
       (True, 1), 
       (False, 'a'), 
       (True, None))
   
  @validParam(i=nullOk(int, '10<x<20'))
  def foo2(i): pass
  _unittest(foo2, 
       (True, 11), 
       (False, 'a'), 
       (True, None), 
       (False, 1))
 
 
def _test4_multitype():
  @validParam(s=multiType(int, str))
  def foo1(s): pass
  _unittest(foo1, 
       (True, 1),
       (True, 'a'),
       (False, None),
       (False, 1.1))
   
  @validParam(s=multiType((int, 'x>20'), nullOk(str, '/^\d+$/')))
  def foo2(s): pass
  _unittest(foo2, 
       (False, 1),
       (False, 'a'),
       (True, None),
       (False, 1.1),
       (True, 21),
       (True, '21'))
 
def _main():
  d = globals()
  from types import FunctionType
  print
  for f in d:
    if f.startswith('_test'):
      f = d[f]
      if isinstance(f, FunctionType):
        f()
 
if __name__ == '__main__':
  _main()