Esempio n. 1
0
    def test_get_class(self):
        from geoalchemy2 import Geometry
        import c2cgeoportal.lib.dbreflection
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table("table_a")
        modelclass = get_class("table_a")

        # test the class
        self.assertEquals(modelclass.__name__, "Table_a")
        self.assertEquals(modelclass.__table__.name, "table_a")
        self.assertEquals(modelclass.__table__.schema, "public")

        self.assertTrue(isinstance(modelclass.point.type, Geometry))
        self.assertTrue(isinstance(modelclass.linestring.type, Geometry))
        self.assertTrue(isinstance(modelclass.polygon.type, Geometry))
        self.assertTrue(isinstance(modelclass.multipoint.type, Geometry))
        self.assertTrue(isinstance(modelclass.multilinestring.type, Geometry))
        self.assertTrue(isinstance(modelclass.multipolygon.type, Geometry))

        # test the Table object
        table = modelclass.__table__
        self.assertTrue("id" in table.c)
        self.assertTrue("child1_id" in table.c)
        self.assertTrue("child2_id" in table.c)
        self.assertTrue("point" in table.c)
        self.assertTrue("linestring" in table.c)
        self.assertTrue("polygon" in table.c)
        self.assertTrue("multipoint" in table.c)
        self.assertTrue("multilinestring" in table.c)
        self.assertTrue("multipolygon" in table.c)
        col_child1_id = table.c["child1_id"]
        self.assertEqual(col_child1_id.name, "child1_id")
        col_child2_id = table.c["child2_id"]
        self.assertEqual(col_child2_id.name, "child2_id")
        col_point = table.c["point"]
        self.assertEqual(col_point.name, "point")
        self.assertEqual(col_point.type.geometry_type, "POINT")
        col_linestring = table.c["linestring"]
        self.assertEqual(col_linestring.name, "linestring")
        self.assertEqual(col_linestring.type.geometry_type, "LINESTRING")
        col_polygon = table.c["polygon"]
        self.assertEqual(col_polygon.name, "polygon")
        self.assertEqual(col_polygon.type.geometry_type, "POLYGON")
        col_multipoint = table.c["multipoint"]
        self.assertEqual(col_multipoint.name, "multipoint")
        self.assertEqual(col_multipoint.type.geometry_type, "MULTIPOINT")
        col_multilinestring = table.c["multilinestring"]
        self.assertEqual(col_multilinestring.name, "multilinestring")
        self.assertEqual(col_multilinestring.type.geometry_type,
                         "MULTILINESTRING")
        col_multipolygon = table.c["multipolygon"]
        self.assertEqual(col_multipolygon.name, "multipolygon")
        self.assertEqual(col_multipolygon.type.geometry_type, "MULTIPOLYGON")

        # the class should now be in the cache
        self.assertTrue(("public", "table_a",
                         "") in c2cgeoportal.lib.dbreflection._class_cache)
        _modelclass = get_class("table_a")
        self.assertTrue(_modelclass is modelclass)
