Fix recommendations

.gitignore

@@ -1,3 +1,4 @@
 **/creds*
 build/*
+src/test*.py
 docs/build/*

src/phub/__init__.py

@@ -1,5 +1,5 @@
 '''
-## PHUB v4.4
+## PHUB v4.5.3
 
 See https://phub.readthedocs.io/en/latest/ for docs.
 See https://github.com/EchterAlsFake/PHUB for code.
@@ -9,7 +9,7 @@
 __author__ = 'Egsagon, EchterAlsFake'
 __copyright__ = 'Copyright 2024, PHUB'
 __license__ = 'GPLv3'
-__version__  = '4.4'
+__version__  = '4.5.3'
 
 __all__ = ['Client', 'Quality', 'core', 'utils',
            'consts', 'errors', 'objects', 'modules']

src/phub/consts.py

@@ -23,6 +23,15 @@
     'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/114.0'
 }
 
+COOKIES = {
+    'accessAgeDisclaimerPH': '1',
+    'accessAgeDisclaimerUK': '1',
+    'accessPH': '1',
+    'age_verified': '1',
+    'cookieBannerState': '1',
+    'platform': 'pc'
+}
+
 LOGIN_PAYLOAD = {
     'from': 'pc_login_modal_:homepage_redesign',
 }

src/phub/core.py

@@ -79,12 +79,9 @@ def reset(self) -> None:
         self.session = requests.Session()
         self._clear_granted_token()
 
-        # Bypass age disclaimer
-        self.session.cookies.set('accessAgeDisclaimerPH', '1')
-        self.session.cookies.set('accessAgeDisclaimerUK', '1')
-        self.session.cookies.set('accessPH', '1')
-        self.session.cookies.set('age_verified', '1')
-
+        # Insert cookies
+        self.session.cookies.update(consts.COOKIES)
+
     def call(self,
              func: str,
              method: str = 'GET',

src/phub/objects/account.py

@@ -104,6 +104,23 @@ def refresh(self, refresh_login: bool = False) -> None:
                 logger.debug('Deleting key %s', key)
                 delattr(self, key)
 
+    def fix_recommendations(self) -> None:
+        '''
+        Allow recommandations cookies.
+        '''
+
+        logger.info('Fixing account recommendations')
+
+        payload = utils.urlify({
+            'token': self.client._granted_token,
+            'cookie_selection': 3,
+            'site_id': 1
+        })
+
+        res = self.client.call('user/log_user_cookie_consent' + payload, silent = True)
+
+        assert res.json().get('success')
+
     @cached_property
     def recommended(self) -> queries.VideoQuery:
         '''
@@ -112,6 +129,9 @@ def recommended(self) -> queries.VideoQuery:
 
         from . import queries
 
+        # Called each account refresh
+        self.fix_recommendations()
+
         return queries.VideoQuery(self.client, 'recommended')
 
     @cached_property

src/phub/objects/query.py

@@ -277,7 +277,7 @@ def _parse_item(self, raw: str) -> Video:
 
             obj.data = {
                 # Property overrides
-                'page@title': data["title"],
+                'page@video_title': data["title"],
                 'data@thumb': data["image"],
                 'page@id': id,