def restriction_area_test_data(dbsession, transact): del transact from c2cgeoportal_commons.models.main import LayerWMS, OGCServer, RestrictionArea, Role roles = [] for i in range(0, 4): roles.append(Role("secretary_" + str(i))) dbsession.add_all(roles) ogc_server = OGCServer(name="test_server") layers = [] for i in range(0, 4): layer = LayerWMS(name=f"layer_{i}", layer=f"layer_{i}", public=False) layer.ogc_server = ogc_server layers.append(layer) dbsession.add_all(layers) restrictionareas = [] for i in range(0, 4): restrictionarea = RestrictionArea(name=f"restrictionarea_{i}") restrictionarea.area = from_shape(box(485869.5728, 76443.1884, 837076.5648, 299941.7864), srid=21781) restrictionarea.description = f"description_{i}" restrictionarea.roles = [roles[i % 4], roles[(i + 2) % 4]] restrictionarea.layers = [layers[i % 4], layers[(i + 2) % 4]] dbsession.add(restrictionarea) restrictionareas.append(restrictionarea) dbsession.flush() yield { "layers": layers, "restriction_areas": restrictionareas, "roles": roles, }
def restriction_area_test_data(dbsession, transact): del transact from c2cgeoportal_commons.models.main import RestrictionArea, Role roles = [] for i in range(0, 4): roles.append(Role("secretary_" + str(i))) dbsession.add(roles[i]) restrictionareas = [] for i in range(0, 4): restrictionarea = RestrictionArea(name='restrictionarea_{}'.format(i)) restrictionarea.area = \ from_shape(box(485869.5728, 76443.1884, 837076.5648, 299941.7864), srid=21781) restrictionarea.description = 'description_{}'.format(i) restrictionarea.roles = [roles[i % 4], roles[(i + 2) % 4]] dbsession.add(restrictionarea) restrictionareas.append(restrictionarea) dbsession.flush() yield {'restriction_areas': restrictionareas, 'roles': roles}