'''
@author ldc
'''
import os
import pymysql
'''
初始化服務(wù)：
1、在mysql上新建一個(gè)數(shù)據(jù)庫(kù)“homework”和建表"t_usr"
2、開啟redis服務(wù)程序
'''
# 建立數(shù)據(jù)庫(kù)連接
conn = pymysql.connect(
  host='localhost',
  user='root',
  password="123456",
  port=3306
)
# 獲取游標(biāo)
cursor = conn.cursor()
# 創(chuàng)建數(shù)據(jù)庫(kù)
dbname = 'homework'
sql='''
   create database if not EXISTS %s charset=utf8;
  '''%dbname
cursor.execute(sql)
# 使用數(shù)據(jù)庫(kù)
cursor.execute('use %s'%dbname)
# 創(chuàng)建表
sql = '''
  create table if not EXISTS t_usr(
     id INTEGER PRIMARY KEY auto_increment,
     username varchar(20) unique not null,
     password varchar(20) not null
    );
'''
cursor.execute(sql)
# 關(guān)閉游標(biāo)與連接
cursor.close()
conn.close()
# 開啟redis服務(wù)，新建一個(gè)啟動(dòng)redisd.bat文件，
#以后開啟redis服務(wù)就可以直接打開這個(gè)文件了
def openRedisd(path):
  rPath = """@echo off
      redis-server %s
      pause"""%path
  with open(r"C:\Users\LDCPC\Desktop\啟動(dòng)redisd.bat","w",encoding="ANSI")
  as f:
   f.write(rPath)
openRedisd(r"D:\ruanjian\redis-64.2.8.2101\redis.windows.conf")
# 打開文件“啟動(dòng)redisd.bat”
os.popen(r"C:\Users\LDCPC\Desktop\啟動(dòng)redisd.bat")

'''
@author ldc
'''
from controller import urls
from model.model import User
from utils.dbUtil import RedisUtil
'''
需求：登錄注冊(cè)驗(yàn)證
1、登錄
2、注冊(cè)
3、改密
4、注銷
'''
# 主界面接口
def index():
  while True:
    #登錄界面
    print("********************************")
    print("*               *")
    print("*  (1) 登錄   (2)注冊(cè)   *")
    print("*  (3) 改密   (4)注銷   *")
    print("*      (5)退出      *")
    print("********************************")
    print()
    num = input("請(qǐng)輸入功能序號(hào)：")
    if num in ['1','2','3','4','5']:
      return num
    else:
      print("輸入有誤，請(qǐng)重新輸入!!!")
# 輸入賬號(hào)與密碼
def inputInfo():
  return input("請(qǐng)輸入賬號(hào)和密碼(逗號(hào)隔開)：").split(',')
if __name__ == '__main__':
  # 連接redis數(shù)據(jù)庫(kù)
  RedisUtil.connect()
  while True:
    # 初始化界面
    num = index()
    # 輸入賬號(hào)密碼
    username, password = inputInfo()
    # 實(shí)例化一個(gè)用戶類
    user = User(username, password)
    if num == '1':
      urls.login(user) #登錄
    elif num == '2':
      urls.regist(user) # 注冊(cè)
    elif num == '3':
      urls.changePasswd(user) # 改密
    elif num == '4':
      urls.deleteUser(user) # 注銷
    else:
      break

