from pymongo import MongoClient
from bson import ObjectId, json_util
# 主环境 (生产环境为production，开发环境为development)
ENV = 'production'

def col(arg):
    """
    获取数据集
    """
    # 链接数据库
    conn = MongoClient('mongodb://cherry:fR1jW2xG3bE9@39.96.28.83:27017/cherry')
    # 判断环境
    if ENV == 'development':
        arg += '_test'
    return conn.cherry[arg]

def updateCookie(new_cookie):
    """
    更新cookie
    """
    # 字符串化
    new_cookie = str(new_cookie)
    try:
        col('config').update({'key': 'cookie'}, {'$set': {'value': new_cookie}}, {'upsert': 'true'})
    except Exception as e:
        print(e)
        return 'cookie数据库更新失败', 400

def findCookie():
    """
    获取cookie
    """
    try:
        res = col('config').find_one({'key': 'cookie'}, {'_id': 0})
        value = res['value']
        cookie = eval(value)
        return cookie, 200
    except Exception as e:
        print(e)
        return 'cookie数据库查询失败', 300

def updateLastUpdate(sid, now):
    """
    更新最后一次课程更新信息
    """
    try:
        col('user').update({'sid': sid}, {'$set': {'last_update': now}})
    except Exception as e:
        print(e)
        return '数据更新时间数据库更新失败', 403
def insertUser(userinfo):
    """"
    插入新学生信息
    """
    try:
        col('user').insert_one(userinfo)
        return 'OK', 200
    except Exception as e:
        print(e)
        return '学生信息数据库插入失败', 100

def findUser(cid):
    """
    获取学生信息
    在group_list里边存放了[{'group_id'}]
    取出之后需要遍历组的信息
    """
    try:
        userinfo = col('user').aggregate([
            {
                '$match': {
                    'cid': cid
                }
            },
            {
                '$lookup': {
                    'from': 'link',
                    'localField': 'sid',
                    'foreignField': 'sid',
                    'as': 'group_list'
                }
            },
            {
                '$project': {
                    '_id': 0,
                    'group_list._id': 0,
                    'group_list.sid': 0
                }
            }
        ])
        return userinfo, 200
    except Exception as e:
        print(e)
        return '学生信息数据库查询失败', 301

def insertInvite(cid, group_id):
    """
    对用户添加用户组邀请
    测试：多次重复邀请，用户接受或者删除的时候是否能删掉所有，前端Set去重
    """
    try:
        col('user').update({'cid': cid}, {'$push': {'invite_list': group_id}})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户组邀请数据库插入失败', 101

def deleteInvite(cid, group_id):
    """
    用户或者管理员删除用户组邀请
    """
    try:
        col('user').update({'cid': cid}, {'$pull': {'invite': {'group_id': group_id}}})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户组邀请数据库删除失败', 500

def bindUserGroup(sid, group_id):
    """
    绑定用户以及用户组
    """
    try:
        col('link').insert_one({'sid': sid, 'group_id': group_id})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户与用户组绑定数据库插入失败', 102

def unbindUserGroup(sid, group_id):
    """
    解绑用户以及用户组
    """
    try:
        col('link').remove({'sid': sid, 'group_id': group_id})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户与用户组解绑数据库删除失败', 501

def updateAvatar(cid, img_id):
    """
    用户更新头像
    """
    try:
        col('user').update({'cid': cid}, {'$set': {'avatar': img_id}})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '头像数据库更新失败', 401

def updateBg(cid, img_id):
    """
    用户更新背景图片
    """
    try:
        col('user').update({'cid': cid}, {'$set': {'setting.bg': img_id}})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '背景图数据库更新失败', 402

def insertGroup(group_info):
    """
    用户创建用户组
    """
    try:
        col('group').insert_one(group_info)
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户组数据库插入失败', 103

def findGroup(group_id):
    """
    查询用户组信息，附带用户组中用户信息
    """
    try:
        groupinfo = col('group').aggregate([
            {
                '$match': {
                    'group_id': group_id
                }
            },
            {
                '$lookup': {
                    'from': 'link',
                    'localField': 'group_id',
                    'foreignField': 'group_id',
                    'as': 'user_list'
                }
            },
            {
                '$lookup': {
                    'from': 'user',
                    'localField': 'cid',
                    'foreignField': 'cid',
                    'as': 'user_list'
                }
            },
            {
                '$project': {
                    '_id': 0,
                    'user_list._id': 0,
                    'user_list.invite_list': 0,
                    'user_list.cid':0,
                    'user_list.pwd':0,
                    'user_list.setting':0,
                    'user_list.last_update': 0,
                }
            }
        ])
        return groupinfo, 200
    except Exception as e:
        print(e)
        return '用户组信息数据库查询失败', 302

def deleteGroup(group_id):
    """
    解散用户组
    """
    try:
        col('group').remove({'group_id': group_id})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '解散用户组数据库删除失败', 504

def addLog(group_id, log):
    """
    向用户组中添加操作记录
    """
    try:
        col('group').update({'group_id': group_id}, {'$push': {'log_list': log}})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '操作记录数据库插入失败', 104

def groupInsertAdmin(cid, group_id):
    """
    向管理组中添加管理
    """
    try:
        col('group').update({'group_id': group_id}, {'$push': {'admin_list': cid}})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户组侧加入管理数据库插入失败', 106

def groupDeleteAdmin(cid, group_id):
    """
    从管理组中删除管理
    """
    try:
        col('group').update({'group_id': group_id}, {'$pull': {'admin_list': cid}})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户组侧移除管理数据库删除失败', 503

def userInsertCrouse(crouse):
    """
    用户添加自定义课程
    """
    try:
        col('user_crouse').insert_one(crouse)
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户自定义课程数据库插入失败', 107

def userDeleteCrouse(crouse_id):
    """
    用户删除自定义课程
    加入是否自定义的判断，防止用户改前端导致删除非定义课程
    """
    try:
        col('user_crouse').remove({'crouse_id': crouse_id, 'is_personal': False})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户自定义课程数据库删除失败', 505

def userDeleteAllCrouse(sid):
    """
    用户删除所有课程（刷新课表用）
    不删除自定义课程
    """
    try:
        col('user_crouse').remove({'sid': sid, 'is_personal': False})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户所有课程数据库删除失败', 502

def userInsertAllCrouse(crouses):
    """
    用户批量添加课表（刷新课表用）
    """
    try:
        col('user_crouse').insert_many(crouses)
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户所有课程数据库插入失败', 108

def insertRank(cid, nick, count, time):
    """
    向排名表里增加或者覆写数据
    """
    try:
        col('rank').update({"cid": cid}, {'$setOnInsert': {"nick": nick}, '$set': {
            "count": count, "time": time}}  , {'upsert': 'true'})
        # col('rank').insert_one({"count":count,"time":time,"nick":nick})
        return 'OK', 200
    except Exception as e:
        # 失败
        return '排名表数据库插入失败', 109

def findRank():
    """
    获取所有排名信息
    """
    time_rank = []
    count_rank = []
    try:
        for i in col('rank').find({}, {"_id": 0}).sort([("time", 1), ("count", 1)]).limit(10):
            time_rank.append(i)
        for i in col('rank').find({}, {"_id": 0}).sort([("count", 1), ("time", 1)]).limit(10):
            count_rank.append(i)
        return {'time_rank': time_rank, 'count_rank': count_rank}, 200
    except Exception as e:
        print(e)
        return '排名表数据库查询失败', 303

def findUserCrouse(sid):
    """
    获取所有用户课程
    """
    crouse_list = []
    try:
        for i in col('user_crouse').find({'sid': sid}, {'_id': 0}):
            crouse_list.append(i)
        return crouse_list, 200
    except Exception as e:
        print(e)
        return '用户课程数据库查询失败', 304

def groupInsertCrouse(crouse):
    """
    用户组添加课程
    """
    try:
        col('group_crouse').insert_one(crouse)
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户组课程数据库插入失败', 110

def groupDeleteCrouse(crouse_id):
    """
    用户组删除课程
    """
    try:
        col('group_crouse').remove({'crouse_id': crouse_id})
        return 'OK', 200
    except Exception as e:
        print(e)
        return '用户组课程数据库删除失败', 506

def findGroupCrouse(group_id):
    """
    获取所有指定用户组课程
    """
    crouse_list = []
    try:
        for i in col('group_crouse').find({'group_id': group_id}):
            crouse_list.append(i)
        return crouse_list, 200
    except Exception as e:
        print(e)
        return '用户组课程数据库查询失败', 305

def insertLog(log):
    """
    插入操作记录
    """
    try:
        col('log').insert_one(log)
        return 'OK', 200
    except Exception as e:
        print(e)
        return '系统操作记录数据库插入失败', 111

def findLog(has_read_page):
    """
    查询操作记录，默认50条分割
    """
    skip = 50 * (has_read_page - 1)
    log_list = []
    try:
        for i in col('log').find({}).limit(50).skip(skip):
            log_list.append(i)
        return log_list, 200
    except Exception as e:
        print(e)
        return '系统操作记录数据库查询失败', 306