diff --git a/app/utils/site.py b/app/utils/site.py index 48666469..1fd41cf2 100644 --- a/app/utils/site.py +++ b/app/utils/site.py @@ -22,14 +22,12 @@ class SiteUtils: ' or contains(@href, "mybonus") ' ' or contains(@onclick, "logout")' ' or contains(@href, "usercp")]', - '//form[contains(@action, "logout")]'] + '//form[contains(@action, "logout")]', + '//div[@class="user-info-side"]', + '//a[contains(@href, "/profile/detail/")]'] for xpath in xpaths: if html.xpath(xpath): return True - user_info_div = html.xpath('//div[@class="user-info-side"]') - if user_info_div: - return True - return False @classmethod