Page MenuHomec4science
Files F62207786

class_decriptor.py
No OneTemporary
Actions

File Metadata

Created
Sat, May 11, 15:31

class_decriptor.py
View Options

#!/usr/bin/env python
# -*- coding: utf-8 -*-
""" Descriptor for classes """
################################################################
class Typename(object):
" Describes a typename "
def __init__(self, name, encapsulation):
self.name = name
self.encapsulation = encapsulation
################################################################
class Method(object):
" Describes a method "
def __init__(self, name, args, ret, encapsulation,
virtual, static, const, comments):
self.name = name
self.virtual = virtual
self.static = static
self.args = dict()
self.args.update(args)
self.ret = ret
self.encapsulation = encapsulation
if self.encapsulation == '':
self.encapsulation = 'public'
self.comments = comments
self.const = const
#print "creating method {0}".format(name)
#print self.__dict__
def __str__(self):
sstr = self.encapsulation + " "
if not self.virtual == '':
sstr += self.virtual + " "
sstr += self.ret + " " + self.name + "("
pairs = list(self.args.iteritems())
pairs = [b + " " + a for a, b in pairs]
sstr += ", ".join(pairs)
sstr += ")"
return sstr
def __eq__(self, other):
return (isinstance(other, self.__class__)
and self.__dict__ == other.__dict__)
def __ne__(self, other):
return not self.__eq__(other)
def __hash__(self):
return hash(str(self))
################################################################
class Member(object):
" Descriptor of a method "
def __init__(self, name, _type, encapsulation, static, comments):
self.name = name
self.type = _type
self.encapsulation = encapsulation
self.static = static
self.comments = comments
def __str__(self):
return self.encapsulation + " " + self.type + " " + self.name
def __eq__(self, other):
return (isinstance(other, self.__class__)
and self.__dict__ == other.__dict__)
################################################################
class ClassDescriptor(object):
" Describe a class "
def __init__(self, name, inheritance=None):
self.name = name
self.inheritance = inheritance
self.members = {'private':{}, 'public':{}, 'protected':{}}
self.methods = {'private':{}, 'public':{}, 'protected':{}}
self.types = {'private':{}, 'public':{}, 'protected':{}}
def add_method(self, name, args, ret, encapsulation,
virtual, static, const, comments):
" append a method to the class "
new_method = Method(name, args, ret, encapsulation, virtual, static, const, comments)
if name not in self.methods[encapsulation]:
self.methods[encapsulation][name] = set()
self.methods[encapsulation][name].add(new_method)
def add_member(self, name, _type, encapsulation, static, comments):
" append a member to the class "
new_member = Member(name, _type, encapsulation, static, comments)
self.members[encapsulation][name] = new_member
def add_type(self, name, encapsulation):
" add a type to the list of typenames "
new_type = Typename(name, encapsulation)
self.types[encapsulation][name] = new_type
def get_members(self, encapsulation=None):
" return the members "
return self.members[encapsulation]
def get_types(self, encapsulation=None):
" return the typenames "
return self.types[encapsulation]
def get_methods(self, encapsulation=None):
" return the methods "
return self.methods[encapsulation]
def __str__(self):
sstr = "Class " + self.name + "\n"
if self.inheritance:
sstr += "Inherit: "
sstr += ",".join(self.inheritance) + "\n"
sstr += "Methods:\n"
for encaps, meths in self.methods.iteritems():
sstr += encaps + ":\n"
for dummy_name, m_list in meths.iteritems():
for method in m_list:
sstr += str(method) + "\n"
sstr += "\n"
sstr += "Members:\n"
for encaps, membs in self.members.iteritems():
sstr += encaps + ":\n"
for dummy_name, memb in membs.iteritems():
sstr += str(memb) + "\n"
return sstr
################################################################
def main():
my_class = ClassDescriptor('dummy')
my_class.add_method('compute', [('int', 'arg1'), ('double', 'arg2')], 'bool', 'public', '')
my_class.add_member('res', 'double', 'private')
print my_class
################################################################
if __name__ == '__main__':
main()

Event Timeline

c4science · Help