Lichen

Annotated results.py

611:d1d907801d42
2017-02-23 Paul Boddie Replaced list comprehension usage. method-wrapper-for-context
paul@11 1
#!/usr/bin/env python
paul@11 2
paul@11 3
"""
paul@11 4
Result abstractions.
paul@11 5
paul@510 6
Copyright (C) 2016, 2017 Paul Boddie <paul@boddie.org.uk>
paul@11 7
paul@11 8
This program is free software; you can redistribute it and/or modify it under
paul@11 9
the terms of the GNU General Public License as published by the Free Software
paul@11 10
Foundation; either version 3 of the License, or (at your option) any later
paul@11 11
version.
paul@11 12
paul@11 13
This program is distributed in the hope that it will be useful, but WITHOUT
paul@11 14
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
paul@11 15
FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
paul@11 16
details.
paul@11 17
paul@11 18
You should have received a copy of the GNU General Public License along with
paul@11 19
this program.  If not, see <http://www.gnu.org/licenses/>.
paul@11 20
"""
paul@11 21
paul@27 22
from referencing import Reference
paul@27 23
paul@11 24
# Classes representing inspection and translation observations.
paul@11 25
paul@11 26
class Result:
paul@11 27
paul@11 28
    "An abstract expression result."
paul@11 29
paul@11 30
    def is_name(self):
paul@11 31
        return False
paul@118 32
paul@603 33
    def is_global_name(self):
paul@603 34
        return False
paul@603 35
paul@226 36
    def reference(self):
paul@226 37
        return None
paul@226 38
paul@552 39
    def references(self):
paul@552 40
        return None
paul@552 41
paul@554 42
    def access_location(self):
paul@553 43
        return None
paul@553 44
paul@226 45
    def get_name(self):
paul@226 46
        return None
paul@226 47
paul@11 48
    def get_origin(self):
paul@11 49
        return None
paul@11 50
paul@226 51
    def static(self):
paul@118 52
        return None
paul@118 53
paul@226 54
    def final(self):
paul@226 55
        return None
paul@226 56
paul@226 57
    def has_kind(self, kinds):
paul@226 58
        return False
paul@226 59
paul@11 60
class AccessRef(Result):
paul@11 61
paul@11 62
    """
paul@11 63
    A reference to an attribute access that is generally only returned from a
paul@11 64
    processed access for possible initialiser resolution for assignments.
paul@11 65
    """
paul@11 66
paul@11 67
    def __init__(self, original_name, attrnames, number):
paul@11 68
        self.original_name = original_name
paul@11 69
        self.attrnames = attrnames
paul@11 70
        self.number = number
paul@11 71
paul@11 72
    def __repr__(self):
paul@11 73
        return "AccessRef(%r, %r, %r)" % (self.original_name, self.attrnames, self.number)
paul@11 74
paul@11 75
class InvocationRef(Result):
paul@11 76
paul@11 77
    "An invocation of a name reference."
paul@11 78
paul@11 79
    def __init__(self, name_ref):
paul@11 80
        self.name_ref = name_ref
paul@11 81
paul@27 82
    def reference(self):
paul@27 83
        origin = self.name_ref.get_origin()
paul@27 84
        if origin:
paul@27 85
            return Reference("<invoke>", origin)
paul@27 86
        else:
paul@27 87
            return Reference("<var>")
paul@27 88
paul@11 89
    def __repr__(self):
paul@11 90
        return "InvocationRef(%r)" % self.name_ref
paul@11 91
paul@11 92
class NameRef(Result):
paul@11 93
paul@11 94
    "A reference to a name."
paul@11 95
paul@603 96
    def __init__(self, name, expr=None, is_global=False):
paul@11 97
        self.name = name
paul@11 98
        self.expr = expr
paul@603 99
        self.is_global = is_global
paul@11 100
paul@11 101
    def is_name(self):
paul@11 102
        return True
paul@11 103
paul@603 104
    def is_global_name(self):
paul@603 105
        return self.is_global
paul@603 106
paul@11 107
    def final(self):
paul@11 108
        return None
paul@11 109
paul@11 110
    def __repr__(self):
