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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
from rpython.rlib import jit
class Signature(object):
_immutable_ = True
_immutable_fields_ = ["argnames[*]", "kwonlyargnames[*]"]
__slots__ = ("argnames", "kwonlyargnames", "varargname", "kwargname")
def __init__(self, argnames, varargname=None, kwargname=None, kwonlyargnames=None):
self.argnames = argnames
self.varargname = varargname
self.kwargname = kwargname
if kwonlyargnames is None:
kwonlyargnames = []
self.kwonlyargnames = kwonlyargnames
@jit.elidable
def find_argname(self, name):
try:
return self.argnames.index(name)
except ValueError:
pass
try:
return len(self.argnames) + self.kwonlyargnames.index(name)
except ValueError:
pass
return -1
def num_argnames(self):
return len(self.argnames)
def num_kwonlyargnames(self):
return len(self.kwonlyargnames)
def has_vararg(self):
return self.varargname is not None
def has_kwarg(self):
return self.kwargname is not None
def scope_length(self):
scopelen = len(self.argnames)
scopelen += len(self.kwonlyargnames)
scopelen += self.has_vararg()
scopelen += self.has_kwarg()
return scopelen
def getallvarnames(self):
argnames = self.argnames
if self.varargname is not None:
argnames = argnames + [self.varargname]
argnames = argnames + self.kwonlyargnames
if self.kwargname is not None:
argnames = argnames + [self.kwargname]
return argnames
def __repr__(self):
return "Signature(%r, %r, %r, %r)" % (
self.argnames, self.varargname, self.kwargname, self.kwonlyargnames)
def __eq__(self, other):
if not isinstance(other, Signature):
return NotImplemented
return (self.argnames == other.argnames and
self.varargname == other.varargname and
self.kwargname == other.kwargname and
self.kwonlyargnames == other.kwonlyargnames)
def __ne__(self, other):
if not isinstance(other, Signature):
return NotImplemented
return not self == other
|
