55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
from __future__ import division
|
|
|
|
from threading import Thread
|
|
import time
|
|
import logging
|
|
|
|
import db
|
|
import state
|
|
import settings
|
|
import update
|
|
|
|
logger = logging.getLogger('oml.downloads')
|
|
|
|
class Downloads(Thread):
|
|
|
|
def __init__(self):
|
|
self._running = True
|
|
Thread.__init__(self)
|
|
self.daemon = True
|
|
self.start()
|
|
|
|
def download_updates(self):
|
|
now = int(time.mktime(time.gmtime()))
|
|
if now > settings.server.get('last_update_check', 0) + 24*60*60:
|
|
settings.server['last_update_check'] = now
|
|
update.download()
|
|
|
|
def download_next(self):
|
|
import item.models
|
|
self.download_updates()
|
|
for t in item.models.Transfer.query.filter(
|
|
item.models.Transfer.added!=None,
|
|
item.models.Transfer.progress<1).order_by(item.models.Transfer.added):
|
|
for u in t.item.users:
|
|
if state.nodes.is_online(u.id):
|
|
logger.debug('DOWNLOAD %s %s', t.item, u)
|
|
r = state.nodes.download(u.id, t.item)
|
|
logger.debug('download ok? %s', r)
|
|
return True
|
|
return False
|
|
|
|
def run(self):
|
|
time.sleep(2)
|
|
with db.session():
|
|
while self._running:
|
|
self.download_next()
|
|
time.sleep(0.5)
|
|
|
|
def join(self):
|
|
self._running = False
|
|
self._q.put(None)
|
|
return Thread.join(self)
|
|
|