paul@603 111
        return "NameRef(%r, %r, %r)" % (self.name, self.expr, self.is_global)
paul@11 112
paul@11 113
class LocalNameRef(NameRef):
paul@11 114
paul@11 115
    "A reference to a local name."
paul@11 116
paul@11 117
    def __init__(self, name, number):
paul@603 118
        NameRef.__init__(self, name, is_global=False)
paul@11 119
        self.number = number
paul@11 120
paul@11 121
    def __repr__(self):
paul@11 122
        return "LocalNameRef(%r, %r)" % (self.name, self.number)
paul@11 123
paul@317 124
class ResolvedRef:
paul@11 125
paul@317 126
    "A resolved reference mix-in."
paul@11 127
paul@552 128
    def __init__(self, ref):
paul@552 129
        self.ref = ref
paul@552 130
paul@11 131
    def reference(self):
paul@11 132
        return self.ref
paul@11 133
paul@552 134
    def references(self):
paul@552 135
        return [self.ref]
paul@552 136
paul@11 137
    def get_name(self):
paul@11 138
        return self.ref and self.ref.get_name() or None
paul@11 139
paul@11 140
    def get_origin(self):
paul@11 141
        return self.ref and self.ref.get_origin() or None
paul@11 142
paul@11 143
    def static(self):
paul@11 144
        return self.ref and self.ref.static() or None
paul@11 145
paul@11 146
    def final(self):
paul@11 147
        return self.ref and self.ref.final() or None
paul@11 148
paul@11 149
    def has_kind(self, kinds):
paul@11 150
        return self.ref and self.ref.has_kind(kinds)
paul@11 151
paul@338 152
    def is_constant_alias(self):
paul@338 153
        return self.ref and self.ref.is_constant_alias()
paul@338 154
paul@317 155
class ResolvedNameRef(ResolvedRef, NameRef):
paul@317 156
paul@317 157
    "A resolved name-based reference."
paul@317 158
paul@603 159
    def __init__(self, name, ref, expr=None, is_global=False):
paul@603 160
        NameRef.__init__(self, name, expr, is_global)
paul@552 161
        ResolvedRef.__init__(self, ref)
paul@317 162
paul@11 163
    def __repr__(self):
paul@603 164
        return "ResolvedNameRef(%r, %r, %r, %r)" % (self.name, self.ref, self.expr, self.is_global)
paul@11 165
paul@11 166
class ConstantValueRef(ResolvedNameRef):
paul@11 167
paul@11 168
    "A constant reference representing a single literal value."
paul@11 169
paul@11 170
    def __init__(self, name, ref, value, number=None):
paul@11 171
        ResolvedNameRef.__init__(self, name, ref)
paul@11 172
        self.value = value
paul@11 173
        self.number = number
paul@11 174
paul@11 175
    def __repr__(self):
paul@11 176
        return "ConstantValueRef(%r, %r, %r, %r)" % (self.name, self.ref, self.value, self.number)
paul@11 177
paul@317 178
class InstanceRef(ResolvedRef, Result):
paul@11 179
paul@11 180
    "An instance reference."
paul@11 181
paul@11 182
    def reference(self):
paul@11 183
        return self.ref
paul@11 184
paul@11 185
    def __repr__(self):
paul@11 186
        return "InstanceRef(%r)" % self.ref
paul@11 187
paul@11 188
class LiteralSequenceRef(ResolvedNameRef):
paul@11 189
paul@11 190
    "A reference representing a sequence of values."
paul@11 191
paul@11 192
    def __init__(self, name, ref, node, items=None):
paul@11 193
        ResolvedNameRef.__init__(self, name, ref)
paul@11 194
        self.node = node
paul@11 195
        self.items = items
paul@11 196
paul@11 197
    def __repr__(self):
paul@11 198
        return "LiteralSequenceRef(%r, %r, %r, %r)" % (self.name, self.ref, self.node, self.items)
paul@11 199
paul@226 200
class VariableRef(Result):
paul@226 201
paul@226 202
    "A variable reference."
paul@226 203
paul@226 204
    def __repr__(self):
paul@226 205
        return "VariableRef()"
paul@226 206
paul@11 207
# vim: tabstop=4 expandtab shiftwidth=4