Python中实现验证数据库redis 和 mysql 中的账号密码

# _*_ coding:utf-8 _*_

import sys

import pymysql
import redis

def con_mysql(sql):
    db = pymysql.connect(
        host='127.0.0.1',
        user='root',
        passwd='123456',
        port=3306,
        db='srs',
        charset='utf8')
    cursor = db.cursor()
    data = cursor.execute(sql)
    db.close()
    return data
def con_redis(name, passwd):
    r = redis.Redis(
        host='120.79.46.135',
        port=6379,
        passwd='123456')
    r_name = r.hget('user', 'name')
    r_passwd = r.hget ('user', 'passwd')
    r_name = r_name.decode('utf8')
    r_passwd = r_passwd.decode('utf8')
    if name == r_name and passwd == r_passwd:
        return True, '登录成功'
    else:
        return False, '登录失败'

def main():
    name = sys.argv[1]
    passwd = sys.argv[2]
    result = con_redis(name,passwd)
    if not result[0]:
        sql = '''select * from tb_login where name="%s" and passwd="%s"''' % (name, passwd)
        data = con_mysql(sql)
        if data:
            r = redis.Redis(host='120.79.46.135',port=6379,password='123456')
            r.hset('user', 'name', name)
            r.hset('user', 'passwd', passwd)
            print('redis数据更新成功，账号密码正确)
        else:
            print('账号或者密码错误')
    else:
        print('登录成功')

if __name__ == '__main__':
    main()