Refactored code into better modules.

src/db/account.rs

@@ -1,7 +1,7 @@
 use rusqlite::{Connection, params, Result, Row, Transaction};
 
-use crate::securities::Security;
+use crate::db::securities::Security;
-use crate::users::User;
+use crate::db::users::User;
 
 pub struct Account {
     id: i64,
@@ -45,8 +45,8 @@ pub fn create_account(tx: &Transaction, sec: &Security, user: &User) -> Result<A
 #[cfg(test)]
 mod test {
     use rusqlite::{Connection};
-    use crate::users;
+    use crate::db::users;
-    use crate::securities;
+    use crate::db::securities;
     use crate::migration;
 
     use super::*;

src/db/mod.rs

@@ -0,0 +1,6 @@
+mod users;
+mod securities;
+mod account;
+
+pub use securities::{list_securities, Security};
+pub use users::{create_user, list_users, User};

src/main.rs

@@ -7,12 +7,11 @@ use serde::Serialize;
 use simple_logger::SimpleLogger;
 
 use crate::migration::apply_migrations;
-use crate::securities::list_securities;
+use db::list_securities;
+use crate::db::{create_user, list_users};
 
 mod migration;
-mod users;
+mod db;
-mod securities;
-mod account;
 
 #[derive(Debug, Serialize)]
 struct Output<T> {
@@ -269,12 +268,12 @@ impl ProCommandLineLogic {
 impl CommandLineUserLogic for ProCommandLineLogic {
     fn create_user(&mut self, conn: &mut Connection, name: &String, public_key: &String) {
         let tx = conn.transaction().unwrap();
-        users::create_user(& tx, name, public_key);
+        create_user(& tx, name, public_key);
         tx.commit().unwrap();
     }
 
     fn list_user(&mut self, conn: &Connection) {
-        let users = users::list_users(conn);
+        let users = list_users(conn);
         match users {
             Ok(u) => {
                 let out = Output::success(u);