Add saving periods

src/db.rs

@@ -1,20 +1,18 @@
+use std::env;
+
+use diesel::dsl::sql;
+use diesel::prelude::*;
+use diesel_migrations::embed_migrations;
+use dotenv::dotenv;
+
+use errors::DatabaseError;
+use models::*;
+use schema::schedules::dsl::*;
+use crate::types::EmgauwaUid;
+
 pub mod errors;
 pub mod models;
 pub mod schema;
-mod types;
-
-use diesel::prelude::*;
-
-use diesel::dsl::sql;
-use dotenv::dotenv;
-use std::env;
-
-use models::*;
-use schema::schedules::dsl::*;
-
-use diesel_migrations::embed_migrations;
-use errors::DatabaseError;
-use types::EmgauwaUid;
 
 embed_migrations!("migrations");
 
@@ -34,18 +32,27 @@ pub fn run_migrations() {
 pub fn get_schedules() -> Vec<Schedule> {
     let connection = get_connection();
     schedules
-        .limit(5)
         .load::<Schedule>(&connection)
         .expect("Error loading schedules")
 }
 
-pub fn create_schedule(new_name: &str) -> Result<Schedule, DatabaseError> {
+pub fn get_schedule_by_uid(filter_uid: EmgauwaUid) -> Result<Schedule, DatabaseError> {
+    let connection = get_connection();
+    let result = schedules
+        .filter(uid.eq(filter_uid))
+        .first::<Schedule>(&connection)
+        .or(Err(DatabaseError::NotFound))?;
+
+    Ok(result)
+}
+
+pub fn create_schedule(new_name: &str, new_periods: &Periods) -> Result<Schedule, DatabaseError> {
     let connection = get_connection();
 
     let new_schedule = NewSchedule {
         uid: &EmgauwaUid::default(),
         name: new_name,
-        periods: "",
+        periods: new_periods
     };
 
     diesel::insert_into(schedules)