Probably fixed "Cannot find Follow button" #128

alexal1 · Oct 8, 2020 · c2fdfb9 · c2fdfb9
1 parent 553342b
commit c2fdfb9
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/action_handle_blogger.py b/src/action_handle_blogger.py
@@ -9,6 +9,7 @@
 
 FOLLOWERS_BUTTON_ID_REGEX = 'com.instagram.android:id/row_profile_header_followers_container' \
                             '|com.instagram.android:id/row_profile_header_container_followers'
+TEXTVIEW_OR_BUTTON_REGEX = 'android.widget.TextView|android.widget.Button'
 
 
 def handle_blogger(device,
@@ -309,15 +310,15 @@ def _follow(device, username, follow_percentage):
 
     random_sleep()
 
-    follow_button = device.find(className='android.widget.Button',
+    follow_button = device.find(classNameMatches=TEXTVIEW_OR_BUTTON_REGEX,
                                 clickable=True,
                                 text='Follow')
     if not follow_button.exists():
-        follow_button = device.find(className='android.widget.Button',
+        follow_button = device.find(classNameMatches=TEXTVIEW_OR_BUTTON_REGEX,
                                     clickable=True,
                                     text='Follow Back')
     if not follow_button.exists():
-        unfollow_button = device.find(className='android.widget.Button',
+        unfollow_button = device.find(classNameMatches=TEXTVIEW_OR_BUTTON_REGEX,
                                       clickable=True,
                                       text='Following')
         if unfollow_button.exists():