python学习之成员信息增删改查

主要实现了成员信息的增加，修改，查询，和删除功能，写着玩玩，
在写的过程中，遇到的问题，旧新成员信息数据的合并，手机号和邮箱的验证，

#!/usr/bin/env python
# coding=utf8
#author:shantuwqk@163.com

import os, sys, time,json
import re

member_dict = \
    {}
member_name_list = []
member_list = []
def handler_member_storage(name,passwd,mobile,email):

    member_detailed = {}
    member_detailed['name'] = name
    member_detailed['passwd'] = passwd
    member_detailed['mobile'] = mobile
    member_detailed['email'] = email
    member_list.append(member_detailed)
    member_name_list.append(name)
    member_dict['namelist'] = member_name_list
    member_dict['namedetail'] = member_list
    #with open('member_info.json','w') as f:
    #   json.dump(member_dict,f)
    return member_dict

def handel_query_user(x):
    """
    判断x如果是str，查询name字段，如果是int,查询手机号字段
    :param x:
    :return:
    """
    #print "字符串",isinstance(x,str)
    #print "数字",isinstance(x,int)
    if isinstance(x,str):
        if x in load_member_data()['namelist']:
            print "精确匹配查询用户[%s]信息....."%(x)
            print "==================================================="
            user_index = load_member_data()['namelist'].index(x)
            user_detail = load_member_data()['namedetail'][user_index]
            for k,v in user_detail.items():
                print k,v
            print "----------------------------------------------------"
        else:
            print "\033[33;1m用户名或密码不在本数据库，你是不是记错了?您再想想!!\033[0m"

    if isinstance(x,int):
        if check_mobileORmail_member(str(x)) != "mobile_true":
            print "这是手机号吗,打打试试...."
        else:
            print "精确匹配查询用户手机[%s]信息......"%(x)
            print "===================================================="
            user_detail = load_member_data()['namedetail']
            for user_dict in user_detail:
                if x == int(user_dict['mobile']):
                    for k,v in user_dict.items():
                        print k,v
                    print "------------------------------------------------"

def load_member_data():
    with open("member_info.json",'r') as f:
        try:
            member_info_dict = json.load(f)
            return member_info_dict
        except ValueError:
            #print "\033[33;1m没有数据可加载\033[0m"
            return False

def check_login(user,passwd):
    """
    验证登录用户名和密码是否正确
    :param user:
    :param passwd:
    :return:
    """
    with open("member_info.json",'r') as f:
        member_info_dict = json.load(f)
        #print member_info_dict
        if user in member_info_dict['namelist']:
            user_index = member_info_dict['namelist'].index(user)
            user_detail = member_info_dict['namedetail'][user_index]
            #print user_index,user_detail
            if passwd == user_detail['passwd'] and user == user_detail['name']:
                print "\033[32;1m输入用户[%s]密码[%s]正确\033[0m"%(user,passwd)
                return True
            else:
                print "\033[31;1m 您输入的密码不正确\033[0m"
        else:
            print "\033[31;1m你查询的用户不存在\033[0m"

def check_mobileORmail_member(x):
    p=re.compile(r'^[\w\d]+[\d\w\_\.]+@([\d\w]+)\.([\d\w]+)(?:\.[\d\w]+)?$|^(?:\+86)?(\d{3})\d{8}$|^(?:\+86)?(0\d{2,3})\d{7,8}$')
    m = p.match(x)
    if m == None:
        print "\033[33;1mmail or mobile number is wrong!!\033[0m"
        return False
    else:
        if m.group(1)!=None:
            if m.group(1) == 'vip':
                print 'It is %s mail,the address is %s' %(m.group(2),m.group())
                return "mail_true"
            else:
                print 'It is %s mail,the address is %s' %(m.group(1),m.group())
                return "mail_true"
        else:
            if m.group(3)!=None:
                print 'It is mobilephone number,the number is %s' %m.group()
                return "mobile_true"
            else:
                print 'It is telephone number,the number is %s' %m.group()
                return True

