pymysql实现用户注册登录功能

#-*- coding:utf8 -*-  
  2 import pymysql

def print_tips(str_info):   #blue
    print '33[1;34m%s33[0m'%str_info

def print_info(str_info):   #green
    print '33[1;32m%s33[0m'%str_info

def print_warn(str_info):   #yellow
    print '33[1;33m%s33[0m'%str_info

def print_err(str_info):    #red
    print '33[1;31m%s33[0m'%str_info


###数据库连接
def func_connect_mysql(host,user,db_name,db_pwd):
    message = {
            "host":host,
            "user":user,
            "password":db_pwd,
            "database":db_name,
            "autocommit":True
            }
    db = pymysql.connect(**message)
    return db



# 4 success
# 1 goto
###登陆函数
def func_login(cursor):
    print('33[1;35mLogin33[0m')

    flag_login = 0
    flag_registration = 0
    flag_find_pwd = 0

    while True:
        user_name = raw_input("33[1;32musername:33[0m")
        #print type(user_name)
        user_pwd = input("33[1;32mpassword:33[0m")
        sql = "select * from users where username = '%s' and password = '%s'"%(user_name,user_pwd)
        result = cursor.execute(sql)
 
        if result:
            print_tips("welcome %s,login successful"%user_name)
            flag_login = 4
            break
        else:
            sql_2 = "select * from users where username = '%s'"%user_name
            result2 = cursor.execute(sql_2)
            if result2:
                print_err("password error!")
        
                while True:
                    print_warn("if you want to try again,enter 1")
                    print_warn("if you want to find password,enter 2")
                    num = input("33[1;32m1/2:33[0m")
                    if num == 1:
                        flag_login = 1
                        break
                    elif num == 2:
                        flag_login = 2
                        break
                    else:
                        print_err("Input error!please input 1 or 2")
                        continue
                
                if flag_login == 1:
                    continue
                else:
                    flag_find_pwd = 1
                    break
            
            else:
                print_err("user not exists!")
                while True:
                    print_warn("if you want to try again,enter 1")
                    print_warn("if you want to Registration,enter 2")
                    num = input("33[1;32m1/2:33[0m")
                    if num == 1:
                        flag_login = 1
                        break
                    elif num == 2:
                        flag_registration = 1
                        break
                    else:
                        print_err("please enter 1 or 2")
                        continue
                break

    return flag_login,flag_registration,flag_find_pwd,user_name

