Apply default agency when importing

interline-io · Oct 13, 2023 · 9feaada · 9feaada
1 parent 721ba03
commit 9feaada
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/dmfr/importer/importer.go b/dmfr/importer/importer.go
@@ -9,6 +9,7 @@ import (
 	"github.com/interline-io/transitland-lib/dmfr"
 	"github.com/interline-io/transitland-lib/dmfr/store"
 	"github.com/interline-io/transitland-lib/ext/builders"
+	"github.com/interline-io/transitland-lib/filters"
 	"github.com/interline-io/transitland-lib/log"
 	"github.com/interline-io/transitland-lib/tl"
 	"github.com/interline-io/transitland-lib/tl/causes"
@@ -201,6 +202,11 @@ func ImportFeedVersion(atx tldb.Adapter, fv tl.FeedVersion, opts Options) (dmfr.
 	if err != nil {
 		return fvi, err
 	}
+
+	// Avoid foreign key errors when writing to database
+	cp.AddExtension(&filters.ApplyDefaultAgencyFilter{})
+
+	// Builder extensions
 	cp.AddExtension(builders.NewRouteGeometryBuilder())
 	cp.AddExtension(builders.NewRouteStopBuilder())
 	cp.AddExtension(builders.NewRouteHeadwayBuilder())