Beispiel #1
0
 def testUpdateNotExist(self):
     route = Route(None)
     route.osm_relation_id = 10
     update_route(route)
     self.assertTrue(route.relation_not_found)
     self.assertIsNone(route.check_result)
Beispiel #2
0
def route_add(context, request):
    class RouteVariantSchema(colander.Schema):
        name = colander.SchemaNode(
            colander.String(),
            description="Variant name",
            missing=u'')
        osm_relation_id = colander.SchemaNode(
            colander.Int(),
            title="Openstreetmap Relation ID (type=route)",
            description="Openstreetmap relation ID of this route variant, if "
                "it exists in openstreetmap. Relation should have "
                "type=route key.",
            missing=None)
        stops = colander.SchemaNode(
            colander.String(),
            description="Stops, one stop name per line",
            missing=u'',
            widget=deform.widget.TextAreaWidget(rows=10, cols=60))

    class VariantsSequence(colander.SequenceSchema):
        variant = RouteVariantSchema()

    class Schema(colander.Schema):
        type = colander.SchemaNode(
            colander.String(),
            widget=deform.widget.SelectWidget(
                values=[('', '- Select -')] + [(v, v) for v in allowed_types]))
        ref = colander.SchemaNode(
            colander.String(),
            description="Route ref, i.e. '22'")
        name = colander.SchemaNode(
            colander.String(),
            description="Route name",
            missing=u'')
        osm_relation_id = colander.SchemaNode(
            colander.Int(),
            title="Openstreetmap Relation ID (type=route_master)",
            description="Openstreetmap relation ID of this route variant, if "
                "it exists in openstreetmap. Relation should have "
                "type=route key.",
            missing=None)
        variants = VariantsSequence()

    def validator(form, value):
        session = DBSession()
        if session.query(RouteMaster).filter_by(ref=value['ref'],
                city=context, type=value['type']).count() != 0:
            t = 'Route with this ref and type already exists'
            exc = colander.Invalid(form, t)
            exc['ref'] = t
            raise exc

    form_schema = Schema(validator=validator)
    form = deform.Form(form_schema, buttons=('submit',))
    form['variants'].widget = deform.widget.SequenceWidget(min_len=1)

    if 'submit' in request.POST:
        controls = request.POST.items()
        try:
            struct = form.validate(controls)
        except deform.ValidationFailure, e:
            return dict(form=e.render())

        session = DBSession()

        master = RouteMaster()
        master.city = context
        master.ref = struct['ref']
        master.type = struct['type']
        master.name = struct['name']
        master.osm_relation_id = struct['osm_relation_id']

        for v_s in struct['variants']:
            variant = Route()
            variant.name = v_s['name']
            variant.osm_relation_id = v_s['osm_relation_id']
            variant.stops = v_s['stops']
            master.routes.append(variant)

        context.route_masters.append(master)

        session.flush()
        transaction.commit()

        return HTTPFound(location=resource_url(context, request))
Beispiel #3
0
 def testUpdate380499(self):
     route = Route(None)
     route.osm_relation_id = 380499
     update_route(route)
     self.assertFalse(route.relation_not_found)
     self.assertIsNotNone(route.check_result)