add flag for db path

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"database/sql"
+	"flag"
 	"fmt"
 	"html/template"
 	"log"
@@ -58,8 +59,8 @@ func GetDefaultTasks() []Task {
 	}
 }
 
-func InitDB() *sql.DB {
-	db, err := sql.Open("sqlite3", DB_NAME)
+func InitDB(db_name string) *sql.DB {
+	db, err := sql.Open("sqlite3", db_name)
 	if err != nil {
 		log.Fatal(err)
 	}
@@ -198,7 +199,10 @@ const TASK_DAY = time.Tuesday
 const AMOUNT_DAYS = 4
 
 func main() {
-	db := InitDB()
+	dbName := flag.String("d", DB_NAME, "path to db file")
+	flag.Parse()
+
+	db := InitDB(*dbName)
 	defer db.Close()
 
 	if db == nil {