From b3dfa9177c538756e256748c099d259687ca0284 Mon Sep 17 00:00:00 2001
From: virusdefender <1670873886@qq.com>
Date: Fri, 11 Sep 2015 23:22:42 +0800
Subject: [PATCH] =?UTF-8?q?=E9=A6=96=E9=A1=B5=E4=B8=8A=E9=A2=98=E7=9B=AE?=
 =?UTF-8?q?=E9=9A=BE=E5=BA=A6=E7=9A=84=E6=8E=92=E5=BA=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 problem/views.py                          | 15 ++++++++++++++-
 template/src/oj/problem/problem_list.html |  4 ++--
 2 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/problem/views.py b/problem/views.py
index daf3a74..ab15e43 100644
--- a/problem/views.py
+++ b/problem/views.py
@@ -228,6 +228,19 @@ def problem_list_page(request, page=1):
     if keyword:
         problems = problems.filter(Q(title__contains=keyword) | Q(description__contains=keyword))
 
+    difficulty_order = request.GET.get("order_by", None)
+    if difficulty_order:
+        if difficulty_order[0] == "-":
+            problems = problems.order_by("-difficulty")
+            difficulty_order = "difficulty"
+        else:
+            problems = problems.order_by("difficulty")
+            difficulty_order = "-difficulty"
+    else:
+        difficulty_order = "difficulty"
+
+
+
     # 按照标签筛选
     tag_text = request.GET.get("tag", None)
     if tag_text:
@@ -262,4 +275,4 @@ def problem_list_page(request, page=1):
                   {"problems": current_page, "page": int(page),
                    "previous_page": previous_page, "next_page": next_page,
                    "keyword": keyword, "tag": tag_text,
-                   "tags": tags})
+                   "tags": tags, "difficulty_order": difficulty_order})
diff --git a/template/src/oj/problem/problem_list.html b/template/src/oj/problem/problem_list.html
index ed57e16..878ded4 100644
--- a/template/src/oj/problem/problem_list.html
+++ b/template/src/oj/problem/problem_list.html
@@ -21,8 +21,8 @@
                             <th></th>
                             <th>#</th>
                             <th>题目</th>
-                            <th><a href="/problems/?order_by=difficulty">难度</a></th>
-                            <th><a href="/problems/?order_by=acceptance">通过率</a></th>
+                            <th><a href="/problems/?order_by={{ difficulty_order }}">难度</a></th>
+                            <th>通过率</th>
                         </tr>
                         </thead>
                         <tbody>