def register(self, user, password, email): self._print('register starts') self._lock.acquire() try: if SHOW_TIME: start_time = datetime.utcnow() pwd = get_md5(password) addr = self._get_user_backend(user) rpcclient = RPCClient(addr, BACKEND_PORT) res = rpcclient.request('register', user=user, pwd=pwd, email=email) if SHOW_TIME: self._print('register , time=%d sec' % (datetime.utcnow() - start_time).seconds) if res: if DEBUG: self._register_cnt += 1 self._print('register, count=%d' % self._register_cnt) return True else: show_error(self, 'failed to register %s' % str(user)) return False finally: self._lock.release()
def login(self, user, password): self._print('login starts') try: if SHOW_TIME: start_time = datetime.utcnow() pwd = get_md5(password) addr = self._get_user_backend(user) rpcclient = RPCClient(addr, BACKEND_PORT) uid, key = rpcclient.request('login', user=user, pwd=pwd) if SHOW_TIME: self._print('login , time=%d sec' % (datetime.utcnow() - start_time).seconds) if uid and key: if DEBUG: self._login_cnt += 1 self._print('login, count=%d' % self._login_cnt) return (uid, key) except: show_error(self, 'failed to login')