Read database path from arguments

1 files changed, 2 insertions, 2 deletions

diff --git a/src/db/db.rs b/src/db/db.rs
index a3398e2..86614ec 100644
--- a/src/db/db.rs
+++ b/src/db/db.rs
@@ -9,8 +9,8 @@ use rusqlite::{params, Connection};
 use rusqlite_migration::{Migrations, M};
 use std::time::SystemTime;
 
-pub fn init() -> Result<Connection> {
-    let mut conn = Connection::open("database.db")?;
+pub fn init(database: String) -> Result<Connection> {
+    let mut conn = Connection::open(database)?;
     let migrations = Migrations::new(vec![M::up(include_str!("sql/1-init.sql"))]);
     migrations.to_latest(&mut conn)?;
     Ok(conn)