Esempio n. 2
0
    def test_get_class(self):
        from geoalchemy import SpatialComparator
        from geoalchemy import (Point, LineString, Polygon,
                                MultiPoint, MultiLineString, MultiPolygon)
        import c2cgeoportal.lib.dbreflection
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table('table_a')
        modelclass = get_class('table_a')

        # test the class
        self.assertEquals(modelclass.__name__, 'Table_a')
        self.assertEquals(modelclass.__table__.name, 'table_a')
        self.assertEquals(modelclass.__table__.schema, 'public')
        self.assertTrue(isinstance(modelclass.point.comparator, SpatialComparator))
        self.assertTrue(isinstance(modelclass.linestring.comparator, SpatialComparator))
        self.assertTrue(isinstance(modelclass.polygon.comparator, SpatialComparator))
        self.assertTrue(isinstance(modelclass.multipoint.comparator, SpatialComparator))
        self.assertTrue(isinstance(modelclass.multilinestring.comparator, SpatialComparator))
        self.assertTrue(isinstance(modelclass.multipolygon.comparator, SpatialComparator))

        # test the Table object
        table = modelclass.__table__
        self.assertTrue('id' in table.c)
        self.assertTrue('child1_id' in table.c)
        self.assertTrue('child2_id' in table.c)
        self.assertTrue('point' in table.c)
        self.assertTrue('linestring' in table.c)
        self.assertTrue('polygon' in table.c)
        self.assertTrue('multipoint' in table.c)
        self.assertTrue('multilinestring' in table.c)
        self.assertTrue('multipolygon' in table.c)
        col_child1_id = table.c['child1_id']
        self.assertEqual(col_child1_id.name, 'child1_id')
        col_child2_id = table.c['child2_id']
        self.assertEqual(col_child2_id.name, 'child2_id')
        col_point = table.c['point']
        self.assertEqual(col_point.name, 'point')
        self.assertTrue(isinstance(col_point.type, Point))
        col_linestring = table.c['linestring']
        self.assertEqual(col_linestring.name, 'linestring')
        self.assertTrue(isinstance(col_linestring.type, LineString))
        col_polygon = table.c['polygon']
        self.assertEqual(col_polygon.name, 'polygon')
        self.assertTrue(isinstance(col_polygon.type, Polygon))
        col_multipoint = table.c['multipoint']
        self.assertEqual(col_multipoint.name, 'multipoint')
        self.assertTrue(isinstance(col_multipoint.type, MultiPoint))
        col_multilinestring = table.c['multilinestring']
        self.assertEqual(col_multilinestring.name, 'multilinestring')
        self.assertTrue(isinstance(col_multilinestring.type, MultiLineString))
        col_multipolygon = table.c['multipolygon']
        self.assertEqual(col_multipolygon.name, 'multipolygon')
        self.assertTrue(isinstance(col_multipolygon.type, MultiPolygon))

        # the class should now be in the cache
        self.assertTrue(('public', 'table_a') in \
                            c2cgeoportal.lib.dbreflection._class_cache)
        _modelclass = get_class('table_a')
        self.assertTrue(_modelclass is modelclass)
Esempio n. 3
0
    def test_get_class_exclude_properties(self):
        import c2cgeoportal.lib.dbreflection
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table("table_d")
        get_class("table_d", exclude_properties=["foo", "bar"])

        # the class should now be in the cache
        self.assertTrue(
            ("public", "table_d",
             "foo,bar") in c2cgeoportal.lib.dbreflection._class_cache)
    def test_get_class_exclude_properties(self):
        import c2cgeoportal.lib.dbreflection
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table('table_d')
        get_class('table_d', exclude_properties='foo,bar')

        # the class should now be in the cache
        self.assertTrue(
            ('public', 'table_d', 'foo,bar') in
            c2cgeoportal.lib.dbreflection._class_cache
        )
Esempio n. 5
0
    def test_get_class_exclude_properties(self):
        import c2cgeoportal.lib.dbreflection
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table('table_d')
        get_class('table_d', exclude_properties='foo,bar')

        # the class should now be in the cache
        self.assertTrue(
            ('public', 'table_d', 'foo,bar') in
            c2cgeoportal.lib.dbreflection._class_cache
        )
    def test_get_class_exclude_properties(self):
        import c2cgeoportal.lib.dbreflection
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table("table_d")
        get_class("table_d", exclude_properties=["foo", "bar"])

        # the class should now be in the cache
        self.assertTrue(
            ("public", "table_d", "foo,bar") in
            c2cgeoportal.lib.dbreflection._class_cache
        )
