p15693087
/
django
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

magic-removal: Fixed bugs in custom_methods unit test 

git-svn-id: http://code.djangoproject.com/svn/django/branches/magic-removal@1603 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Adrian Holovaty 2005-12-12 03:35:17 +00:00
parent 141a5382d8
commit 24cf934578
1 changed files with 5 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tests/testapp/models/custom_methods.py
View File

@ -1,23 +1,7 @@
""" """
3. Giving models custom methods and custom module-level functions 3. Giving models custom methods and custom managers
Any method you add to a model will be available to instances. Any method you add to a model will be available to instances.
Custom methods have the same namespace as if the model class were defined
in the dynamically-generated module. That is, methods can access
``get_list()``, ``get_object()``, ``AddManipulator``, and all other
module-level objects.
Also, custom methods have access to a few commonly-used objects for
convenience:
* The ``datetime`` module from Python's standard library.
* The ``db`` object from ``django.core.db``. This represents the database
connection, so you can do custom queries via a cursor object.
If your model method starts with "_module_", it'll be a module-level function
instead of a method. Otherwise, custom module-level functions have the same
namespace as custom methods.
""" """
from django.core import meta from django.core import meta
@ -33,22 +17,23 @@ class Article(meta.Model):
return self.pub_date == datetime.date.today() return self.pub_date == datetime.date.today()
def get_articles_from_same_day_1(self): def get_articles_from_same_day_1(self):
return get_list(id__ne=self.id, pub_date__exact=self.pub_date) return self.objects.get_list(id__ne=self.id, pub_date__exact=self.pub_date)
def get_articles_from_same_day_2(self): def get_articles_from_same_day_2(self):
""" """
Verbose version of get_articles_from_same_day_1, which does a custom Verbose version of get_articles_from_same_day_1, which does a custom
database query for the sake of demonstration. database query for the sake of demonstration.
""" """
from django.core.db import db
cursor = db.cursor() cursor = db.cursor()
cursor.execute(""" cursor.execute("""
SELECT id, headline, pub_date SELECT id, headline, pub_date
FROM custom_methods_articles FROM custom_methods_articles
WHERE pub_date = %s WHERE pub_date = %s
AND id != %s""", [str(self.pub_date), self.id]) AND id != %s""", [str(self.pub_date), self.id])
# The asterisk in "Article(*row)" tells Python to expand the list into # The asterisk in "(*row)" tells Python to expand the list into
# positional arguments to Article(). # positional arguments to Article().
return [Article(*row) for row in cursor.fetchall()] return [self.__class__(*row) for row in cursor.fetchall()]
API_TESTS = """ API_TESTS = """
# Create a couple of Articles. # Create a couple of Articles.