managers.py

from django.db import models


class JSDependencyManager(models.Manager):
	def get_active(self, **kwargs):
		return self.get_query_set().filter(active=True,**kwargs)

class JSLibraryManager(models.Manager):
	def get_active(self, **kwargs):
		return self.get_query_set().filter(active=True,**kwargs)

class PastieManager(models.Manager):
	def all_examples(self):
		return self.get_query_set().filter(example=True)

	def all_examples_by_groups(self):
		examples = self.all_examples()
		libs = {}
		for ex in examples:
			group_name = ex.favourite.js_lib.library_group.name
			if not libs.has_key(group_name):
				libs[group_name] = []
			libs[group_name].append(ex)
		return libs

class ShellManager(models.Manager):
	def all(self):
		public = self.get_query_set().filter(private=False)
		return public

	def all_available(self, user=None):
		public = self.get_query_set().filter(private=False)
		if user and user.is_authenticated():
			owned = self.get_query_set().filter(private=True, author__id=user.id)
			return public | owned
		return public

	def all_owned(self, user=None):
		return self.get_query_set().filter(private=True, author__id=user.id)

	def all_with_private(self):
		return super(ShellManager, self).all()
 
	def get_public_or_owned(self, **kwargs):
		try:
			return self.get_public(**kwargs)
		except:
			return self.get_owned(user, **kwargs)

	def get_public(self, **kwargs):
		return self.get_query_set().get(private=False, **kwargs)

	def get_owned(self, user, **kwargs):
		return self.get_query_set().get(private=True, author__id=user.id, **kwargs)