Esempio n. 7
0
 def _import_layer_wms(self, layer, messages):
     server = layer.ogc_server
     url = server.url_wfs or server.url
     if url is None:
         return
     for wms_layer in layer.layer.split(","):
         self._import_layer_attributes(
             url, wms_layer, layer.item_type, layer.name, layer.id, messages
         )
     if layer.geo_table is not None:
         exclude = [] if layer.exclude_properties is None else layer.exclude_properties.split(",")
         last_update_date = layer.get_metadatas("lastUpdateDateColumn")
         if len(last_update_date) == 1:
             exclude.append(last_update_date[0].value)
         last_update_user = layer.get_metadatas("lastUpdateUserColumn")
         if len(last_update_user) == 1:
             exclude.append(last_update_user[0].value)
         try:
             cls = get_class(layer.geo_table, exclude_properties=exclude)
             for column_property in class_mapper(cls).iterate_properties:
                 if isinstance(column_property, ColumnProperty) and \
                         len(column_property.columns) == 1 and \
                         not column_property.columns[0].primary_key and \
                         not column_property.columns[0].foreign_keys and \
                         not isinstance(column_property.columns[0].type, Geometry):
                     messages.append(Message(
                         None, column_property.key, None, [], "", "",
                         (".".join(["edit", layer.item_type, str(layer.id)]), layer.name)
                     ))
         except NoSuchTableError:
             exit(colorize("No such table '{}' for layer '{}'.".format(layer.geo_table, layer.name), RED))
Esempio n. 8
0
    def test_get_class_dotted_notation(self):
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table('table_b')
        modelclass = get_class('public.table_b')

        self.assertEquals(modelclass.__name__, 'Table_b')
        self.assertEquals(modelclass.__table__.name, 'table_b')
        self.assertEquals(modelclass.__table__.schema, 'public')
Esempio n. 9
0
    def test_get_class_dotted_notation(self):
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table('table_b')
        modelclass = get_class('public.table_b')

        self.assertEquals(modelclass.__name__, 'Table_b')
        self.assertEquals(modelclass.__table__.name, 'table_b')
        self.assertEquals(modelclass.__table__.schema, 'public')
Esempio n. 10
0
    def test_get_class_dotted_notation(self):
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table("table_b")
        modelclass = get_class("public.table_b")

        self.assertEquals(modelclass.__name__, "Table_b")
        self.assertEquals(modelclass.__table__.name, "table_b")
        self.assertEquals(modelclass.__table__.schema, "public")
Esempio n. 11
0
def get_layer_metadatas(layer):
    # exclude the columns used to record the last features update
    exclude = [] if layer.exclude_properties is None else layer.exclude_properties.split(
        ",")

    date_metadata = layer.get_metadatas("lastUpdateDateColumn")
    last_update_date = date_metadata[0] if len(date_metadata) == 1 else None
    if last_update_date is not None:
        exclude.append(last_update_date.value)
    user_metadata = layer.get_metadatas("lastUpdateUserColumn")
    last_update_user = user_metadata[0] if len(user_metadata) == 1 else None
    if last_update_user is not None:
        exclude.append(last_update_user.value)

    cls = get_class(layer.geo_table, exclude_properties=exclude)

    edit_columns = []

    for column_property in class_mapper(cls).iterate_properties:
        if isinstance(column_property, ColumnProperty):

            if len(column_property.columns) != 1:
                raise NotImplementedError  # pragma: no cover

            column = column_property.columns[0]

            # Exclude columns that are primary keys
            if not column.primary_key:
                properties = _convert_column_type(column.type)
                properties["name"] = column.key

                if column.nullable:
                    properties["nillable"] = True
                edit_columns.append(properties)
        else:
            for k, p in cls.__dict__.items():
                if not isinstance(p, _AssociationProxy):
                    continue

                relationship_property = class_mapper(cls) \
                    .get_property(p.target)
                target_cls = relationship_property.argument
                query = models.DBSession.query(
                    getattr(target_cls, p.value_attr))
                properties = {}
                if column.nullable:
                    properties["nillable"] = True

                properties["name"] = k
                properties["restriction"] = "enumeration"
                properties["type"] = "xsd:string"
                properties["enumeration"] = []
                for value in query:
                    properties["enumeration"].append(value[0])

                edit_columns.append(properties)
    return edit_columns
