64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
from django.contrib.gis import feeds
|
|
from django.contrib.gis.tests.utils import mysql
|
|
from models import City, Country
|
|
|
|
class TestGeoRSS1(feeds.Feed):
|
|
link = '/city/'
|
|
title = 'Test GeoDjango Cities'
|
|
|
|
def items(self):
|
|
return City.objects.all()
|
|
|
|
def item_link(self, item):
|
|
return '/city/%s/' % item.pk
|
|
|
|
def item_geometry(self, item):
|
|
return item.point
|
|
|
|
class TestGeoRSS2(TestGeoRSS1):
|
|
def geometry(self, obj):
|
|
# This should attach a <georss:box> element for the extent of
|
|
# of the cities in the database. This tuple came from
|
|
# calling `City.objects.extent()` -- we can't do that call here
|
|
# because `extent` is not implemented for MySQL/Oracle.
|
|
return (-123.30, -41.32, 174.78, 48.46)
|
|
|
|
def item_geometry(self, item):
|
|
# Returning a simple tuple for the geometry.
|
|
return item.point.x, item.point.y
|
|
|
|
class TestGeoAtom1(TestGeoRSS1):
|
|
feed_type = feeds.GeoAtom1Feed
|
|
|
|
class TestGeoAtom2(TestGeoRSS2):
|
|
feed_type = feeds.GeoAtom1Feed
|
|
|
|
def geometry(self, obj):
|
|
# This time we'll use a 2-tuple of coordinates for the box.
|
|
return ((-123.30, -41.32), (174.78, 48.46))
|
|
|
|
class TestW3CGeo1(TestGeoRSS1):
|
|
feed_type = feeds.W3CGeoFeed
|
|
|
|
# The following feeds are invalid, and will raise exceptions.
|
|
class TestW3CGeo2(TestGeoRSS2):
|
|
feed_type = feeds.W3CGeoFeed
|
|
|
|
class TestW3CGeo3(TestGeoRSS1):
|
|
feed_type = feeds.W3CGeoFeed
|
|
|
|
def item_geometry(self, item):
|
|
from django.contrib.gis.geos import Polygon
|
|
return Polygon(((0, 0), (0, 1), (1, 1), (1, 0), (0, 0)))
|
|
|
|
# The feed dictionary to use for URLs.
|
|
feed_dict = {
|
|
'rss1' : TestGeoRSS1,
|
|
'rss2' : TestGeoRSS2,
|
|
'atom1' : TestGeoAtom1,
|
|
'atom2' : TestGeoAtom2,
|
|
'w3cgeo1' : TestW3CGeo1,
|
|
'w3cgeo2' : TestW3CGeo2,
|
|
'w3cgeo3' : TestW3CGeo3,
|
|
}
|