adam/yr
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: Add interval to forecast cmd

Browse source 
Signed-off-by: Tim Hårek Andreassen <tim@harek.no>
This commit is contained in:
Tim Hårek Andreassen 2024-10-02 20:40:49 +02:00
parent 65dd6ad1f5
commit 6cb3994ae9
No known key found for this signature in database
GPG key ID: E59C7734F0E10EB5
2 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cmd/flags/flags.go
View file

@ -1,7 +1,8 @@
package flags package flags
const ( const (
LON = "lon" LON = "lon"
LAT = "lat" LAT = "lat"
JSON = "json" JSON = "json"
INTERVAL = "interval"
) )

6
cmd/forecast.go
View file

@ -24,6 +24,7 @@ var forecastCmd = &cobra.Command{
func init() { func init() {
rootCmd.AddCommand(forecastCmd) rootCmd.AddCommand(forecastCmd)
rootCmd.PersistentFlags().IntP(flags.INTERVAL, "i", 0, "Number of hours to output")
} }
func forecast(cmd *cobra.Command, args []string) { func forecast(cmd *cobra.Command, args []string) {
@ -31,6 +32,7 @@ func forecast(cmd *cobra.Command, args []string) {
cobra.CheckErr(err) cobra.CheckErr(err)
lon, _ := cmd.Flags().GetFloat64(flags.LON) lon, _ := cmd.Flags().GetFloat64(flags.LON)
lat, _ := cmd.Flags().GetFloat64(flags.LAT) lat, _ := cmd.Flags().GetFloat64(flags.LAT)
interval, _ := cmd.Flags().GetInt(flags.INTERVAL)
c, err := yr.New() c, err := yr.New()
cobra.CheckErr(err) cobra.CheckErr(err)
@ -43,12 +45,14 @@ func forecast(cmd *cobra.Command, args []string) {
if len(args) == 0 { if len(args) == 0 {
f, err = c.ForecastCoords(&nominatim.Coordinates{Longitude: lon, Latitude: lat}, nil) f, err = c.ForecastCoords(&nominatim.Coordinates{Longitude: lon, Latitude: lat}, nil)
cobra.CheckErr(err) cobra.CheckErr(err)
} else { } else {
location := args[0] location := args[0]
f, err = c.Forecast(location) f, err = c.Forecast(location)
cobra.CheckErr(err) cobra.CheckErr(err)
}
if interval > 0 {
f = f[:interval]
} }
if isJson { if isJson {