Esempio n. 12
0
    def test_get_class_dotted_notation(self):
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table("table_b")
        modelclass = get_class("public.table_b")

        self.assertEquals(modelclass.__name__, "Table_b")
        self.assertEquals(modelclass.__table__.name, "table_b")
        self.assertEquals(modelclass.__table__.schema, "public")
Esempio n. 13
0
def _get_geom_col_info(layer):
    """ Return information about the layer's geometry column, namely
    a ``(name, srid)`` tuple, where ``name`` is the name of the
    geometry column, and ``srid`` its srid.

    This function assumes that the names of geometry attributes
    in the mapped class are the same as those of geometry columns.
    """
    mapped_class = get_class(str(layer.geoTable))
    for p in class_mapper(mapped_class).iterate_properties:
        if not isinstance(p, ColumnProperty):
            continue  # pragma: no cover
        col = p.columns[0]
        if isinstance(col.type, Geometry):
            return col.name, col.type.srid
    raise HTTPInternalServerError()  # pragma: no cover
Esempio n. 14
0
def _get_geom_col_info(layer):
    """ Return information about the layer's geometry column, namely
    a ``(name, srid)`` tuple, where ``name`` is the name of the
    geometry column, and ``srid`` its srid.

    This function assumes that the names of geometry attributes
    in the mapped class are the same as those of geometry columns.
    """
    mapped_class = get_class(str(layer.geoTable))
    for p in class_mapper(mapped_class).iterate_properties:
        if not isinstance(p, ColumnProperty):
            continue  # pragma: no cover
        col = p.columns[0]
        if isinstance(col.type, Geometry):
            return col.name, col.type.srid
    raise HTTPInternalServerError()  # pragma: no cover
Esempio n. 15
0
    def metadata(self):
        set_common_headers(self.request, "layers", PRIVATE_CACHE)

        layer = self._get_layer_for_request()
        if not layer.public and self.request.user is None:
            raise HTTPForbidden()

        # exclude the columns used to record the last features update
        exclude = [] if layer.exclude_properties is None else layer.exclude_properties.split(
            ",")
        last_update_date = self._get_metadata(layer, "lastUpdateDateColumn")
        if last_update_date is not None:
            exclude.append(last_update_date)
        last_update_user = self._get_metadata(layer, "lastUpdateUserColumn")
        if last_update_user is not None:
            exclude.append(last_update_user)

        return get_class(layer.geo_table, exclude_properties=exclude)
Esempio n. 16
0
    def test_mixing_get_class_and_queries(self):
        """ This test shows that we can mix the use of DBSession
        and the db reflection API. """
        from c2cgeoportal.lib.dbreflection import get_class
        from c2cgeoportal.models import DBSession
        from sqlalchemy import text
        import transaction

        self._create_table("table_c")

        DBSession.execute(text("SELECT id FROM table_c"))

        modelclass = get_class("table_c")
        self.assertEquals(modelclass.__name__, "Table_c")

        # This commits the transaction created by DBSession.execute. This
        # is required here in the test because tearDown does table.drop,
        # which will block forever if the transaction is not committed.
        transaction.commit()
Esempio n. 17
0
    def test_mixing_get_class_and_queries(self):
        """ This test shows that we can mix the use of DBSession
        and the db reflection API. """
        from c2cgeoportal.lib.dbreflection import get_class
        from c2cgeoportal.models import DBSession
        from sqlalchemy import text
        import transaction

        self._create_table('table_c')

        DBSession.execute(text('SELECT id FROM table_c'))

        modelclass = get_class('table_c')
        self.assertEquals(modelclass.__name__, 'Table_c')

        # This commits the transaction created by DBSession.execute. This
        # is required here in the test because tearDown does table.drop,
        # which will block forever if the transaction is not committed.
        transaction.commit()
