# -*- coding: utf-8 -*-
#!/usr/bin/python
# filename: login_admin.py
# codedtime: 2014-9-7 11:26:11

import bottle
import mysql.connector # 導入mysql數據庫連接器

def check_userinfo():
 a_list = [] # 創建一個空列表
 username = bottle.request.GET.get('loginname','').strip() # 用戶名
 password = bottle.request.GET.get('password','').strip() # 密碼
 if username is not None or password is not None:
 try:
 # 連接數據庫 
 conn = mysql.connector.connect(user='root', password='123456', database='myblog') 
 cursor = conn.cursor() # 創建數據游標
 
 # 執行查詢
 query = ("SELECT username, password FROM mb_users "
 "WHERE username=%s and password=%s")
 cursor.execute(query, (username, password))

 a_list = cursor.fetchall() # fetchone獲取一個元組
 #count = int(cursor.rowcount) # 獲取元組個數 
 return a_list

 except mysql.connector.Error as err:
 print("Something went wrong: {}".format(err))
 exit()
 
 finally:
 conn.commit() # 提交修改
 cursor.close() # 關閉數據庫
 conn.close()
 else:
 return a_list

def login_admin():
 if bottle.request.GET.get('bs-submit','').strip(): #點擊登錄按鈕
 a_list = check_userinfo()
 if a_list:
 a_name = a_list[0][0] # 獲得用戶名
 return bottle.template('templates/index_user.tpl', username = a_name)
 else:
 return bottle.template('templates/login_admin.tpl', action='/login_admin', 
 error_info='請輸入正確的用戶名或密碼！')
 else:
 return bottle.template('templates/login_admin.tpl', action='', error_info=' ')