Lichen

Annotated results.py

210:263f32338b37
2016-11-22 Paul Boddie Prevent lambdas with defaults being referenced as static function objects.
paul@11 1
#!/usr/bin/env python
paul@11 2
paul@11 3
"""
paul@11 4
Result abstractions.
paul@11 5
paul@11 6
Copyright (C) 2016 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@11 33
    def get_origin(self):
paul@11 34
        return None
paul@11 35
paul@118 36
    def reference(self):
paul@118 37
        return None
paul@118 38
paul@11 39
class AccessRef(Result):
paul@11 40
paul@11 41
    """
paul@11 42
    A reference to an attribute access that is generally only returned from a
paul@11 43
    processed access for possible initialiser resolution for assignments.
paul@11 44
    """
paul@11 45
paul@11 46
    def __init__(self, original_name, attrnames, number):
paul@11 47
        self.original_name = original_name
paul@11 48
        self.attrnames = attrnames
paul@11 49
        self.number = number
paul@11 50
paul@11 51
    def __repr__(self):
paul@11 52
        return "AccessRef(%r, %r, %r)" % (self.original_name, self.attrnames, self.number)
paul@11 53
paul@11 54
class InvocationRef(Result):
paul@11 55
paul@11 56
    "An invocation of a name reference."
paul@11 57
paul@11 58
    def __init__(self, name_ref):
paul@11 59
        self.name_ref = name_ref
paul@11 60
paul@27 61
    def reference(self):
paul@27 62
        origin = self.name_ref.get_origin()
paul@27 63
        if origin:
paul@27 64
            return Reference("<invoke>", origin)
paul@27 65
        else:
paul@27 66
            return Reference("<var>")
paul@27 67
paul@11 68
    def __repr__(self):
paul@11 69
        return "InvocationRef(%r)" % self.name_ref
paul@11 70
paul@11 71
class NameRef(Result):
paul@11 72
paul@11 73
    "A reference to a name."
paul@11 74
paul@11 75
    def __init__(self, name, expr=None):
paul@11 76
        self.name = name
paul@11 77
        self.expr = expr
paul@11 78
paul@11 79
    def is_name(self):
paul@11 80
        return True
paul@11 81
paul@11 82
    def final(self):
paul@11 83
        return None
paul@11 84
paul@11 85
    def __repr__(self):
paul@11 86
        return "NameRef(%r, %r)" % (self.name, self.expr)
paul@11 87
paul@11 88
class LocalNameRef(NameRef):
paul@11 89
paul@11 90
    "A reference to a local name."
paul@11 91
paul@11 92
    def __init__(self, name, number):
paul@11 93
        NameRef.__init__(self, name)
paul@11 94
        self.number = number
paul@11 95
paul@11 96
    def __repr__(self):
paul@11 97
        return "LocalNameRef(%r, %r)" % (self.name, self.number)
paul@11 98
paul@11 99
class ResolvedNameRef(NameRef):
paul@11 100
paul@11 101
    "A resolved name-based reference."
paul@11 102
paul@11 103
    def __init__(self, name, ref, expr=None):
paul@11 104
        NameRef.__init__(self, name, expr)
paul@11 105
        self.ref = ref
paul@11 106
paul@11 107
    def reference(self):
paul@11 108
        return self.ref
paul@11 109
paul@11 110
    def get_name(self):
paul@11 111
        return self.ref and self.ref.get_name() or None
paul@11 112
paul@11 113
    def get_origin(self):
paul@11 114
        return self.ref and self.ref.get_origin() or None
paul@11 115
paul@11 116
    def static(self):
paul@11 117
        return self.ref and self.ref.static() or None
paul@11 118
paul@11 119
    def final(self):
paul@11 120
        return self.ref and self.ref.final() or None
paul@11 121
paul@11 122
    def has_kind(self, kinds):
paul@11 123
        return self.ref and self.ref.has_kind(kinds)
paul@11 124
paul@11 125
    def __repr__(self):
paul@11 126
        return "ResolvedNameRef(%r, %r, %r)" % (self.name, self.ref, self.expr)
paul@11 127
paul@11 128
class ConstantValueRef(ResolvedNameRef):
paul@11 129
paul@11 130
    "A constant reference representing a single literal value."
paul@11 131
paul@11 132
    def __init__(self, name, ref, value, number=None):
paul@11 133
        ResolvedNameRef.__init__(self, name, ref)
paul@11 134
        self.value = value
paul@11 135
        self.number = number
paul@11 136
paul@11 137
    def __repr__(self):
paul@11 138
        return "ConstantValueRef(%r, %r, %r, %r)" % (self.name, self.ref, self.value, self.number)
paul@11 139
paul@11 140
class InstanceRef(Result):
paul@11 141
paul@11 142
    "An instance reference."
paul@11 143
paul@11 144
    def __init__(self, ref):
paul@11 145
        self.ref = ref
paul@11 146
paul@11 147
    def reference(self):
paul@11 148
        return self.ref
paul@11 149
paul@11 150
    def __repr__(self):
paul@11 151
        return "InstanceRef(%r)" % self.ref
paul@11 152
paul@11 153
class LiteralSequenceRef(ResolvedNameRef):
paul@11 154
paul@11 155
    "A reference representing a sequence of values."
paul@11 156
paul@11 157
    def __init__(self, name, ref, node, items=None):
paul@11 158
        ResolvedNameRef.__init__(self, name, ref)
paul@11 159
        self.node = node
paul@11 160
        self.items = items
paul@11 161
paul@11 162
    def __repr__(self):
paul@11 163
        return "LiteralSequenceRef(%r, %r, %r, %r)" % (self.name, self.ref, self.node, self.items)
paul@11 164
paul@11 165
# vim: tabstop=4 expandtab shiftwidth=4