class Modify_User_Info:
    """
    修改用户的相关信息
    """
    def update_passwd(self,user,newpwd):
        if user in load_member_data()['namelist']:
            user_index = load_member_data()['namelist'].index(user)
            user_detail = load_member_data()['namedetail'][user_index]
            user_detail['passwd'] = newpwd
            newpwd_dict = load_member_data()
            newpwd_dict['namedetail'][user_index] = user_detail
            return newpwd_dict
        else:
            print "你有没有搞错！你输入的用户不在数据库"

    def update_mobile(self,user,newmobile):
        if user in load_member_data()['namelist']:
            user_index = load_member_data()['namelist'].index(user)
            user_detail = load_member_data()['namedetail'][user_index]
            user_detail['mobile'] = newmobile
            newpwd_dict = load_member_data()
            newpwd_dict['namedetail'][user_index] = user_detail
            return newpwd_dict
        else:
            print "你有没有搞错！你输入的用户不在数据库"

    def update_email(self,user,newmail):
        if user in load_member_data()['namelist']:
            user_index = load_member_data()['namelist'].index(user)
            user_detail = load_member_data()['namedetail'][user_index]
            user_detail['email'] = newmail
            newpwd_dict = load_member_data()
            newpwd_dict['namedetail'][user_index] = user_detail
            return newpwd_dict
        else:
            print "你有没有搞错！你输入的用户不在数据库"

def del_user_info(user):
    if user in load_member_data()['namelist']:
        user_index = load_member_data()['namelist'].index(user)
        user_detail = load_member_data()['namedetail'][user_index]
        new_del_dict = load_member_data()
        new_del_dict['namedetail'].remove(user_detail)
        new_del_dict['namelist'].remove(user)
        return new_del_dict

    else:
        print "你要删除的用户[%s]不在数据库里"%user

