def submit_advanced(cls, args, kwargs, delay=0, user_id=None):
task = AsyncTask(user_id=user_id)
task.set_status(*cls.get_initial_status())
task.type = cls.get_type()
Session.add(task)
Session.commit()
queue_id = job.submit_advanced(cls, (task.id, args, kwargs), {}, delay=delay)
task.queue_id = queue_id
Session.commit()
return task
def run_for_user_now(cls, user_id):
last = AsyncTask.get_for_user(user_id=user_id,
type=cls.get_type(),
limit=1).first()
if not last or last.is_completed:
cls.submit_advanced((user_id,), {}, user_id=user_id)
else:
last.run_now()
def status(self):
last = AsyncTask.get_for_user(type=tasks.FullSync.get_type(),
limit=1).first()
data = None
if last:
if last.time_left:
data = None
else:
data = dict((attr, getattr(last, attr))
for attr in
('total_units','completed_units','status_data'))
return json.dumps(data)
def submit_advanced(cls, args, kwargs, delay=0, user_id=None):
task = AsyncTask(user_id=user_id)
task.set_status(*cls.get_initial_status())
task.type = cls.get_type()
Session.add(task)
Session.commit()
queue_id = job.submit_advanced(cls, (task.id, args, kwargs), {}, delay=delay)
task.queue_id = queue_id
Session.commit()
return task
def index(self):
c.user = self.logged_in_user
c.fb_connect_url = fb.get_authorization_url()
c.flickr_connect_url = flickr.get_authorization_url()
if self.logged_in_user:
if self.logged_in_user.fb_access_token:
c.fb_user = fb.GraphUser(
id=self.logged_in_user.fb_uid,
access_token=self.logged_in_user.fb_access_token)
if self.logged_in_user.flickr_token:
c.flickr_user = flickr.FlickrUser(
token=self.logged_in_user.flickr_token)
if self.logged_in_user.flickr_token and self.logged_in_user.fb_access_token:
c.tasks = AsyncTask.get_for_user(
limit=2,
type=FullSync.get_type()).all()
c.sync_url = '/sync/full_sync'
for task in c.tasks:
if task.is_completed:
c.last_task = task
elif task.time_left:
c.next_task = task
else:
c.current_task = task
bytes_in, bytes_out = SyncRecord.get_bytes_transfered_by_user(self.logged_in_user.id)
c.bytes_transferred = bytes_in/1024.**2
# cost is $0.10 per GB transferred in and
# $0.15 per GB transferred out
c.cost = bytes_in/1024.**3 * 0.1 + bytes_out/1024.**3 * 0.15
if c.tasks:
return render('/homepage.mako')
return render('/frontpage.mako')
if session.get('user_id'):
return self.home()
else:
c.fb_connect_url = fb.get_authorization_url([
'user_photos',
'publish_stream',
'offline_access',
])
# Return a rendered template
return render('/frontpage.mako')
def command(self):
"""Main command that starts the worker."""
super(KickCommand, self).command()
users = Session.query(User)\
.filter(User.fb_uid != None)\
.filter(User.flickr_nsid != None)
if self.options.simulate:
log.info("--SIMULATING--")
for user in users:
tasks = AsyncTask.get_for_user(user.id)
for task in tasks:
if not (task.is_completed or task.time_left) and task.is_buried:
log.info("Running task %r", task)
if not self.options.simulate:
task.run_now()
def home(self):
c.picasa_connect_url = picasa.get_authorization_url()
c.flickr_connect_url = flickr.get_authorization_url()
c.fb_connect_url = fb.get_authorization_url()
c.fbuser = fb.GraphUser()
c.picasa_user = None
if session.get('picasa_token'):
c.picasa_user = gdata.photos.service.PhotosService()
c.flickr_user = None
if session.get('flickr_token'):
c.flickr_user = flickr.FlickrUser()
tasks = AsyncTask.get_for_user(limit=2, type=FullSync.get_type()).all()
c.tasks = tasks
for task in reversed(tasks):
if task.is_completed:
c.last_task = task
elif task.time_left:
c.next_task = task
else:
c.current_task = task
return render('/homepage.mako')
def run_for_user_now(cls, user_id):
last = AsyncTask.get_for_user(user_id=user_id, type=cls.get_type(), limit=1).first()
if not last or last.is_completed:
cls.submit_advanced((user_id,), {}, user_id=user_id)
else:
last.run_now()
