#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Author: zhw
#读取文件中的内容
def open_file(filename ,file_type , *args):
    #读取文件中的数据
    if file_type == 'r':
        flie_list = []
        with open(filename,file_type) as sub_all:
            for file_line in sub_all:
                flie_list.append(file_line.strip())
            #print(flie_list)
            return flie_list
    #读取文件中的用户号
    elif file_type == 'ru':
        flie_list = []
        with open(filename,'r') as sub_all:
            for file_line in sub_all:
                file_l = file_line.split('|')
                user_no = file_l[0]
                flie_list.append(user_no.strip())
            #print(flie_list)
            return flie_list
    #写入文件
    elif file_type == 'a+':
        data = args[0]
        with open(filename, file_type) as sub_all:
            sub_all.write(data +"\n")
    else:
        print('您输入的格式没有，请检查')

class User(object):
    def __init__(self, user_no, user_pw):
        self.user_no = user_no
        self.user_pw = user_pw
        self.file = 'login.log'
    @property
    #检查账号是否已经存在
    def check_user(self):
        flie_list = open_file(self.file, 'ru')
        if self.user_no in flie_list:
            return '1'
        else:
            return '0'
    #注册账号
    def regist(self):
        user_info = '%s|%s' % (self.user_no, self.user_pw)
        open_file(self.file, 'a+',user_info)
        #return'注册成功'
    #账号登录
    def login(self):
        flie_list = open_file(self.file, 'r')
        user_info = '%s|%s' % (self.user_no, self.user_pw)
        if user_info in flie_list:
            return '1'
        else:
            return '0'


def return_more(args):
    pass

#循环验证登录，循环注册账号
while True:
    print('请选择\n  '
          '1. 注册\n'
          '2. 登录\n'
          '3. 退出')
    choose = input('请选择').strip()
    if choose == '1':

        for i in range(3):
            user_no = input('请输入注册账号：').strip()
            user_pw = ''
            user = User(user_no,user_pw)
            #检查用户号是否存在
            return_more = user.check_user
            if return_more == '0':
                user_pw = input('请输入注册密码：').strip()
                user = User(user_no, user_pw)
                return_more = user.regist()
                print(return_more)
                break
            else:
                print('您输入的账号已经存在，请选择其他账号')
        #print('您输入的错误次数太多，已退出！')
    elif choose == '2':
         for i in range(3):
             user_no = input('请输入登录账号：').strip()
             user_pw = input('请输入登录密码').strip()
             user = User(user_no, user_pw)
             #用户登录
             return_mk = user.login()
             if return_mk =='0':
                 print("登录失败，请重新输入")
             else:
                 print('登录成功')
                 break
    elif choose == '3':
        break
    else:
        print('输入序号有无，请重新输入')