Merge pull request #2 from kluzynskn6/master

Added a user import from csv.

Author: timothyhollabaugh

backend/Cargo.lock

@@ -12,3 +12,8 @@ serde = { version = "1.0", features = ["derive"]}
 serde_json = "1.0"
 log = "0.4"
 simplelog = "0.5"
+csv = "1.0.5"
+
+[[bin]]
+name = "csv_user_import"
+path = "src/bin/csv_user_import.rs"

backend/Cargo.toml

@@ -0,0 +1,104 @@
+use log::debug;
+use log::error;
+use log::info;
+use log::trace;
+use log::warn;
+use diesel::prelude::*;
+use diesel::MysqlConnection;
+use dotenv::dotenv;
+use csv;
+use web_dev::users::models::{NewUser,UserRequest};
+use web_dev::users::requests;
+use serde::Deserialize;
+use serde::Serialize;
+
+#[derive(Serialize, Deserialize, Debug)]
+//Struct to take the data from the csv
+struct Csv_User {
+	#[serde(rename = "Banner ID")]
+	banner_id: i32,
+	#[serde(rename = "Last Name")]
+	last_name: String,
+	#[serde(rename = "First Name")]
+	first_name: String,
+	#[serde(rename = "Email")]
+	email: String,
+	#[serde(rename = "Year")]
+	year: String,
+	#[serde(rename = "Department")]
+	department: String,
+}
+fn main(){
+	//Diesel things
+	dotenv().ok();
+
+    simplelog::TermLogger::init(simplelog::LevelFilter::Trace, simplelog::Config::default())
+        .unwrap();
+
+    info!("Connecting to database");
+
+    let database_url = match env::var("DATABASE_URL") {
+        Ok(url) => url,
+        Err(e) => {
+            error!("Could not read DATABASE_URL environment variable");
+            return;
+        }
+    };
+
+    debug!("Connecting to {}", database_url);
+
+    let connection = match MysqlConnection::establish(&database_url) {
+        Ok(c) => c,
+        Err(e) => {
+            error!("Could not connect to database: {}", e);
+            return;
+        }
+    };
+
+    debug!("Connected to database");
+	//Get file name and path from args
+	use std::env;
+	let arg = env::args().nth(1);
+	let filename = 	match arg {
+		Some(name) => name,
+		None => {
+			error!("Needs a filename");
+			return;
+		}
+	};
+	debug!("{}", filename);
+	//Import the csv into an iterator
+	let mut user_count = 0;
+	let all_users_result = csv::Reader::from_path(filename);
+	let mut all_users = match all_users_result{
+		Ok(data) => data,
+		Err(e) => {
+				error!("Bad file. Error {}",e);
+				return;
+			}
+	};		
+	//Go through each item in the iterator
+	for result in all_users.deserialize(){
+		//Check to see if it's valid
+		let csv_user: Csv_User = match result{
+			Ok(data) => data,
+			Err(e) => {
+				error!("Bad data, {:?}", e);
+				return;
+			}
+		};
+		//Convert the user data from the csv and create a New User from it
+		let new_user:NewUser = NewUser{
+			first_name: csv_user.first_name,
+			last_name: csv_user.last_name,
+			email:  Some(csv_user.email),
+			banner_id: csv_user.banner_id as u32,
+		};
+		//Import new user into database
+		let import_user = UserRequest::CreateUser(new_user);
+		requests::handle_user(import_user, &connection);
+		user_count = user_count+1;
+	}
+	info!("Imported {} user(s)",user_count);
+}
+

backend/src/bin/csv_user_import.rs

@@ -0,0 +1,6 @@
+#[macro_use]
+extern crate diesel;
+
+
+pub mod errors;
+pub mod users;

backend/src/lib.rs

@@ -1,6 +1,3 @@
-#[macro_use]
-extern crate diesel;
-
 use std::env;
 use std::sync::Mutex;
 
@@ -18,8 +15,6 @@ use dotenv::dotenv;
 use self::errors::WebdevError;
 use self::errors::WebdevErrorKind;
 
-mod errors;
-mod users;
 
 use self::users::models::UserRequest;
 use self::users::requests::handle_user;

backend/src/main.rs

@@ -23,7 +23,7 @@ pub struct User {
     pub email: Option<String>,
 }
 
-#[derive(Insertable, Serialize, Deserialize)]
+#[derive(Insertable, Serialize, Deserialize, Debug)]
 #[table_name = "users"]
 pub struct NewUser {
     pub first_name: String,