regular matching of READMD.md failed, unable to parse the field

Author: bestgopher

src/file.rs

@@ -6,7 +6,7 @@ use regex::Regex;
 use crate::http::{Resp, Data, Ques, Difficulty};
 
 lazy_static!(
-    static ref RE: Regex = Regex::new(r"\|\s*([0-9]*)\s*\|\s*(.*?)\s*\|.*?bin/(.*?)\.rs.*?\|.*?\|").unwrap();
+    static ref RE: Regex = Regex::new(r"\|\s*([0-9]*)\s*\|\s*(.*?)\s*\|.*?bin/(.*?)\.rs.*?\|.*?\|\s*?(\w*)\s*?\|").unwrap();
 );
 
 /// 将结果写入README.md中
@@ -77,7 +77,7 @@ fn parse(contents: &str) -> Vec<Resp> {
                         translated_title: i.get(2).unwrap().as_str().to_string(),
                         title_slug: i.get(3).unwrap().as_str().to_string(),
                         code_snippets: vec![],
-                        difficulty: Difficulty::Easy,
+                        difficulty: Difficulty::new(i.get(4).unwrap().as_str()),
                     }
                 },
             })
@@ -99,7 +99,12 @@ mod tests {
         println!("{}", x.len());
 
         for i in x {
-            println!("{}, {}, {}", i.data.question.translated_title, i.data.question.question_id, i.data.question.title_slug);
+            println!("{}, {}, {}, {:?}",
+                     i.data.question.translated_title,
+                     i.data.question.question_id,
+                     i.data.question.title_slug,
+                     i.data.question.difficulty
+            );
         }
     }
 

src/http.rs

@@ -20,6 +20,17 @@ pub enum Difficulty {
     Hard,
 }
 
+impl Difficulty {
+    pub fn new(s: &str) -> Self {
+        match s {
+            "Easy" => Self::Easy,
+            "Medium" => Self::Medium,
+            "Hard" => Self::Hard,
+            _ => Self::Easy
+        }
+    }
+}
+
 impl Serialize for Difficulty {
     fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
         where
@@ -117,7 +128,7 @@ pub fn get_question_info(mut ques: &str) -> Resp {
         .send()
         .unwrap()
         .json::<Resp>()
-        .unwrap()
+        .expect(format!("{} download failed", ques).as_str())
 }
 
 #[cfg(test)]