예제 #1
0
    def login_and_signup(self):
        msg = self.connect1.recv(1024)
        msg = msg.split()
        if msg[0] == 'exit':
            return 'EXIT'
        username = msg[1]
        password = msg[2]
        myreg = Register()
        if msg[0] == 'log':
            if myreg.login(username,password) == True:
                self.username = username
                self.au = Authority(self.username)
                self.connect1.send('success')
                return 'OK'
            else:
                self.connect1.send('fail')
                return 'FAIL'

        elif msg[0] == 'sign':
            if myreg.sign_up(username,password) == True:
                self.username = username
                #use username to create a root directory
                self.au = Authority(self.username)
                self.create_dir('.')
                self.connect1.send('success')
                return 'OK'
            else:
                self.connect1.send('fail')
                return 'FAIL'