Niko_Public/checkIn/functions.py

from django.http import HttpResponse, HttpResponseRedirect
from django.shortcuts import render
import json
import datetime

class DateEncoder(json.JSONEncoder):  
    def default(self, obj):  
        if isinstance(obj, datetime.datetime):  
            return obj.strftime('%Y-%m-%d %H:%M:%S')  
        elif isinstance(obj, date):  
            return obj.strftime("%Y-%m-%d")  
        else:  
            return json.JSONEncoder.default(self, obj)

def qs2json(querySet):
    qdict =[]
    for i in querySet:
        qdict.append(i)
    qJson = {}
    qJson = json.dumps(qdict,cls=DateEncoder)
    print(qJson)
    return qJson

def qs2dict(querySet):
    result = []
    for i in querySet:
        result.append(i)
    return result

# 验证用户登录状态
def checkLoginUser(request):
   try:
       request.session['loginID']
       request.session['loginUser']
       request.session['loginNickname']
       return True
   except:
       return False

# 获取session
def getSession(request, session):
    try:
        sessionName = request.session[session]
        return sessionName
    except:
        return None

# 检验session id的有效性
def checkSession(request, session):
    try:
        sessionName = request.session[session]
        return True
    except:
        return False


def checkGET(request, GET):
    try:
        result = request.GET[GET]
        return True
    except:
        return False

def checkPOST(request, GET):
    try:
        result = request.POST[GET]
        return True
    except:
        return False


# 在mongoDB里面保存签到数据
def mongoAddRecord(checkInID, content):
    from pymongo import MongoClient
    client = MongoClient('localhost', 27017)
    db = client['checkIn']
    collectionName = 'checkIn_%s' % (str(checkInID))
    db[collectionName]
    db[collectionName].insert(content)
    return True

# 在mongoDB里面，根据checkInID找出所有的签到数据
def mongoSearchByID(checkInID):
    from pymongo import MongoClient
    client = MongoClient('localhost', 27017)
    db = client['checkIn']
    collectionName = 'checkIn_%s' % (str(checkInID))
    result = []
    for x in db[collectionName].find():
        del x['_id']
        result.append(x)
    return result

# 验证checkIn的组织者是否为登录的用户
def checkCorrectO_ID(request, ID):
    checkSession(request, 'loginID')
    if ID != getSession(request, 'loginID'):
        return False
    else:
        return True