'''
@author ldc
'''
from model.model import UserDao
# 先查詢?cè)撚脩羰欠翊嬖跀?shù)據(jù)庫(kù)中
def exists(user):
  '''先查看Redis緩存中是否有該用戶數(shù)據(jù)'''
  if not UserDao.exists(user.username, 'redis'):
   '''然后在mysql中查詢?cè)撚脩羰欠翊嬖?''
   if UserDao.exists(user.username, 'mysql'):
     # 若在mysql存在就把該用戶寫進(jìn)redis,
     UserDao.redis.set(user.username, user.password)
     return 'mysql'
   else :
     return None
  return 'redis'
'''
# 登錄模塊
先在redis上驗(yàn)證，驗(yàn)證成功則提示在redis上驗(yàn)證成功
否則到mysql中驗(yàn)證，驗(yàn)證成功則提示在mysql上驗(yàn)證成功
否則提示用戶不存在
'''
def login(user):
  print("------------登錄界面------------")
  # 查詢?cè)撚脩粜畔⑹欠翊嬖跀?shù)據(jù)庫(kù)中
  whereDB = exists(user)
  if whereDB == 'redis':
   # 匹配密碼是否正確
   if UserDao.query(user, 'redis') == user.password:
     print("[在redis中查詢到該用戶]登錄成功!!!")
     return 1
   else:
     print("[在redis中查詢到該用戶] 登錄失敗,用戶名或者密碼不正確!!!")
  elif whereDB == 'mysql':
   # 匹配密碼是否正確
   if UserDao.query(user, 'mysql'):
     print("[在mysql中查詢到該用戶] 登錄成功!!!")
     return 1
   else:
     print("[在mysql中查詢到該用戶] 登錄失敗,用戶或者密碼不正確!!!")
  else:
   print("[在mysql中查詢不到該用戶]登錄失敗,該用戶不存在，請(qǐng)注冊(cè)后再登錄!!!")
  return 0
'''
# 注冊(cè)模塊
先在redis上查詢賬號(hào)是否存在，存在則注冊(cè)失敗
否則到mysql上查詢，用戶存在則注冊(cè)失敗
否則注冊(cè)成功，把賬號(hào)寫進(jìn)mysql，寫進(jìn)redis
'''
def regist(user):
  print("------------注冊(cè)界面------------")
  # 查詢?cè)撚脩粜畔⑹欠翊嬖跀?shù)據(jù)庫(kù)中
  whereDB = exists(user)
  if whereDB :
   print("注冊(cè)失敗，該用戶已存在!!!")
  else:
   if UserDao.insert(user):
     print("注冊(cè)成功!!!")
   else:
     print("注冊(cè)失敗!!!")
'''
# 修改密碼模塊
先在redis上和mysql上查詢，用戶存在就在mysql上修改該用戶密碼，
然后把該用戶信息重新寫進(jìn)redis中
在mysql中查詢不到該用戶，就返回該用戶不存在，改密失敗
'''
def changePasswd(user):
  print("------------改密界面------------")
  # 查詢?cè)撚脩粜畔⑹欠翊嬖跀?shù)據(jù)庫(kù)中
  whereDB = exists(user)
  if whereDB:
   user.password = input("請(qǐng)輸入新密碼：")
   if UserDao.changePasswd(user):
     print("改密成功!!!")
   else:
     print("改密失敗!!!")
  else:
   print("用戶不存在，改密失敗!!!")
'''
# 注銷用戶模塊
先在在redis上和mysql上查詢，用戶存在就在mysql和redis上刪除該用戶
在mysql中查詢不到該用戶，就返回該用戶不存在，注銷失敗
'''
def deleteUser(user):
  print("------------注銷界面------------")
  # 查詢?cè)撚脩粜畔⑹欠翊嬖跀?shù)據(jù)庫(kù)中
  if login(user):
   if UserDao.deleteUser(user):
     print("注銷成功!!!")
     return
  print("注銷失敗!!!")

'''
@author ldc
'''
from utils.dbUtil import RedisUtil, MySQLUtil
# 用戶模型類
class User:
  def __init__(self,username,password):
    self.username = username
    self.password = password
# UserDao
# 封裝了對(duì)User數(shù)據(jù)的增刪改查
# Dao=Database Access Object 數(shù)據(jù)庫(kù)訪問對(duì)象
class UserDao:
  # 創(chuàng)建數(shù)據(jù)庫(kù)對(duì)象
  redis = RedisUtil()
  mySQL = MySQLUtil('homework','t_usr')
  # 執(zhí)行數(shù)據(jù)庫(kù)查詢操作，返回查詢結(jié)果
  @classmethod
  def query(cls,user,dbType):
    dataDict = {}
    dataDict["username"] = user.username
    dataDict["password"] = user.password
    if dbType == 'redis':
      return cls.redis.get(user.username)
    elif dbType == 'mysql':
      return cls.mySQL.query(dataDict)
  # 執(zhí)行數(shù)據(jù)庫(kù)查詢操作，查詢用戶是否存在，返回查詢結(jié)果
  @classmethod
  def exists(cls,username,dbType):
    dataDict = {}
    dataDict["username"] = username
    if dbType == 'redis':
      return cls.redis.exists(username)
    elif dbType == 'mysql':
      return cls.mySQL.exists(dataDict)
    else:
      pass
  # 執(zhí)行數(shù)據(jù)插入操作，先把用戶信息添加進(jìn)mysql，然后再添加進(jìn)redis
  @classmethod
  def insert(cls, user):
    dataDict = {}
    dataDict["username"] = user.username
    dataDict["password"] = user.password
    if cls.mySQL.insert(dataDict):
      cls.redis.set(user.username,user.password)
      return 1
    else:
      print("注冊(cè)失敗，服務(wù)器繁忙!!!")
      return 0
  # 修改密碼
  @classmethod
  def changePasswd(cls, user):
    dataDict = {'changeCol': 'password = %s'%user.password,
     'caluse' : 'username = %s'%user.username}
    if cls.mySQL.update(dataDict):
      cls.redis.set(user.username,user.password)
      return 1
    else:
      print("修改密碼失敗，服務(wù)器繁忙!!!")
      return 0
  # 注銷用戶
  @classmethod
  def deleteUser(cls, user):
    dataDict = {'username' : user.username}
    if cls.mySQL.delete(dataDict):
      cls.redis.delete(user.username)
      return 1
    else:
      print("修改密碼失敗，服務(wù)器繁忙!!!")
      return 0

'''
@author ldc
'''
import pymysql
import redis as redis
'''
MySQL增刪改查操作類
'''
class MySQLUtil:
  def __init__(self,dbName,tableName):
   self.dbName = dbName
   self.tableName = tableName
  # 連接數(shù)據(jù)庫(kù),并生成全局可用的連接對(duì)象和查詢游標(biāo)
  def connect(self):
   self.conn = pymysql.connect(
     host='localhost', user='root', password="123456",
     database=self.dbName, port=3306,
   )
   self.cursor = self.conn.cursor()
  # 關(guān)閉全局游標(biāo)，斷開全局連接
  def disconnect(self):
   self.cursor.close()
   self.conn.close()
  # 查詢用戶名是否存在
  def exists(self,dataDict):
   caluse = ''
   for key,value in dataDict.items():
     caluse += key + '="'+ value + '"'
   # print(caluse)
   sql = """
      select * from %s where %s ;
      """ % (self.tableName, caluse)
   return self.execute(sql)
  # 驗(yàn)證用戶名和密碼是否正確
  def query(self, dataDict):
   # 查詢子條件拼接
   caluse = ''
   for key, value in dataDict.items():
     caluse += key + '="' + value + '" and '
   caluse = caluse[:-4]
   # print(caluse)
   sql = """
      select * from %s where %s;
      """% (self.tableName, caluse)
   return self.execute(sql)
  # 添加新用戶
  def insert(self, dataDict):
   # sql語(yǔ)句拼接
   columns = ''
   values = ''
   for key, value in dataDict.items():
     columns += key + ','
     values += '"' + value + '",'
   columns = columns[:-1]
   values = values[:-1]
   sql = """
      insert into %s (%s) VALUES (%s);
      """ % (self.tableName, columns,values)
   # print(sql)
   return self.execute(sql)
  # 更新
  def update(self, dataDict):
   # sql語(yǔ)句拼接
   changeCol = dataDict['changeCol'] #要改變值的列名
   caluse = dataDict['caluse'] #要改變值的子條件
   sql = 'update %s set %s where %s'%(self.tableName,changeCol,caluse)
   return self.execute(sql)
  # 刪除
  def delete(self, dataDict):
   # sql語(yǔ)句拼接
   caluse = ''
   for key,value in dataDict.items():
     caluse += key + '="' + value + '"'
   sql = """
      delete from %s where %s;
      """ % (self.tableName,caluse)
   # print(sql)
   return self.execute(sql)
  # print(sql)
  # 執(zhí)行sql語(yǔ)句
  def execute(self, sql):
   self.connect()
   affected = 0
   try:
     affected = self.cursor.execute(sql)
   except BaseException as e:
     print(e)
     affected = 0
   finally:
     self.conn.commit()
     self.disconnect()
     return affected
'''
redis增刪改查操作類
'''
class RedisUtil:
  # redis連接
  @classmethod
  def connect(cls):
   cls.client = redis.Redis(
     host='localhost', port=6379,
     db=1, password='123456',
   )
  # 判斷鍵是否存在
  @classmethod
  def exists(cls,key):
   return cls.client.exists(key)
  # 存儲(chǔ)鍵值,
  @classmethod
  def set(cls,key,value):
   # 鍵值存儲(chǔ)在緩存中，保留時(shí)間為30秒
   cls.client.setex(key,value,30)
  # 獲取鍵值
  @classmethod
  def get(cls,key):
   res = cls.client.get(key).decode("utf-8")
   return res
  # 刪除鍵值
  def delete(cls, key):
   cls.client.delete(key)