lang: funcs: Add a weekday function to the datetime package
This returns the day of the week. It also includes a helpful example demonstrating how this functionality can be fun!
This commit is contained in:
30
examples/lang/readonlyfriday.mcl
Normal file
30
examples/lang/readonlyfriday.mcl
Normal file
@@ -0,0 +1,30 @@
|
|||||||
|
import "datetime"
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
$now = datetime.now()
|
||||||
|
$day = datetime.weekday($now)
|
||||||
|
$is_friday = $day == "friday"
|
||||||
|
|
||||||
|
$s1 = template("Hello! It is now: {{ datetime_print . }}\n", $now)
|
||||||
|
$s2 = if $is_friday {
|
||||||
|
"It's friday!!! (don't break anything, read-only)"
|
||||||
|
} else {
|
||||||
|
if $day == "saturday" || $day == "sunday" {
|
||||||
|
"It's the weekend!"
|
||||||
|
} else {
|
||||||
|
fmt.printf("Unfortunately, it is %s. Go to work!", $day)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
print "msg" {
|
||||||
|
msg => $s1 + $s2,
|
||||||
|
}
|
||||||
|
|
||||||
|
file "/tmp/files/" {
|
||||||
|
state => "exists",
|
||||||
|
mode => if $is_friday { # this updates the mode, the instant it changes!
|
||||||
|
"0550"
|
||||||
|
} else {
|
||||||
|
"0770"
|
||||||
|
},
|
||||||
|
}
|
||||||
49
lang/funcs/core/datetime/weekday_func.go
Normal file
49
lang/funcs/core/datetime/weekday_func.go
Normal file
@@ -0,0 +1,49 @@
|
|||||||
|
// Mgmt
|
||||||
|
// Copyright (C) 2013-2019+ James Shubin and the project contributors
|
||||||
|
// Written by James Shubin <james@shubin.ca> and the project contributors
|
||||||
|
//
|
||||||
|
// This program is free software: you can redistribute it and/or modify
|
||||||
|
// it under the terms of the GNU General Public License as published by
|
||||||
|
// the Free Software Foundation, either version 3 of the License, or
|
||||||
|
// (at your option) any later version.
|
||||||
|
//
|
||||||
|
// This program is distributed in the hope that it will be useful,
|
||||||
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
// GNU General Public License for more details.
|
||||||
|
//
|
||||||
|
// You should have received a copy of the GNU General Public License
|
||||||
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
package coredatetime
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/purpleidea/mgmt/lang/funcs/simple"
|
||||||
|
"github.com/purpleidea/mgmt/lang/types"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
simple.ModuleRegister(moduleName, "weekday", &types.FuncValue{
|
||||||
|
T: types.NewType("func(a int) str"),
|
||||||
|
V: Weekday,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Weekday returns the lowercased day of the week corresponding to the input
|
||||||
|
// time. The time is the number of seconds since the epoch, and matches what
|
||||||
|
// comes from our Now function.
|
||||||
|
func Weekday(input []types.Value) (types.Value, error) {
|
||||||
|
epochDelta := input[0].Int()
|
||||||
|
if epochDelta < 0 {
|
||||||
|
return nil, fmt.Errorf("epoch delta must be positive")
|
||||||
|
}
|
||||||
|
|
||||||
|
weekday := time.Unix(epochDelta, 0).Weekday()
|
||||||
|
return &types.StrValue{
|
||||||
|
V: strings.ToLower(weekday.String()),
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user