From e7d6341787369bb823219c56a36d22ca5b02bcd2 Mon Sep 17 00:00:00 2001
From: root <root@debian>
Date: Mon, 12 Sep 2016 20:12:28 +0800
Subject: [PATCH] add a pymongo version check

---
 controllers.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/controllers.py b/controllers.py
index 47f8507..47c6ea6 100644
--- a/controllers.py
+++ b/controllers.py
@@ -6,9 +6,15 @@
 from bottle import static_file, redirect, HTTPResponse
 from bottle import mako_view as view
 from PIL import Image
-from pymongo.objectid import ObjectId
 from models import Message
 
+from pymongo import version
+version_num = int(''.join(version.split('.')))
+if version_num > 220:
+    from bson.objectid import ObjectId
+else:
+    from pymongo.objectid import ObjectId
+
 PAGE_SIZE = 5
 
 @get(['/', '/list', '/list/:page#\d+#'])