Author: Adam Evyčędo <git@apiote.xyz>
save to TRAFFIC file lua scripts if they exist
file/file.go | 12 ++++++++++--
diff --git a/file/file.go b/file/file.go index e5865ba71e0e91762fbc42890a779cc8b4f4db53..020b851b21f27bcf81f7148f4a02aad27da60c71 100644 --- a/file/file.go +++ b/file/file.go @@ -89,11 +89,19 @@ defer xzWriter.Close() tarWriter := tar.NewWriter(xzWriter) defer tarWriter.Close() - files := []string{"calendar.bare", "ix_stop_codes.bare", "ix_stop_names.bare", "lines.bare", "stops.bare", "trips.bare", "vehicles.bare", "ix_lines.bare", "feed_info.bare", "agencies.bare", "ix_trips.bare", "updates.lua"} + files := []string{"calendar.bare", "ix_stop_codes.bare", "ix_stop_names.bare", "lines.bare", "stops.bare", "trips.bare", "vehicles.bare", "ix_lines.bare", "feed_info.bare", "agencies.bare", "ix_trips.bare", "updates.lua", "vehicles.lua", "alerts.lua"} + optionalFiles := map[string]struct{}{ + "updates.lua": {}, + "vehicles.lua": {}, + "alerts.lua": {}, + } for _, file := range files { stat, err := os.Stat(filepath.Join(path, file)) if err != nil { - // TODO if updates.lua does not exist, ignore + if _, ok := optionalFiles[file]; ok { + log.Printf("while stating %s: %v; ignoring\n", file, err) + continue + } return err } bareFile, err := os.Open(filepath.Join(path, file))