diff --git a/main.go b/main.go index 55489e0..afcf51f 100644 --- a/main.go +++ b/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 {