def watch_button(context, package):
user = context['user']
next = context['request'].path
if user.is_authenticated():
if package.watch_set.filter(user=user).exists():
action, text = url('delete_watch', package.name), 'Stop watching this package'
else:
action, text = url('create_watch', package.name), 'Watch this package'
else:
action = url('login') + '?next=' + next
text = 'Log in to watch this package'
return {'action': action, 'text': text, 'user': user}
def get_url(self):
if not self.package:
self.package = PackageFactory.create()
return url('delete_watch', self.package.name)