-
Notifications
You must be signed in to change notification settings - Fork 0
/
typepy.py
37 lines (34 loc) · 1.39 KB
/
typepy.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Thu Aug 24 17:29:00 2017
@author: misakawa
"""
class strict:
@staticmethod
def args(*typeargs : (*(),), **typekwargs:dict):
def _1(func):
def _2(*args, **kwargs):
for arg_idx, (arg,typearg) in enumerate(zip(args, typeargs)):
if not isinstance(arg, typearg):
raise TypeError(f"Type of argnument {arg_idx} should {typearg}")
for key in kwargs:
if not isinstance(kwargs[key], typekwargs[key]):
raise TypeError(f"Type of argnument named {key} should {typekwargs[key]}")
return func(*args,**kwargs)
return _2
return _1
def ret(*typerets):
def _1(func):
def _2(*args, **kwargs):
ret = func(*args, **kwargs)
if len(typerets) > 1:
for ret_idx,(ret_i, typeret) in enumerate(zip(ret, typerets)):
if not isinstance(ret_i, typeret):
raise TypeError(f"Type of return value {ret_idx} should {typeret}")
else:
if not isinstance(ret, typerets[0]):
raise TypeError(f"Type of return value should be {typerets[0]}")
return ret
return _2
return _1