Browse Source

Fixed session

Giulio 2 years ago
parent
commit
bf38af0485
1 changed files with 7 additions and 5 deletions
  1. 7 5
      api.py

+ 7 - 5
api.py

@@ -67,11 +67,13 @@ class Api():
 		return self.stream_url
 
 	def login(self):
-		login = requests.get('https://www.vvvvid.it/user/login', headers={'User-Agent': Api.ua}).json()
-		self.conn_id = login['data']['conn_id']
+		s = requests.session()
+		login = s.get('https://www.vvvvid.it/user/login', headers={'User-Agent': Api.ua})
+		self.conn_id = login.json()['data']['conn_id']
+		self.session = s
 
 	def get_info(self, show_id):
-		info = requests.get('https://www.vvvvid.it/vvvvid/ondemand/' + str(show_id) + '/info/?conn_id=' + self.conn_id, headers={'User-Agent': Api.ua})
+		info = self.session.get('https://www.vvvvid.it/vvvvid/ondemand/' + str(show_id) + '/info/?conn_id=' + self.conn_id, headers={'User-Agent': Api.ua})
 		info.encoding = 'utf-8'
 		info = info.json()
 		if info['result'] == 'ok':
@@ -80,14 +82,14 @@ class Api():
 			return False
 
 	def get_seasons(self, show_id):
-		seasons = requests.get('https://www.vvvvid.it/vvvvid/ondemand/' + str(show_id) + '/seasons/?conn_id=' + self.conn_id, headers={'User-Agent': Api.ua}).json()
+		seasons = self.session.get('https://www.vvvvid.it/vvvvid/ondemand/' + str(show_id) + '/seasons/?conn_id=' + self.conn_id, headers={'User-Agent': Api.ua}).json()
 		if seasons['result'] == 'ok' and seasons['data']:  # and seasons['data'][0]['episodes']:
 			return seasons['data']
 		else:
 			return []
 
 	def get_episodes(self, season_id, show_id):
-		episodes = requests.get('https://www.vvvvid.it/vvvvid/ondemand/' + str(show_id) + '/season/' + str(season_id) + '?conn_id=' + self.conn_id, headers={'User-Agent': Api.ua}).json()
+		episodes = self.session.get('https://www.vvvvid.it/vvvvid/ondemand/' + str(show_id) + '/season/' + str(season_id) + '?conn_id=' + self.conn_id, headers={'User-Agent': Api.ua}).json()
 		if episodes['result'] == 'ok' and episodes['data']:  # and episodes['data'][0]['embed_info']:
 			return episodes['data']
 		else: