From 5b7459cb7109a0634fe8db74e45fd63c460ad836 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Wed, 24 Jan 2018 18:38:18 +0100
Subject: [PATCH] Fix infinite loop in case a flood wait is triggered

---
 pyrogram/client/client.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py
index bf8b1e51..9c282a82 100644
--- a/pyrogram/client/client.py
+++ b/pyrogram/client/client.py
@@ -299,7 +299,9 @@ class Client:
                             print("Hint: {}".format(r.hint))
                             self.password = input("Enter password: ")  # TODO: Use getpass
 
-                        self.password = r.current_salt + self.password.encode() + r.current_salt
+                        if type(self.password) is str:
+                            self.password = r.current_salt + self.password.encode() + r.current_salt
+
                         password_hash = sha256(self.password).digest()
 
                         r = self.send(functions.auth.CheckPassword(password_hash))