def index(self, *args):
from mako.template import Template
tmpl = Template(filename='template.mako', output_encoding='utf-8')
data = {}
cd = collections.defaultdict
data['status'] = cd(lambda: cd(lambda: cd(lambda: cd(int))))
db = DB()
rows = db.get_status()
db.close()
for year, clip, bw, state, count in rows:
print year, clip, bw, state, count
state = {
'no': u'未下載',
'404': u'404 not found (to try)',
'404skip': u'404 not found',
'downloading': u'下載中',
'downloaded': u'已下載(a)',
'stored': u'已下載(b)',
}.get(state, u'其他')
data['status'][year][clip][bw][state] = count
print data
result = tmpl.render(**data)
return result
def change(self, *args, **argd):
key = argd.get('key')
date, url = json.loads(key)
state = argd.get('state')
info = argd.get('info')
name = argd.get('name')
token = argd.get('token')
db = DB()
if db.get_user_token(name) != token:
return 'invalid token'
if state == 'downloaded' and not info:
return 'invalid'
assert state in ('404', 'downloaded', 'failed')
with db.conn:
db.change_job_state(url, name, state)
if info:
db.add_video_info(url, name, info)
if state == '404':
result = db.query('''
SELECT count(*)
FROM download_state_history
WHERE key = %s AND state = '404'
''', url)
if int(result[0][0]) >= 10:
db.change_job_state(url, name, '404skip')
db.close()
return 'ok'
def register(self, *args, **argd):
name = argd.get('name')
contact = argd.get('contact')
if not name or not contact:
return json.dumps(dict(result='error', msg='empty name or contact'))
pattern = r'^[[email protected]]+$'
if not re.match(pattern, name) or not re.match(pattern, contact):
return json.dumps(dict(result='error', msg='valid pattern: %s' % pattern))
db = DB()
if db.get_user_token(name):
return json.dumps(dict(result='error', msg='user exists'))
assert name and contact
token = str(random.getrandbits(32))
db.add_user_token(name, contact, token)
db.close()
return json.dumps(dict(result='ok', token=token))
def next(self, *args, **argd):
bw = argd.get('bw')
name = argd.get('name')
token = argd.get('token')
if not name or not token:
return 'Empty name or token'
db = DB()
if db.get_user_token(name) != token:
return json.dumps('invalid token')
now = datetime.datetime.now()
if 0 <= now.weekday() <= 4 and 6 <= now.hour <= 19:
result = db.query('''
SELECT count(*)
FROM download_state
WHERE state = 'downloading' AND last_modified > now() - interval '24 hours'
''')
if result and int(result[0][0]) > 10:
return 'wait'
# bw, clip
# 0, 0 200KB/s
# 0, 1 500KB/s
# 1, 0 70KB/s
# 1, 1 70KB/s
if bw == 'high':
bw_cond = 'true'
elif bw == 'mid':
# bw, clip != 0, 1
bw_cond = '(bw = 1 OR clip = 0)'
elif bw == 'low':
# bw = 1
bw_cond = '(bw = 1)'
else:
bw_cond = 'true'
while True:
t0 = time.time()
result = db.query('''
SELECT key, videodate, last_modified, state
FROM download_state
WHERE (
(state = 'no')
OR (state = 'downloading' AND clip = 1 AND last_modified < now() - interval '12 hours')
OR (state = 'downloading' AND clip = 0 AND last_modified < now() - interval '24 hours')
OR (state = 'failed' AND last_modified < now() - interval '10 minutes')
OR (state = '404' AND last_modified < now() - interval '10 minutes')
) AND state != '404skip' AND %s
ORDER BY last_modified
LIMIT 1
FOR UPDATE
''' % bw_cond,
)
print 't', time.time() - t0
if not result or not result[0][0]:
return 'done'
row = result[0]
key, videodate, last_modified, state = row
if state == '404':
result = db.query('''
SELECT count(*)
FROM download_state_history
WHERE key = %s AND state = '404'
''', key)
if int(result[0][0]) >= 10:
with db.conn:
db.change_job_state(key, name, '404skip')
continue
with db.conn:
db.change_job_state(key, name, 'downloading')
db.close()
return json.dumps((str(videodate).replace('-','/'), key))
def metadata(self, vid):
db = DB()
cherrypy.response.headers['Content-Type'] = 'application/json; charset=utf-8'
data = db.get_metadata(vid)
db.close()
return data
def status(self):
db = DB()
rows = db.get_status()
db.close()
return json.dumps(rows)