[Zope3-checkins] CVS: Zope3/lib/python/Zope/App/schemagen - __init__.py:1.1 interfaces.py:1.1 typereg.py:1.1

[Martijn Faassen]

Update of /cvs-repository/Zope3/lib/python/Zope/App/schemagen
In directory cvs.zope.org:/tmp/cvs-serv30330

Added Files:
	__init__.py interfaces.py typereg.py 
Log Message:
Added code to generate schema python code from schema. Not done yet,
but this basic framework passes the tests.


=== Added File Zope3/lib/python/Zope/App/schemagen/__init__.py ===
# this is empty



=== Added File Zope3/lib/python/Zope/App/schemagen/interfaces.py ===
##############################################################################
#
# Copyright (c) 2002 Zope Corporation and Contributors.
# All Rights Reserved.
#
# This software is subject to the provisions of the Zope Public License,
# Version 2.0 (ZPL).  A copy of the ZPL should accompany this distribution.
# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
# FOR A PARTICULAR PURPOSE.
#
##############################################################################
"""XXX short summary goes here.

XXX longer description goes here.

$Id: interfaces.py,v 1.1 2002/12/11 16:11:14 faassen Exp $
"""

from Interface import Interface
from Interface.Attribute import Attribute

class ITypeRepresentation(Interface):

    importList = Attribute('List of two-string tuples for use in '
                           'from X import Y')

    text = Attribute('Textual representation of object')


    def getTypes():
        """Return the sequence of types this representation can represent.
        """


=== Added File Zope3/lib/python/Zope/App/schemagen/typereg.py ===
##############################################################################
#
# Copyright (c) 2002 Zope Corporation and Contributors.
# All Rights Reserved.
#
# This software is subject to the provisions of the Zope Public License,
# Version 2.0 (ZPL).  A copy of the ZPL should accompany this distribution.
# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
# FOR A PARTICULAR PURPOSE.
#
##############################################################################
"""XXX short summary goes here.

XXX longer description goes here.

$Id: typereg.py,v 1.1 2002/12/11 16:11:14 faassen Exp $
"""

from Interface.Implements import visitImplements
from Zope.Schema import getFields

from interfaces import ITypeRepresentation

class TypeRepresentationRegistry:
    def __init__(self, default):
        self._registry = {}
        self._default = default
        
    #def registerType(self, type, factory):
    #    self._registry[type] = factory

    def represent(self, object):
        # returns an ITypeRepresentation object
        factory = self._registry.get(type(object))
        if factory is not None:
            return factory(object)
        return self._default(object)

    def register(self, representation):
        for type in representation.getTypes():
            self._registry[type] = representation
        
class DefaultTypeRepresentation:
    __implements__ = ITypeRepresentation

    def __init__(self, object):
        self.text = repr(object)
        import_name = type(object).__name__
        import_module = type(object).__module__
        if import_module != '__builtin__':
            self.importList = [(import_module, import_name)]
        else:
            self.importList = []
            
    def getTypes():
        return ()
    getTypes = staticmethod(getTypes)

class DatetimeRepresentation:
    __implements__ = ITypeRepresentation

    def __init__(self, dt):
        self.text = repr(dt)
        self.importList = [('datetime', type(dt).__name__)]

    def getTypes():
        import datetime
        # XXX not supporting tz or timedelta yet
        return [
            datetime.date,
            datetime.datetime,
          # datetime.datetimetz,
            datetime.time,
          #  datetime.timedelta,
          # datetime.timetz
        ]
    getTypes = staticmethod(getTypes)

class DefaultFieldRepresentation:
    __implements__ = ITypeRepresentation

    def __init__(self, field):
        # This field is described by a schema, or schemas, as found in its
        # __implements__ attribute. The fields of this field's schema are
        # its properties -- that is, the things we give as arguments when
        # constructing this field.
        # We're going to get these properties, get source-code
        # representations of them, and sort out appropriate imports.
        names = {} # used as set of property names, ignoring values
        visitImplements(field.__implements__, field,
                        lambda interface: names.update(getFields(interface)))
        # getFields only returns data for Fields in the interface.
        # Otherwise, it returns an empty dict.
    
        # At this point, name_property is a dict with keys of the
        # property names, and values of property objects.
        # XXX I don't know if we're following proper MRO though.

        global typeRegistry
        self.importList =  self._getImportList(field)
        arguments = []
        # don't represent order, as that will be implicit
        if 'order' in names:
            del names['order']
        for name in names:
            representation = typeRegistry.represent(getattr(field, name))
            arguments.append((name, representation.text))
            for import_spec in representation.importList:
                self.importList.append(import_spec) 

        arguments_text = ', '.join(["%s=%s" % item for item in arguments])

        self.text = "%s(%s)" % (type(field).__name__, arguments_text)
        
    def getTypes():
        return ()
    getTypes = staticmethod(getTypes)

    def _getImportList(field):
        import Zope.Schema

        field_class = type(field)
        if getattr(Zope.Schema, field_class.__name__, None) is field_class:
            module_name = 'Zope.Schema'
        else:
            module_name = field_class.__module__
        return [(module_name, field_class.__name__)]
    
    _getImportList = staticmethod(_getImportList)
    
typeRegistry = TypeRepresentationRegistry(DefaultTypeRepresentation)
typeRegistry.register(DatetimeRepresentation)

fieldRegistry = TypeRepresentationRegistry(DefaultFieldRepresentation)