if __name__ == "__main__":
    Operation_type = {
        '1':'Add user information',
        '1.1':'name,passwd,mobile,email',
        '2':'Query the user information',
        '3':'Modify the user information',
        '4':'Delete user information',
        '5':'logout'
    }

    while True:
        print """
\033[32;1m********************************************************************************
Welcome to login user information center, you can make the following operation:
        1,%s
        2,%s
        3,%s
        4,%s
        5,%s
        What you need to operate?
********************************************************************************\033[0m
        """%(Operation_type['1'],Operation_type['2'],Operation_type['3'],Operation_type['4'],Operation_type['5'])
        option = raw_input("Choice is?>>")

        if option == '1':
            print "\033[32;1m you have choice is [%s]\033[0m"%Operation_type['1']
            print "You need to complete these input according to [%s] the guidelines"%Operation_type['1.1']
            while True:
                name = raw_input("name:")
                passwd = raw_input("passwd:")
                mobile = raw_input("mobile:")
                email = raw_input("email:")
                if len(name) == 0 or len(passwd) == 0 or len(mobile) == 0 or len(email) == 0:
                    print "\033[33;1mThe user information you entered any item can't be empty or incorrect.Please fill in again!!\033[0m"
                    continue
                if check_mobileORmail_member(mobile) != "mobile_true" or check_mobileORmail_member(email) != "mail_true":
                    continue

                #print "#########3",name,passwd,mobile,email

                print """\033[32;1m 您输入的信息如下:
                        name:%s
                        passwd:%s
                        mobile:%s
                        email:%s
                    \033[0m"""%(name,passwd,mobile,email)
                s1 = raw_input("y 保存/w 重新填写>>")
                if s1 == "y":

                    old_user_dict = load_member_data()
                    new_user_dict = handler_member_storage(name,passwd,mobile,email)
                    #print "############33",old_user_dict,'\n',new_user_dict
                    #合并旧的和新的用户数据
                    if old_user_dict == False:
                        print "没有旧数据可加载，直接写入新数据"
                        with open('member_info.json','w') as f:
                            json.dump(new_user_dict,f)
                        s11 = raw_input("t 继续添加/b 返回上一级/q退出操作?>>")
                        if s11 == "t":
                            continue
                        elif s11 == "b":
                            break
                        elif s11 == "q":
                            sys.exit(0)
                    else:
                        for user in new_user_dict['namelist']:
                            old_user_dict['namelist'].append(user)
                        for user_detail in new_user_dict['namedetail']:
                            old_user_dict['namedetail'].append(user_detail)
                        #print old_user_dict,
                        with open('member_info.json','w') as f:
                            json.dump(old_user_dict,f)
                        print "保存成功"
                        s12 = raw_input("t 继续添加/b 返回上一级/q退出操作?>>")
                        if s12 == "t":
                            continue
                        elif s12 == "b":
                            break
                        elif s12 == "q":
                            sys.exit(0)
                elif s1 == "w":
                    continue
                else:
                    pass

        elif option == '2':
            print "\033[32;1m you have choice is [%s]\033[0m"%Operation_type['2']
            print "\033[33;1m友情提示:输入正确的用户名和密码才能查询用户信息\033[0m"
            luser = raw_input("用户名:").strip()
            lpasswd = raw_input("用户密码:").strip()
            if check_login(luser,lpasswd) == True:
                print "用户验证通过"
                print "请输入需要查询的用户名或手机号[目前只支持精确查询]"
                while True:
                    query_info = raw_input("查询:>>").strip()
                    if check_mobileORmail_member(query_info) == "mobile_true":
                        handel_query_user(int(query_info))
                    else:
                        handel_query_user(query_info)
                    s2 = raw_input("b 返回上一级/t 继续查询")
                    if s2 == "b":
                        break
                    elif s2 == "t":
                        continue

            else:
                print "用户名或密码不正确,不能查询其它人信息"

        elif option == '3':
            print "\033[32;1m you have choice is [%s]\033[0m"%Operation_type['3']
            print "友情提示:目前支持修改用户的密码,手机号,邮箱地址"
            modify = Modify_User_Info()
            modify_option = raw_input("p 修改密码/m 修改手机号/e 修改邮箱地址")
            if modify_option == "p":
                user_name = raw_input("用户名:").strip()
                user_pwd = raw_input("新密码:").strip()
                print "开始修改用户[%s]新的密码为[%s]....."%(user_name,user_pwd)
                update_dict = modify.update_passwd(user_name,user_pwd)
                with open('member_info.json','w') as f:
                    json.dump(update_dict,f)
                print "密码修改成功!"
            elif modify_option == "m":
                user_name = raw_input("用户名:").strip()
                user_mobile = raw_input("新手机号:").strip()
                print "开始修改用户[%s]新的手机号为[%s]....."%(user_name,user_mobile)
                update_dict = modify.update_mobile(user_name,user_mobile)
                with open('member_info.json','w') as f:
                    json.dump(update_dict,f)
                print "手机号修改成功!"
            elif modify_option == "e":
                user_name = raw_input("用户名:").strip()
                user_mail = raw_input("新邮箱:").strip()
                print "开始修改用户[%s]新的邮箱为[%s]....."%(user_name,user_mail)
                update_dict = modify.update_email(user_name,user_mail)
                with open('member_info.json','w') as f:
                    json.dump(update_dict,f)
                print "邮箱修改成功!"

        elif option == '4':
            print "\033[32;1m you have choice is [%s]\033[0m"%Operation_type['4']
            print "友情提示：请输入用户名"
            user_name = raw_input("用户名:")
            new_del_dict = del_user_info(user_name)
            with open('member_info.json','w') as f:
                json.dump(new_del_dict,f)
            print "用户[%s]删除成功!!"%(user_name)
        elif option == '5':
            print "\033[32;1m you have choice is [%s]\033[0m"%Operation_type['5']
            sys.exit(0)
        else:
            print "\033[31;1m The choice of the invalid \033[0m"

    #print check_mobileORmail_member('123456')
    #print check_login('testwqk','testwqk')
    #print handel_query_user(18600404875)