Fix some err bugs

xalanq · Apr 11, 2019 · b02abc8 · b02abc8
1 parent 4f0d2ea
commit b02abc8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/cmd/cmd.go b/cmd/cmd.go
@@ -119,6 +119,9 @@ func getCode(args map[string]interface{}, templates []config.CodeTemplate) (code
 		return
 	}
 	paths, err := ioutil.ReadDir(path)
+	if err != nil {
+		return
+	}
 
 	for _, path := range paths {
 		name := path.Name()

diff --git a/cmd/parse.go b/cmd/parse.go
@@ -26,7 +26,7 @@ func Parse(args map[string]interface{}) error {
 		}
 		return cln.ParseContest(contestID, currentPath)
 	}
-	if err := work(); err != nil {
+	if err = work(); err != nil {
 		if err = loginAgain(cfg, cln, err); err == nil {
 			err = work()
 		}