Added validation for V.I.P. account status in titulky providerv1.0.3-beta.2

1 files changed, 11 insertions, 3 deletions

diff --git a/libs/subliminal_patch/providers/titulky.py b/libs/subliminal_patch/providers/titulky.py
index 7e7b63d09..7b33acd04 100644
--- a/libs/subliminal_patch/providers/titulky.py
+++ b/libs/subliminal_patch/providers/titulky.py
@@ -5,6 +5,7 @@ import math
 import re
 import zipfile
 from random import randint
+from urllib.parse import urlparse, parse_qs
 from threading import Thread
 
 import rarfile
@@ -266,9 +267,14 @@ class TitulkyProvider(Provider, ProviderSubtitleArchiveMixin):
                                 allow_redirects=False,
                                 timeout=self.timeout)
 
+        location_qs = parse_qs(urlparse(res.headers['Location']).query)
+
         # If the response is a redirect and doesnt point to an error message page, then we are logged in
-        if res.status_code == 302 and 'msg_type=i' in res.headers['Location']:
-            return True
+        if res.status_code == 302 and location_qs['msg_type'][0] == 'i':
+            if 'omezené' in location_qs['msg'][0]:
+                raise AuthenticationError("V.I.P. account is required for this provider to work!")
+            else:
+                return True
         else:
             raise AuthenticationError("Login failed")
 
@@ -279,8 +285,10 @@ class TitulkyProvider(Provider, ProviderSubtitleArchiveMixin):
                                allow_redirects=False,
                                timeout=self.timeout)
 
+        location_qs = parse_qs(urlparse(res.headers['Location']).query)
+
         # If the response is a redirect and doesnt point to an error message page, then we are logged out
-        if res.status_code == 302 and 'msg_type=i' in res.headers['Location']:
+        if res.status_code == 302 and location_qs['msg_type'][0] == 'i':
             return True
         else:
             raise AuthenticationError("Logout failed.")