refactor api
This commit is contained in:
parent
8583540619
commit
5803ee46d8
2
.gitignore
vendored
2
.gitignore
vendored
@ -1,2 +1,4 @@
|
||||
index.html
|
||||
vvvvidb.sqlite3
|
||||
|
||||
__pycache__/
|
||||
113
api.py
Normal file
113
api.py
Normal file
@ -0,0 +1,113 @@
|
||||
import requests
|
||||
|
||||
class Api():
|
||||
ua = "Mozilla/5.0 (Windows; U; Win98; en-US; rv:0.9.4.2) Gecko/20020502 CS 2000 7.0/7.0"
|
||||
|
||||
def __init__(self):
|
||||
print("Staring VVVVID Api")
|
||||
|
||||
def ds(self, h):
|
||||
g = "MNOPIJKL89+/4567UVWXQRSTEFGHABCDcdefYZabstuvopqr0123wxyzklmnghij"
|
||||
|
||||
def f(m):
|
||||
l = []
|
||||
o = 0
|
||||
b = False
|
||||
m_len = len(m)
|
||||
while ((not b) and o < m_len):
|
||||
n = m[o] << 2
|
||||
o += 1
|
||||
k = -1
|
||||
j = -1
|
||||
if o < m_len:
|
||||
n += m[o] >> 4
|
||||
o += 1
|
||||
if o < m_len:
|
||||
k = (m[o - 1] << 4) & 255
|
||||
k += m[o] >> 2
|
||||
o += 1
|
||||
if o < m_len:
|
||||
j = (m[o - 1] << 6) & 255
|
||||
j += m[o]
|
||||
o += 1
|
||||
else:
|
||||
b = True
|
||||
else:
|
||||
b = True
|
||||
else:
|
||||
b = True
|
||||
l.append(n)
|
||||
if k != -1:
|
||||
l.append(k)
|
||||
if j != -1:
|
||||
l.append(j)
|
||||
return l
|
||||
|
||||
c = []
|
||||
for e in h:
|
||||
c.append(g.index(e))
|
||||
|
||||
c_len = len(c)
|
||||
for e in range(c_len * 2 - 1, -1, -1):
|
||||
a = c[e % c_len] ^ c[(e + 1) % c_len]
|
||||
c[e % c_len] = a
|
||||
|
||||
c = f(c)
|
||||
d = ''
|
||||
for e in c:
|
||||
d += chr(e)
|
||||
|
||||
return d
|
||||
|
||||
def get_settings(self):
|
||||
settings = requests.get('https://www.vvvvid.it/vvvvid/settings', headers={'User-Agent': Api.ua})
|
||||
if settings.status_code != 200:
|
||||
return None
|
||||
self.stream_url = settings.json()['data']['defaultStreamingServer']
|
||||
return self.stream_url
|
||||
|
||||
def login(self):
|
||||
login = requests.get('https://www.vvvvid.it/user/login', headers={'User-Agent': Api.ua})
|
||||
self.conn_id = login.json()['data']['conn_id']
|
||||
|
||||
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.encoding = 'utf-8'
|
||||
if info.json()['result'] == 'ok':
|
||||
return info.json()['data']
|
||||
else:
|
||||
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})
|
||||
if seasons.json()['result'] == 'ok' and seasons.json()['data'] and seasons.json()['data'][0]['episodes']:
|
||||
return seasons.json()['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()
|
||||
if episodes['result'] == 'ok' and episodes['data'] and episodes['data'][0]['embed_info']:
|
||||
return episodes['data']
|
||||
else:
|
||||
return []
|
||||
|
||||
def format_episodes(self, season_id, show_id, episodes):
|
||||
eps = []
|
||||
count = 0
|
||||
for k in episodes:
|
||||
count += 1
|
||||
if k['embed_info']:
|
||||
if k['video_type'] == 'video/rcs':
|
||||
embed_info = self.ds(k['embed_info'])
|
||||
embed_info = 'https' + embed_info[4:30] + 'i' + embed_info[31:-12] + 'master.m3u8'
|
||||
elif k['video_type'] == 'video/vvvvid':
|
||||
embed_info = 'https' + self.stream_url[4:] + self.ds(k['embed_info']) + '/playlist.m3u8'
|
||||
elif k['video_type'] == 'video/youtube':
|
||||
embed_info = self.ds(k['embed_info'])
|
||||
elif k['video_type'] == 'video/kenc':
|
||||
embed_info = self.ds(k['embed_info'])
|
||||
else:
|
||||
embed_info = self.ds(k['embed_info'])
|
||||
eps.append((count, show_id, season_id, k['video_type'], embed_info))
|
||||
return eps
|
||||
83
db.py
Normal file
83
db.py
Normal file
@ -0,0 +1,83 @@
|
||||
import os
|
||||
import sqlite3
|
||||
|
||||
class VVVVIDatabase():
|
||||
|
||||
def __init__(self, path):
|
||||
self.path = path
|
||||
self.is_valid = os.path.isfile(self.path)
|
||||
|
||||
def create(self):
|
||||
con = sqlite3.connect(self.path)
|
||||
cur = con.cursor()
|
||||
cur.execute("CREATE TABLE series (id INTEGER, name TEXT NOT NULL, season_id INTEGER, type TEXT, PRIMARY KEY (id, season_id));")
|
||||
cur.execute("CREATE TABLE episodes (id INTEGER, serie_id INTEGER, season_id INTEGER, cdn_url TEXT NOT NULL, type TEXT NOT NULL);")
|
||||
con.commit()
|
||||
con.close()
|
||||
|
||||
def last_serie(self):
|
||||
con = sqlite3.connect(self.path)
|
||||
cur = con.cursor()
|
||||
cur.execute("SELECT id FROM series ORDER BY id DESC LIMIT 1;")
|
||||
rows = cur.fetchall()
|
||||
last = (rows[0][0] + 1) if len(rows) > 0 else 0
|
||||
con.commit()
|
||||
con.close()
|
||||
return last
|
||||
|
||||
def series_id(self):
|
||||
con = sqlite3.connect(self.path)
|
||||
cur = con.cursor()
|
||||
cur.execute("SELECT id, season_id FROM series ORDER BY id, season_id DESC;")
|
||||
rows = cur.fetchall()
|
||||
con.commit()
|
||||
con.close()
|
||||
# getting series in a useful format for later
|
||||
series = {}
|
||||
for i,s in rows:
|
||||
if series.get(i) is None:
|
||||
series[i] = []
|
||||
series[i].append(s)
|
||||
return series
|
||||
|
||||
def series_episodes(self, serie_id):
|
||||
con = sqlite3.connect(self.path)
|
||||
cur = con.cursor()
|
||||
cur.execute("SELECT id, serie_id, season_id FROM episodes ORDER BY id, serie_id, season_id DESC;")
|
||||
rows = cur.fetchall()
|
||||
con.commit()
|
||||
con.close()
|
||||
# getting series in a useful format for later
|
||||
series = {}
|
||||
for i,s in rows:
|
||||
if series.get(i) is None:
|
||||
series[i] = []
|
||||
series[i].append(s)
|
||||
return series
|
||||
|
||||
|
||||
def insert_serie(self, serie):
|
||||
con = sqlite3.connect(self.path)
|
||||
cur = con.cursor()
|
||||
try:
|
||||
cur.execute("INSERT INTO series (id, name, season_id, type) VALUES (?, ?, ?, ?);", serie)
|
||||
con.commit()
|
||||
con.close()
|
||||
return True
|
||||
except sqlite3.IntegrityError:
|
||||
# serie gia' presente
|
||||
pass
|
||||
return False
|
||||
|
||||
def insert_episodes(self, eps):
|
||||
con = sqlite3.connect(self.path)
|
||||
cur = con.cursor()
|
||||
try:
|
||||
cur.executemany("INSERT INTO episodes (id, serie_id, season_id, type, cdn_url) VALUES (?, ?, ?, ?, ?);", eps)
|
||||
con.commit()
|
||||
con.close()
|
||||
return True
|
||||
except sqlite3.IntegrityError:
|
||||
# episodi gia' presenti
|
||||
pass
|
||||
return False
|
||||
184
vvvvget.py
184
vvvvget.py
@ -2,168 +2,44 @@ import os
|
||||
import json
|
||||
import requests
|
||||
import sqlite3
|
||||
from db import VVVVIDatabase
|
||||
from api import Api
|
||||
|
||||
ua = "Mozilla/5.0 (Windows; U; Win98; en-US; rv:0.9.4.2) Gecko/20020502 CS 2000 7.0/7.0"
|
||||
|
||||
def ds(h):
|
||||
g = "MNOPIJKL89+/4567UVWXQRSTEFGHABCDcdefYZabstuvopqr0123wxyzklmnghij"
|
||||
|
||||
def f(m):
|
||||
l = []
|
||||
o = 0
|
||||
b = False
|
||||
m_len = len(m)
|
||||
while ((not b) and o < m_len):
|
||||
n = m[o] << 2
|
||||
o += 1
|
||||
k = -1
|
||||
j = -1
|
||||
if o < m_len:
|
||||
n += m[o] >> 4
|
||||
o += 1
|
||||
if o < m_len:
|
||||
k = (m[o - 1] << 4) & 255
|
||||
k += m[o] >> 2
|
||||
o += 1
|
||||
if o < m_len:
|
||||
j = (m[o - 1] << 6) & 255
|
||||
j += m[o]
|
||||
o += 1
|
||||
else:
|
||||
b = True
|
||||
else:
|
||||
b = True
|
||||
else:
|
||||
b = True
|
||||
l.append(n)
|
||||
if k != -1:
|
||||
l.append(k)
|
||||
if j != -1:
|
||||
l.append(j)
|
||||
return l
|
||||
|
||||
c = []
|
||||
for e in h:
|
||||
c.append(g.index(e))
|
||||
|
||||
c_len = len(c)
|
||||
for e in range(c_len * 2 - 1, -1, -1):
|
||||
a = c[e % c_len] ^ c[(e + 1) % c_len]
|
||||
c[e % c_len] = a
|
||||
|
||||
c = f(c)
|
||||
d = ''
|
||||
for e in c:
|
||||
d += chr(e)
|
||||
|
||||
return d
|
||||
|
||||
def get_settings():
|
||||
settings = requests.get('https://www.vvvvid.it/vvvvid/settings', headers={'User-Agent': ua})
|
||||
if settings.status_code != 200:
|
||||
return None
|
||||
return settings.json()['data']['defaultStreamingServer']
|
||||
|
||||
def login():
|
||||
login = requests.get('https://www.vvvvid.it/user/login', headers={'User-Agent': ua})
|
||||
return login.json()['data']['conn_id']
|
||||
|
||||
def get_info(show_id, conn_id):
|
||||
info = requests.get('https://www.vvvvid.it/vvvvid/ondemand/' + str(show_id) + '/info/?conn_id=' + conn_id, headers={'User-Agent': ua})
|
||||
info.encoding = 'utf-8'
|
||||
if info.json()['result'] == 'ok':
|
||||
return info.json()['data']
|
||||
else:
|
||||
return False
|
||||
|
||||
def get_seasons(show_id, conn_id):
|
||||
seasons = requests.get('https://www.vvvvid.it/vvvvid/ondemand/' + str(show_id) + '/seasons/?conn_id=' + conn_id, headers={'User-Agent': ua})
|
||||
if seasons.json()['result'] == 'ok' and seasons.json()['data'] and seasons.json()['data'][0]['episodes']:
|
||||
return seasons.json()['data']
|
||||
else:
|
||||
return False
|
||||
|
||||
def get_episodes(season_id, show_id, conn_id):
|
||||
episodes = requests.get('https://www.vvvvid.it/vvvvid/ondemand/' + str(show_id) + '/season/' +str(season_id) + '?conn_id=' + conn_id, headers={'User-Agent': ua})
|
||||
if episodes.json()['result'] == 'ok' and episodes.json()['data'] and episodes.json()['data'][0]['embed_info']:
|
||||
return episodes.json()['data']
|
||||
else:
|
||||
return False
|
||||
|
||||
vvvvid_stream_url = get_settings()
|
||||
if vvvvid_stream_url is None:
|
||||
api = Api()
|
||||
stream_url = api.get_settings()
|
||||
if stream_url is None:
|
||||
print("VVVVID is not available at the moment")
|
||||
exit(1)
|
||||
|
||||
vvvvidb = "vvvvidb.sqlite3"
|
||||
last = 0
|
||||
vvvvidb = VVVVIDatabase("vvvvidb.sqlite3")
|
||||
api.login()
|
||||
|
||||
if not os.path.isfile(vvvvidb):
|
||||
con = sqlite3.connect(vvvvidb)
|
||||
cur = con.cursor()
|
||||
cur.execute("CREATE TABLE series (id INTEGER, name TEXT NOT NULL, season_id INTEGER, type TEXT, PRIMARY KEY (id, season_id));")
|
||||
cur.execute("CREATE TABLE episodes (serie_id INTEGER, season_id INTEGER, cdn_url TEXT NOT NULL, type TEXT NOT NULL);")
|
||||
con.commit()
|
||||
con.close()
|
||||
if not vvvvidb.is_valid:
|
||||
# Database file not present
|
||||
# Create a new database
|
||||
vvvvidb.create()
|
||||
else:
|
||||
con = sqlite3.connect(vvvvidb)
|
||||
cur = con.cursor()
|
||||
cur.execute("SELECT id FROM series ORDER BY id DESC LIMIT 1;")
|
||||
rows = cur.fetchall()
|
||||
if len(rows) > 0:
|
||||
last = rows[0][0] + 1
|
||||
con.commit()
|
||||
con.close()
|
||||
|
||||
print("Resuming from...{}".format(last))
|
||||
|
||||
con = sqlite3.connect(vvvvidb)
|
||||
cur = con.cursor()
|
||||
|
||||
stream_url = get_settings()
|
||||
conn_id = login()
|
||||
# Database file is already present
|
||||
# Since we have no information about series lenght prior to their pubblication
|
||||
# We scan all the older id to see if there are new episodes, then we scan every id greater then the last one
|
||||
ids = vvvvidb.series_id()
|
||||
|
||||
last = 0
|
||||
# Scan all the episodes
|
||||
for i in range(last, min(last + 500, 1000)):
|
||||
print("Fetching...{}".format(i))
|
||||
info = get_info(i, conn_id)
|
||||
if info:
|
||||
seasons = get_seasons(i, conn_id)
|
||||
if seasons:
|
||||
for j in seasons:
|
||||
serie = (info['show_id'], info['title'], j['season_id'], j['name'])
|
||||
print("Found: {}".format(info['title']))
|
||||
info = api.get_info(i)
|
||||
if not info:
|
||||
continue
|
||||
seasons = api.get_seasons(i)
|
||||
for j in seasons:
|
||||
print("Found: {}".format(info['title']))
|
||||
|
||||
try:
|
||||
cur.execute("INSERT INTO series (id, name, season_id, type) VALUES (?, ?, ?, ?);", serie)
|
||||
con.commit()
|
||||
except sqlite3.IntegrityError:
|
||||
# serie/stagione gia' presente, salta il fetch degli episodi
|
||||
continue
|
||||
if not vvvvidb.insert_serie((info['show_id'], info['title'], j['season_id'], j['name'])):
|
||||
print("Serie already present")
|
||||
continue
|
||||
|
||||
eps = []
|
||||
episodes = get_episodes(j['season_id'], i, conn_id)
|
||||
if episodes:
|
||||
for k in episodes:
|
||||
if k['embed_info']:
|
||||
if k['video_type'] == 'video/rcs':
|
||||
embed_info = ds(k['embed_info'])
|
||||
embed_info = 'https' + embed_info[4:30] + 'i' + embed_info[31:-12] + 'master.m3u8'
|
||||
elif k['video_type'] == 'video/vvvvid':
|
||||
embed_info = 'https' + vvvvid_stream_url[4:] + ds(k['embed_info']) + '/playlist.m3u8'
|
||||
elif k['video_type'] == 'video/youtube':
|
||||
embed_info = ds(k['embed_info'])
|
||||
elif k['video_type'] == 'video/kenc':
|
||||
embed_info = ds(k['embed_info'])
|
||||
else:
|
||||
embed_info = ds(k['embed_info'])
|
||||
eps.append((info['show_id'], j['season_id'], k['video_type'], embed_info))
|
||||
print("Found {} episodes".format(len(eps)))
|
||||
|
||||
try:
|
||||
cur.executemany("INSERT INTO episodes (serie_id, season_id, type, cdn_url) VALUES (?, ?, ?, ?);", eps)
|
||||
con.commit()
|
||||
except sqlite3.IntegrityError:
|
||||
# episodi gia' presenti
|
||||
pass
|
||||
|
||||
con.close()
|
||||
episodes = api.get_episodes(j['season_id'], i)
|
||||
eps = api.format_episodes(j['season_id'], info['show_id'], episodes)
|
||||
print("Found {} episodes".format(len(eps)))
|
||||
vvvvidb.insert_episodes(eps)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user