From 593bd0cf30c640f9e4123c2189c84460cafcbce1 Mon Sep 17 00:00:00 2001
From: Matej Focko <xfocko@fi.muni.cz>
Date: Mon, 25 May 2020 23:49:52 +0200
Subject: [PATCH] Sort comments by lines and update mail extraction

---
 commands/comments.py |  7 +++++++
 submission.py        | 10 ++++++++--
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/commands/comments.py b/commands/comments.py
index 3993fbd..3d7fe79 100644
--- a/commands/comments.py
+++ b/commands/comments.py
@@ -2,6 +2,7 @@
 
 
 import json
+import math
 
 
 from commands.base import BaseCommand
@@ -22,6 +23,12 @@ class Comments(BaseCommand):
 
             result[author].append(comment)
 
+        # sort by lines
+        for author in result:
+            result[author].sort(key=lambda comment:
+                    comment["position"]["new_line"] if "position" in comment
+                    else math.inf)
+
         return result
 
     @staticmethod
diff --git a/submission.py b/submission.py
index f4a7ab9..e38599d 100644
--- a/submission.py
+++ b/submission.py
@@ -7,6 +7,8 @@ from typing import Dict, List
 
 
 class Submission:
+    MAIL_CONTENT= re.compile(r"<pre>((.*\s+)+)<\/pre>")
+
     def __init__(
         self,
         uco: str,
@@ -44,8 +46,12 @@ class Submission:
             self.flag = "LATE"
 
     def get_mail(self) -> str:
-        match = re.search(r"<pre>((.*\s+)+)<\/pre>", self.mail)
-        return match.group(1) if match else self.mail
+        left = self.mail.find("<pre>")
+        right = self.mail.rfind("</pre>")
+        return self.mail[left + 5:right]
+        # print(self.mail)
+        # match = Submission.MAIL_CONTENT.search(self.mail)
+        # return match.group(1) if match else self.mail
 
 
 def print_submissions(all_submissions: Dict[str, List[Submission]]) -> None: