3类数据库的联动：mysql、mongodb、redis

3类数据库的联动：mysql、mongodb、redis

from pymysql import *
from pymongo import *
from redis import *


class MySQL(object):
    def __init__(self):
        self.con = connect(host='localhost', port=3306,
                           database='stu_info', user='root',
                           password='mysql', charset='utf8')
        self.cur = self.con.cursor()
        self.flag = ["mysql", None]

    def close(self):
        self.cur.close()
        self.con.close()

    def check(self, sid):
        try:
            cur = self.cur
            param = [sid]
            sel_sql = "select name from student where id = %s"
            cur.execute(sel_sql, param)
            res = cur.fetchone()
            if res:
                self.flag[1] = res[0]
        except Exception:
            pass
        finally:
            self.close()
            return self.flag

class Mongo(object):
    def __init__(self):
        self.client = MongoClient(host='localhost', port=27017)
        self.db = self.client.stu_info
        self.col = self.db.student
        self.flag = ["mongo", None]

    def check(self, sid):
        try:
            res = self.col.find_one({"id": sid})
            if res:
                self.flag[1] = res["name"]
        except Exception:
            pass
        finally:
            return self.flag

class Redis(object):
    def __init__(self):
        self.client = StrictRedis()
        self.flag = ["redis", None]

    def check(self, sid):
        try:
            res = self.client.get(sid)
            if res:
                self.flag[1] = res.decode()
        except Exception:
            pass
        finally:
            return self.flag

class Check(object):
    def __init__(self):
        self.msg = ["不在库中", "未找到该学生"]

    def check(self, sid):
        redis = Redis()
        self.msg = redis.check(sid)
        if self.msg[1]:
            return self.msg
        else:
            mongo = Mongo()
            self.msg = mongo.check(sid)
            if self.msg[1]:
                return self.msg
            else:
                mysql = MySQL()
                self.msg = mysql.check(sid)
                if self.msg[1]:
                    return self.msg
                else:
                    self.msg = ["不在库中", "未找到该学生"]
                    return self.msg
def main():
    stu_id = input("请输入要查询的id号：")
    check_id = Check()
    id_msg = check_id.check(stu_id)
    print(id_msg)
if __name__ == '__main__':
    main()