john-okeefe/checkdigitcalculator
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

allowed file to be entered as argument

Browse Source
This commit is contained in:
John O'Keefe 2023-12-09 11:51:31 -05:00
parent 52714efc92
commit d38cfc8036
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
main.go
View File

@ -21,10 +21,18 @@ func main() {
Error(err) Error(err)
// Get name of file to be processed. // Get name of file to be processed.
pterm.Printf(pterm.Cyan("What is the name of the file in this directory?: ")) args := os.Args[1:]
reader := bufio.NewReader(os.Stdin)
input, err := reader.ReadString('\n') var input string
Error(err)
if len(args) > 0 {
input = args[0]
} else {
pterm.Printf(pterm.Cyan("What is the name of the file in this directory?: "))
reader := bufio.NewReader(os.Stdin)
input, err = reader.ReadString('\n')
Error(err)
}
// Create upcs array and push all upc strings into it // Create upcs array and push all upc strings into it
input = strings.TrimSpace(input) input = strings.TrimSpace(input)