Esempio n. 18
0
 def _import_layer_wms(self, layer, messages):
     server = layer.ogc_server
     url = server.url_wfs or server.url
     if url is None:
         return
     for wms_layer in layer.layer.split(","):
         self._import_layer_attributes(url, wms_layer, layer.item_type,
                                       layer.name, messages)
     if layer.geo_table is not None and layer.geo_table != "":
         exclude = [] if layer.exclude_properties is None else layer.exclude_properties.split(
             ",")
         last_update_date = layer.get_metadatas("lastUpdateDateColumn")
         if len(last_update_date) == 1:
             exclude.append(last_update_date[0].value)
         last_update_user = layer.get_metadatas("lastUpdateUserColumn")
         if len(last_update_user) == 1:
             exclude.append(last_update_user[0].value)
         try:
             cls = get_class(layer.geo_table, exclude_properties=exclude)
             for column_property in class_mapper(cls).iterate_properties:
                 if isinstance(column_property, ColumnProperty) and \
                         len(column_property.columns) == 1 and \
                         not column_property.columns[0].primary_key and \
                         not column_property.columns[0].foreign_keys and \
                         not isinstance(column_property.columns[0].type, Geometry):
                     messages.append(
                         Message(
                             None, column_property.key, None, [], "", "",
                             (".".join(
                                 ["edit", layer.item_type,
                                  str(layer.id)]), layer.name)))
         except NoSuchTableError:
             print(
                 colorize(
                     "ERROR! No such table '{}' for layer '{}'.".format(
                         layer.geo_table, layer.name), RED))
             print(colorize(traceback.format_exc(), RED))
             if os.environ.get("IGNORE_I18N_ERRORS", "FALSE") != "TRUE":
                 raise
Esempio n. 19
0
def metadata(request):
    layer = _get_layer_for_request(request)
    if not layer.public and request.user is None:
        raise HTTPNotFound()
    return get_class(str(layer.geoTable))
Esempio n. 20
0
 def _metadata(self, geo_table, exclude_properties):
     return get_class(
         geo_table,
         exclude_properties=exclude_properties
     )
Esempio n. 21
0
def metadata(request):
    layer = _get_layer_for_request(request)
    if not layer.public and request.user is None:
        raise HTTPNotFound()
    return get_class(str(layer.geoTable))
Esempio n. 22
0
    def test_get_class(self):
        from geoalchemy import SpatialComparator
        from geoalchemy import (Point, LineString, Polygon, MultiPoint,
                                MultiLineString, MultiPolygon)
        import c2cgeoportal.lib.dbreflection
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table('table_a')
        modelclass = get_class('table_a')

        # test the class
        self.assertEquals(modelclass.__name__, 'Table_a')
        self.assertEquals(modelclass.__table__.name, 'table_a')
        self.assertEquals(modelclass.__table__.schema, 'public')
        self.assertTrue(
            isinstance(modelclass.point.comparator, SpatialComparator))
        self.assertTrue(
            isinstance(modelclass.linestring.comparator, SpatialComparator))
        self.assertTrue(
            isinstance(modelclass.polygon.comparator, SpatialComparator))
        self.assertTrue(
            isinstance(modelclass.multipoint.comparator, SpatialComparator))
        self.assertTrue(
            isinstance(modelclass.multilinestring.comparator,
                       SpatialComparator))
        self.assertTrue(
            isinstance(modelclass.multipolygon.comparator, SpatialComparator))

        # test the Table object
        table = modelclass.__table__
        self.assertTrue('id' in table.c)
        self.assertTrue('child1_id' in table.c)
        self.assertTrue('child2_id' in table.c)
        self.assertTrue('point' in table.c)
        self.assertTrue('linestring' in table.c)
        self.assertTrue('polygon' in table.c)
        self.assertTrue('multipoint' in table.c)
        self.assertTrue('multilinestring' in table.c)
        self.assertTrue('multipolygon' in table.c)
        col_child1_id = table.c['child1_id']
        self.assertEqual(col_child1_id.name, 'child1_id')
        col_child2_id = table.c['child2_id']
        self.assertEqual(col_child2_id.name, 'child2_id')
        col_point = table.c['point']
        self.assertEqual(col_point.name, 'point')
        self.assertTrue(isinstance(col_point.type, Point))
        col_linestring = table.c['linestring']
        self.assertEqual(col_linestring.name, 'linestring')
        self.assertTrue(isinstance(col_linestring.type, LineString))
        col_polygon = table.c['polygon']
        self.assertEqual(col_polygon.name, 'polygon')
        self.assertTrue(isinstance(col_polygon.type, Polygon))
        col_multipoint = table.c['multipoint']
        self.assertEqual(col_multipoint.name, 'multipoint')
        self.assertTrue(isinstance(col_multipoint.type, MultiPoint))
        col_multilinestring = table.c['multilinestring']
        self.assertEqual(col_multilinestring.name, 'multilinestring')
        self.assertTrue(isinstance(col_multilinestring.type, MultiLineString))
        col_multipolygon = table.c['multipolygon']
        self.assertEqual(col_multipolygon.name, 'multipolygon')
        self.assertTrue(isinstance(col_multipolygon.type, MultiPolygon))

        # the class should now be in the cache
        self.assertTrue(('public', 'table_a') in \
                            c2cgeoportal.lib.dbreflection._class_cache)
        _modelclass = get_class('table_a')
        self.assertTrue(_modelclass is modelclass)
