更新mongo测试

zhanghe06 · zhanghe06 · commit 49dc58a479e7 · 2017-07-08T22:22:48.000+08:00
diff --git a/test/test_mongo.py b/test/test_mongo.py
@@ -6,6 +6,46 @@
 sys.path.append('..')
 from tools.mongo import Mongodb
 
+import time
+import calendar
+from datetime import datetime, timedelta, date
+
+
+def time_local_to_utc(local_time):
+    """
+    本地时间转UTC时间
+    :param local_time:
+    :return:
+    """
+    # 字符串处理
+    if isinstance(local_time, str) and len(local_time) == 10:
+        local_time = datetime.strptime(local_time, '%Y-%m-%d')
+    elif isinstance(local_time, str) and len(local_time) >= 19:
+        local_time = datetime.strptime(local_time[:19], '%Y-%m-%d %H:%M:%S')
+    elif not (isinstance(local_time, datetime) or isinstance(local_time, date)):
+        local_time = datetime.now()
+    # 时间转换
+    utc_time = local_time + timedelta(seconds=time.timezone)
+    return utc_time
+
+
+def time_utc_to_local(utc_time):
+    """
+    UTC时间转本地时间
+    :param utc_time:
+    :return:
+    """
+    # 字符串处理
+    if isinstance(utc_time, str) and len(utc_time) == 10:
+        utc_time = datetime.strptime(utc_time, '%Y-%m-%d')
+    elif isinstance(utc_time, str) and len(utc_time) >= 19:
+        utc_time = datetime.strptime(utc_time[:19], '%Y-%m-%d %H:%M:%S')
+    elif not (isinstance(utc_time, datetime) or isinstance(utc_time, date)):
+        utc_time = datetime.utcnow()
+    # 时间转换
+    local_time = utc_time - timedelta(seconds=time.timezone)
+    return local_time
+
 
 db_config = {
     'host': 'localhost',
@@ -68,8 +108,28 @@ def test():
         print e
 
 
+def test_02():
+    table_name = 'user'
+    conn = Mongodb(db_config)
+    print conn.db
+    print conn.find_one(table_name)
+    test_doc = {
+        '_id': 1,
+        'id': 1,
+        'name': 'admin',
+        'create_time': datetime.strptime('2017-07-07 08:00:00', '%Y-%m-%d %H:%M:%S'),
+        'create_time_utc': time_local_to_utc('2017-07-07 08:00:00'),
+        'create_time_str': '2017-07-07 08:00:00'
+    }
+    print conn.remove(table_name)  # 清空记录
+    print conn.insert(table_name, test_doc)  # 插入记录
+    print conn.find_one(table_name)
+    conn.output_rows(table_name)
+
+
 if __name__ == '__main__':
-    test()
+    # test()
+    test_02()
 
 
 """
