mirror of
https://github.com/python/cpython.git
synced 2025-07-16 07:45:20 +00:00
c59e220000
(Hard to believe these were never handled before) Add misc.mangle() that mangles based on the rules in compile.c. XXX Need to test the corner cases Update CodeGenerator with a class_name attribute bound to None. If a particular instance is created within a class scope, the instance's class_name is bound to that class's name. Add mangle() method to CodeGenerator that mangles if the class_name has a class_name in it. Modify the FunctionCodeGenerator family to handle an extra argument-- the class_name. Wrap all name ops and attrnames in calls to self.mangle()
64 lines
1.4 KiB
Python
64 lines
1.4 KiB
Python
import types
|
|
|
|
def flatten(tup):
|
|
elts = []
|
|
for elt in tup:
|
|
if type(elt) == types.TupleType:
|
|
elts = elts + flatten(elt)
|
|
else:
|
|
elts.append(elt)
|
|
return elts
|
|
|
|
class Set:
|
|
def __init__(self):
|
|
self.elts = {}
|
|
def __len__(self):
|
|
return len(self.elts)
|
|
def __contains__(self, elt):
|
|
return self.elts.has_key(elt)
|
|
def add(self, elt):
|
|
self.elts[elt] = elt
|
|
def elements(self):
|
|
return self.elts.keys()
|
|
def has_elt(self, elt):
|
|
return self.elts.has_key(elt)
|
|
def remove(self, elt):
|
|
del self.elts[elt]
|
|
def copy(self):
|
|
c = Set()
|
|
c.elts.update(self.elts)
|
|
return c
|
|
|
|
class Stack:
|
|
def __init__(self):
|
|
self.stack = []
|
|
self.pop = self.stack.pop
|
|
def __len__(self):
|
|
return len(self.stack)
|
|
def push(self, elt):
|
|
self.stack.append(elt)
|
|
def top(self):
|
|
return self.stack[-1]
|
|
|
|
MANGLE_LEN = 256 # magic constant from compile.c
|
|
|
|
def mangle(name, klass):
|
|
if not name.startswith('__'):
|
|
return name
|
|
if len(name) + 2 >= MANGLE_LEN:
|
|
return name
|
|
if name.endswith('__'):
|
|
return name
|
|
try:
|
|
i = 0
|
|
while klass[i] == '_':
|
|
i = i + 1
|
|
except IndexError:
|
|
return name
|
|
klass = klass[i:]
|
|
|
|
tlen = len(klass) + len(name)
|
|
if tlen > MANGLE_LEN:
|
|
klass = klass[:MANGLE_LEN-tlen]
|
|
|
|
return "_%s%s" % (klass, name)
|