Esempio n. 23
0
 def _get_protocol_for_layer(self, layer, **kwargs):
     """ Returns a papyrus ``Protocol`` for the ``Layer`` object. """
     cls = get_class(layer.geo_table)
     geom_attr = self._get_geom_col_info(layer)[0]
     return Protocol(DBSession, cls, geom_attr, **kwargs)
Esempio n. 24
0
 def _get_protocol_for_layer(self, layer, **kwargs):
     """ Returns a papyrus ``Protocol`` for the ``Layer`` object. """
     cls = get_class(str(layer.geo_table))
     geom_attr = self._get_geom_col_info(layer)[0]
     return Protocol(DBSession, cls, geom_attr, **kwargs)
Esempio n. 25
0
    def test_get_class(self):
        from geoalchemy2 import Geometry
        import c2cgeoportal.lib.dbreflection
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table("table_a")
        modelclass = get_class("table_a")

        # test the class
        self.assertEquals(modelclass.__name__, "Table_a")
        self.assertEquals(modelclass.__table__.name, "table_a")
        self.assertEquals(modelclass.__table__.schema, "public")

        self.assertTrue(isinstance(modelclass.point.type, Geometry))
        self.assertTrue(isinstance(modelclass.linestring.type, Geometry))
        self.assertTrue(isinstance(modelclass.polygon.type, Geometry))
        self.assertTrue(isinstance(modelclass.multipoint.type, Geometry))
        self.assertTrue(isinstance(modelclass.multilinestring.type, Geometry))
        self.assertTrue(isinstance(modelclass.multipolygon.type, Geometry))

        # test the Table object
        table = modelclass.__table__
        self.assertTrue("id" in table.c)
        self.assertTrue("child1_id" in table.c)
        self.assertTrue("child2_id" in table.c)
        self.assertTrue("point" in table.c)
        self.assertTrue("linestring" in table.c)
        self.assertTrue("polygon" in table.c)
        self.assertTrue("multipoint" in table.c)
        self.assertTrue("multilinestring" in table.c)
        self.assertTrue("multipolygon" in table.c)
        col_child1_id = table.c["child1_id"]
        self.assertEqual(col_child1_id.name, "child1_id")
        col_child2_id = table.c["child2_id"]
        self.assertEqual(col_child2_id.name, "child2_id")
        col_point = table.c["point"]
        self.assertEqual(col_point.name, "point")
        self.assertEqual(col_point.type.geometry_type, "POINT")
        col_linestring = table.c["linestring"]
        self.assertEqual(col_linestring.name, "linestring")
        self.assertEqual(col_linestring.type.geometry_type, "LINESTRING")
        col_polygon = table.c["polygon"]
        self.assertEqual(col_polygon.name, "polygon")
        self.assertEqual(col_polygon.type.geometry_type, "POLYGON")
        col_multipoint = table.c["multipoint"]
        self.assertEqual(col_multipoint.name, "multipoint")
        self.assertEqual(col_multipoint.type.geometry_type, "MULTIPOINT")
        col_multilinestring = table.c["multilinestring"]
        self.assertEqual(col_multilinestring.name, "multilinestring")
        self.assertEqual(col_multilinestring.type.geometry_type, "MULTILINESTRING")
        col_multipolygon = table.c["multipolygon"]
        self.assertEqual(col_multipolygon.name, "multipolygon")
        self.assertEqual(col_multipolygon.type.geometry_type, "MULTIPOLYGON")

        # the class should now be in the cache
        self.assertTrue(
            ("public", "table_a", "") in
            c2cgeoportal.lib.dbreflection._class_cache
        )
        _modelclass = get_class("table_a")
        self.assertTrue(_modelclass is modelclass)