###找回密码函数
def func_find_pwd(user_name,cursor):
    print('33[1;35mFind password33[0m')
    flag_login = 0
    flag_find_pwd = 0
    flag_registration = 0
    password = ""

    while True:
        phone_number = input('33[1;32mplease Enter your phone number:33[0m')
        sql = "select password from users where username = '%s' and phone = '%s'"%(user_name,phone_number)
        result = cursor.execute(sql)
        if result:
            password = cursor.fetchone()
            flag_login = 1
            break
        else:
            print_err("phone number error!")
            while True:
                print_warn("Enter 1 try again
Enter 2 go to Registration")
                num = input('33[1;32m1/2:33[0m')
                if num == 1:
                    flag_find_pwd = 1
                    break
                elif num == 2:
                    flag_find_pwd = 2
                    break
                else:
                    continue
            if flag_find_pwd == 1:
                continue
            elif flag_find_pwd == 2:
                flag_registration = 1
                break

    return flag_login,flag_registration,password


###注册函数
def func_registration(db,cursor):
   #print('33[1;35mRegistration33[0m')
    flag_login = 0
    while True:
        #username:
        print('33[1;35mRegistration33[0m')
        username = raw_input('33[1;32mEnter username:33[0m')
        sql1 = "select * from users where username = '%s'"%username
        result1 = cursor.execute(sql1)
        if result1:
            print_err("user exists!")
            print_warn("Enter 1 try again
Enter 2 go to login")
            num = input('33[1;32m1/2:33[0m') 
            if num == 1:
                continue
            else:
                flag_login = 1
                break

        #password
        password = raw_input('33[1;32mEnter password:33[0m')
        password1 = raw_input('33[1;32mEnter password again:33[0m')
        if password1 != password:
            print_err("Passwords are inconsistent!")
            continue
        else:

        #phone
            phone_number = input('33[1;32mEnter phone:33[0m')

        #registration
            sql = "insert into users(username,password,phone) values ('%s','%s','%s');"%(username,password,phone_number)
            result = cursor.execute(sql)
            #db.commit()
            data = cursor.fetchall()
            if result:
                print_tips("registration successful!")
                flag_login = 1
                break
            else:
                print_err("registration fail!")
                print warn("Try again ,Y/N")
                num = input("33[1;32mY/N:33[0m")
                if num =="Y":
                    continue
                else:
                    break
    return flag_login 


###修改用户信息函数
def change_message(num,id_primary,cursor):
    flag = False
    flag_run = False
    if num == 1:
        print("33[1;35malter password33[0m")
        print_warn("please enter your new password")
        new_password = raw_input("33[1;32mnew password:33[0m")
        sql1 = "select password  from users where id = '%d'"%id_primary
        row = cursor.execute(sql1)
        old_password = cursor.fetchone()[0]
        if old_password != new_password:
            sql = "update users set password = '%s' where id = '%d'"%(new_password,id_primary)
            flag_run = True
        else:
            print_err("new password = old password!")

    elif num == 2:
        print("33[1;35malter username33[0m")
        print_warn("please enter your new username")
        new_username = raw_input("33[1;32mnew username:33[0m")
        sql1 = "select username from users where id = '%d'"%id_primary
        row = cursor.execute(sql1)
        old_username = cursor.fetchone()[0]
        if old_username != new_username:
            sql = "update users set username = '%s' where id = '%d'"%(new_username,id_primary)
            flag_run = True
        else:
            print_err("new username = old username!")
    elif num == 3:
        print("33[1;35malter phone33[0m")
        print_warn("please enter your new phone")
        new_phone = raw_input("33[1;32mnew phone:33[0m")
        sql1 = "select phone from users where id = '%d'"%id_primary
        row = cursor.execute(sql1)
        old_phone = cursor.fetchone()[0]
        if old_phone != new_phone:
            sql = "update users set phone = '%s' where id = '%d'"%(new_phone,id_primary)
            flag_run = True
        else:
            print_err("new phone = old phone!")
    elif num == 4:
        print("33[1;35mdelete account33[0m")
        while True:
            print_warn("confirm deletion,enter 1
think again,enter 2")
            num_delete = input("33[1;32m1/2:33[0m")
            if num_delete == 1:
                sql = "delete from users where id = '%d'"%id_primary
                flag_run = True
                break
            elif num_delete == 2:
                break
            else:
                print_err("please enter 1/2!")
                continue
    if flag_run:
        result = cursor.execute(sql)
        if result:
            flag = True
    return flag



###主流程函数
def func_main(db):
    cursor = db.cursor()
    cursor.execute("create table if not exists users(id int primary key auto_increment,username char(10) unique,password char(20),phone char(11))")
    while True:
        login_flag_login = -1
        login_flag_registration = -1
        login_flag_find_pwd = -1
        pwd_flag_login = -1
        pwd_flag_registration = -1
        password = ""
        registration_flag_login = -1
        id_primary = -1
        print_warn("if you want to login,enter 1")
        print_warn("if you want to registration,enter 2")
        print_warn("if you want to quit,enter 5")
        num_login = input("33[1;32m1/2/5:33[0m")
        if num_login != 1 and num_login != 2 and num_login != 5:
            print_err("please enter 1 or 2 or 5!")
            continue
        if num_login == 5:
            break
        elif num_login == 1:
            login_flag_login,login_flag_registration,login_flag_find_pwd,user_name = func_login(cursor)
            if login_flag_login == 4:
                while True:
                    print_warn("if you want to alter password,enter 1")
                    print_warn("if you want to alter username,enter 2")
                    print_warn("if you want to alter phone,enter 3")
                    print_warn("if you want to delete your acount,enter 4")
                    print_warn("if you want to quit,enter 5")
                    num = input('33[1;32m1/2/3:33[0m')
                    sql = "select id from users where username = '%s'"%user_name
                    row = cursor.execute(sql)
                    if row:
                        id_primary = cursor.fetchone()[0]
                    if num == 1 or num == 2 or num == 3:
                        if id_primary != -1:
                            flag = change_message(num,id_primary,cursor)
                            if flag:
                                print_tips("alter success!")
                                break
                            else:
                                print_err("sorry,alter fail,please try again")
                                continue
                    elif num == 4:
                        if id_primary != -1:
                            flag = change_message(num,id_primary,cursor)
                            if flag:
                                print_tips("delete success!")
                                num_login == 1
                                break
                            else:
                                print_err("sorry,delete fail,please try again")
                                continue
                    elif num == 5:
                        break
                    else:
                        print_err("please enter 1/2/3/4")
                        continue
            if login_flag_find_pwd == 1:
                pwd_flag_login,pwd_flag_registration,password = func_find_pwd(user_name,cursor)
                if pwd_flag_login == 1 and password != "":
                    print_tips("password:"+password[0])
                    continue
        if pwd_flag_registration == 1 or login_flag_registration == 1 or num_login == 2:
            registration_flag_login = func_registration(db,cursor)
            if registration_flag_login == 1:
                continue
            else:
                print_err("Registration Failed!")


if __name__=='__main__': 
    db = func_connect_mysql("localhost","root","Student_base","123456")
    func_main(db)