Esempio n. 26
0
 def _metadata(self, geo_table, exclude_properties):
     return get_class(
         geo_table,
         exclude_properties=exclude_properties
     )
Esempio n. 27
0
    def test_get_class(self):
        from geoalchemy2 import Geometry
        import c2cgeoportal.lib.dbreflection
        from c2cgeoportal.lib.dbreflection import get_class

        self._create_table('table_a')
        modelclass = get_class('table_a')

        # test the class
        self.assertEquals(modelclass.__name__, 'Table_a')
        self.assertEquals(modelclass.__table__.name, 'table_a')
        self.assertEquals(modelclass.__table__.schema, 'public')

        self.assertTrue(isinstance(modelclass.point.type, Geometry))
        self.assertTrue(isinstance(modelclass.linestring.type, Geometry))
        self.assertTrue(isinstance(modelclass.polygon.type, Geometry))
        self.assertTrue(isinstance(modelclass.multipoint.type, Geometry))
        self.assertTrue(isinstance(modelclass.multilinestring.type, Geometry))
        self.assertTrue(isinstance(modelclass.multipolygon.type, Geometry))

        # test the Table object
        table = modelclass.__table__
        self.assertTrue('id' in table.c)
        self.assertTrue('child1_id' in table.c)
        self.assertTrue('child2_id' in table.c)
        self.assertTrue('point' in table.c)
        self.assertTrue('linestring' in table.c)
        self.assertTrue('polygon' in table.c)
        self.assertTrue('multipoint' in table.c)
        self.assertTrue('multilinestring' in table.c)
        self.assertTrue('multipolygon' in table.c)
        col_child1_id = table.c['child1_id']
        self.assertEqual(col_child1_id.name, 'child1_id')
        col_child2_id = table.c['child2_id']
        self.assertEqual(col_child2_id.name, 'child2_id')
        col_point = table.c['point']
        self.assertEqual(col_point.name, 'point')
        self.assertEqual(col_point.type.geometry_type, "POINT")
        col_linestring = table.c['linestring']
        self.assertEqual(col_linestring.name, 'linestring')
        self.assertEqual(col_linestring.type.geometry_type, "LINESTRING")
        col_polygon = table.c['polygon']
        self.assertEqual(col_polygon.name, 'polygon')
        self.assertEqual(col_polygon.type.geometry_type, "POLYGON")
        col_multipoint = table.c['multipoint']
        self.assertEqual(col_multipoint.name, 'multipoint')
        self.assertEqual(col_multipoint.type.geometry_type, "MULTIPOINT")
        col_multilinestring = table.c['multilinestring']
        self.assertEqual(col_multilinestring.name, 'multilinestring')
        self.assertEqual(col_multilinestring.type.geometry_type, "MULTILINESTRING")
        col_multipolygon = table.c['multipolygon']
        self.assertEqual(col_multipolygon.name, 'multipolygon')
        self.assertEqual(col_multipolygon.type.geometry_type, "MULTIPOLYGON")

        # the class should now be in the cache
        self.assertTrue(
            ('public', 'table_a', None) in
            c2cgeoportal.lib.dbreflection._class_cache
        )
        _modelclass = get_class('table_a')
        self.assertTrue(_modelclass is modelclass)