Ejemplo n.º 1
0
    def get_xml(self):
        """Return a string containing the XML representation of the collection 
        and related items
        """
        impl = getDOMImplementation()
        doc = impl.createDocument(None, "telemeta", None)
        coll_doc = self.collection.to_dom()
        coll_node = doc.documentElement.appendChild(coll_doc.documentElement)
        coll_doc.unlink()
        items_node_name = MediaItem.get_dom_name() + "List"
        items_node = doc.createElement(items_node_name)
        coll_node.appendChild(items_node)

        items = self.collection.items.all()
        for item in items:
            if item.file:
                item.file = self.__get_media_filename(item)
            item_doc = item.to_dom()
            items_node.appendChild(item_doc.documentElement)
            item_doc.unlink()
        doc.normalize()

        # libxml2 has prettier output than xml.dom:
        tree = libxml2.parseDoc(doc.toxml(encoding="utf-8"))
        doc.unlink()
        xml = unicode(tree.serialize(encoding="utf-8", format=1), "utf-8")
        tree.free()

        return xml
Ejemplo n.º 2
0
    def get_xml(self):
        """Return a string containing the XML representation of the collection 
        and related items
        """
        impl = getDOMImplementation()
        doc = impl.createDocument(None, "telemeta", None)
        coll_doc = self.collection.to_dom()
        coll_node = doc.documentElement.appendChild(coll_doc.documentElement)
        coll_doc.unlink()
        items_node_name = MediaItem.get_dom_name() + "List"
        items_node = doc.createElement(items_node_name)
        coll_node.appendChild(items_node)

        items = self.collection.items.all()
        for item in items:
            if item.file:
                item.file = self.__get_media_filename(item)
            item_doc = item.to_dom()
            items_node.appendChild(item_doc.documentElement)
            item_doc.unlink()
        doc.normalize()

        # libxml2 has prettier output than xml.dom:
        tree = libxml2.parseDoc(doc.toxml(encoding="utf-8"))
        doc.unlink()
        xml = unicode(tree.serialize(encoding="utf-8", format=1), "utf-8")
        tree.free()

        return xml
Ejemplo n.º 3
0
    def get_xml(self):
        """Return a string containing the XML representation of the collection
        and related items
        """
        impl = getDOMImplementation()
        doc = impl.createDocument(None, "telemeta", None)
        coll_doc = self.collection.to_dom()
        coll_node = doc.documentElement.appendChild(coll_doc.documentElement)
        coll_doc.unlink()
        items_node_name = MediaItem.get_dom_name() + "List"
        items_node = doc.createElement(items_node_name)
        coll_node.appendChild(items_node)

        items = self.collection.items.all()
        for item in items:
            if item.file:
                item.file = self.__get_media_filename(item)
            item_doc = item.to_dom()
            items_node.appendChild(item_doc.documentElement)
            item_doc.unlink()
        doc.normalize()

        return doc.toxml()
Ejemplo n.º 4
0
    def get_xml(self):
        """Return a string containing the XML representation of the collection
        and related items
        """
        impl = getDOMImplementation()
        doc = impl.createDocument(None, "telemeta", None)
        coll_doc = self.collection.to_dom()
        coll_node = doc.documentElement.appendChild(coll_doc.documentElement)
        coll_doc.unlink()
        items_node_name = MediaItem.get_dom_name() + "List"
        items_node = doc.createElement(items_node_name)
        coll_node.appendChild(items_node)

        items = self.collection.items.all()
        for item in items:
            if item.file:
                item.file = self.__get_media_filename(item)
            item_doc = item.to_dom()
            items_node.appendChild(item_doc.documentElement)
            item_doc.unlink()
        doc.normalize()

        return doc.toxml()
Ejemplo n.º 5
0
    def setUp(self):
        "Create a test database based on objects created in Django"

#        self.david   = User.objects.create(username="******", level="user")
#        self.olivier = User.objects.create(username="******", level="admin")
        self.david   = User.objects.create_user(username="******")
        self.olivier = User.objects.create_user(username="******")

        self.country = LocationType.objects.create(code="country", name="Country")
        self.continent = LocationType.objects.create(code="continent", name="Continent")
        self.city = LocationType.objects.create(code="city", name="City")

        self.paris = Location.objects.create(name="Paris", type=Location.OTHER_TYPE, complete_type=self.city)
        self.france = Location.objects.create(name="France", type=Location.COUNTRY, complete_type=self.country)
        self.europe = Location.objects.create(name="Europe", type=Location.CONTINENT, complete_type=self.continent)
        self.belgique = Location.objects.create(name="Belgique", type=Location.COUNTRY, complete_type=self.country)

        self.europe.add_child(self.france)
        self.france.add_child(self.paris)
        self.europe.add_child(self.belgique)

        self.a = EthnicGroup.objects.create(value="a")
        self.b = EthnicGroup.objects.create(value="b")
        self.c = EthnicGroup.objects.create(value="c")
        self.d = EthnicGroup.objects.create(value="d")

        self.persepolis = MediaCollection(id=1, code="CNRSMH_E_1970_001_002", reference="A1", title="persepolis",
            creator="Abraham LINCOLN", collector="Friedrich HEINZ", year_published=2009, is_published=True,
            recorded_from_year=1970, recorded_to_year=1980)

        self.persepolis.set_revision(self.david)

        self.volonte = MediaCollection(id=2, reference="A2",  code="CNRSMH_I_1960_001", title="Volonté de puissance",
            creator="Friedrich NIETZSCHE", collector="Jean AMORA", year_published=1999,
            recorded_from_year=1960, recorded_to_year=2000)

        self.volonte.set_revision(self.olivier)

        self.nicolas = MediaCollection(id=3, reference="A3",  code="CNRSMH_I_1967_123", title="petit nicolas",
            creator="Georgette McKenic", collector="Paul MAILLE",  year_published=1999,
            recorded_from_year=1967, recorded_to_year=1968)

        self.nicolas.set_revision(self.olivier)

        self.item_1 = MediaItem(id=1, collection=self.persepolis, code="CNRSMH_E_1970_001_002_44",
            recorded_from_date="1971-01-12", recorded_to_date="1971-02-24", location=self.paris,
            ethnic_group=self.a, title="item 1", author="Mickael SHEPHERD", collector="Charles PREMIER",
            comment="comment 1")

        self.item_1.set_revision(self.david)

        self.item_2 = MediaItem(id=2, collection=self.volonte, code="CNRSMH_I_1960_001_129",
            recorded_from_date="1981-01-12", recorded_to_date="1991-02-24", location=self.france,
            ethnic_group=self.a, title="item 2", author="Rick ROLL", comment="comment 2")

        self.item_2.set_revision(self.david)

        self.item_3 = MediaItem(id=3, collection=self.nicolas, code="CNRSMH_I_1967_123_456_01",
            recorded_from_date="1968-01-12", recorded_to_date="1968-02-24", location=self.belgique,
            ethnic_group=self.b, title="item 3", author="John SMITH", collector="Paul CARLOS",
            comment="comment 3",  )

        self.item_3.set_revision(self.olivier)

        self.item_4 = MediaItem(id=4, collection=self.persepolis, code="CNRSMH_E_1970_001_002_22_33",
            recorded_from_date="1972-01-12", recorded_to_date="1972-02-24", location=self.europe,
            ethnic_group=self.a, title="item 4", alt_title="I4", author="Keanu REAVES",
            collector="Christina BARCELONA", comment="comment 4")

        self.item_4.set_revision(self.olivier)

        self.item_5 = MediaItem(id=5, collection=self.volonte,code="CNRSMH_I_1960_001_789_85_22",
            approx_duration="00:05:00", recorded_from_date="1978-01-12", recorded_to_date="1978-02-24",
            location=self.belgique, ethnic_group=self.a, title="item 5", alt_title="I5",
            author="Simon PAUL", collector="Javier BARDEM",
            comment="comment 5")

        self.item_5.set_revision(self.olivier)

        self.item_6 = MediaItem(id=6, collection=self.persepolis, code="CNRSMH_E_1970_001_002_90",
            recorded_from_date="1968-01-12", recorded_to_date="1968-02-11", location=self.france,
            ethnic_group=self.b, title="item 6", author="Paul ANDERSON",
            collector="Jim CARLSON", comment="comment 10000")

        self.item_6.set_revision(self.david)

        self.collections = MediaCollection.objects.all()
        self.items       = MediaItem.objects.all()
Ejemplo n.º 6
0
class CollectionItemTestCase(unittest.TestCase):
    pytestmark = pytest.mark.django_db

    def setUp(self):
        "Create a test database based on objects created in Django"

#        self.david   = User.objects.create(username="******", level="user")
#        self.olivier = User.objects.create(username="******", level="admin")
        self.david   = User.objects.create_user(username="******")
        self.olivier = User.objects.create_user(username="******")

        self.country = LocationType.objects.create(code="country", name="Country")
        self.continent = LocationType.objects.create(code="continent", name="Continent")
        self.city = LocationType.objects.create(code="city", name="City")

        self.paris = Location.objects.create(name="Paris", type=Location.OTHER_TYPE, complete_type=self.city)
        self.france = Location.objects.create(name="France", type=Location.COUNTRY, complete_type=self.country)
        self.europe = Location.objects.create(name="Europe", type=Location.CONTINENT, complete_type=self.continent)
        self.belgique = Location.objects.create(name="Belgique", type=Location.COUNTRY, complete_type=self.country)

        self.europe.add_child(self.france)
        self.france.add_child(self.paris)
        self.europe.add_child(self.belgique)

        self.a = EthnicGroup.objects.create(value="a")
        self.b = EthnicGroup.objects.create(value="b")
        self.c = EthnicGroup.objects.create(value="c")
        self.d = EthnicGroup.objects.create(value="d")

        self.persepolis = MediaCollection(id=1, code="CNRSMH_E_1970_001_002", reference="A1", title="persepolis",
            creator="Abraham LINCOLN", collector="Friedrich HEINZ", year_published=2009, is_published=True,
            recorded_from_year=1970, recorded_to_year=1980)

        self.persepolis.set_revision(self.david)

        self.volonte = MediaCollection(id=2, reference="A2",  code="CNRSMH_I_1960_001", title="Volonté de puissance",
            creator="Friedrich NIETZSCHE", collector="Jean AMORA", year_published=1999,
            recorded_from_year=1960, recorded_to_year=2000)

        self.volonte.set_revision(self.olivier)

        self.nicolas = MediaCollection(id=3, reference="A3",  code="CNRSMH_I_1967_123", title="petit nicolas",
            creator="Georgette McKenic", collector="Paul MAILLE",  year_published=1999,
            recorded_from_year=1967, recorded_to_year=1968)

        self.nicolas.set_revision(self.olivier)

        self.item_1 = MediaItem(id=1, collection=self.persepolis, code="CNRSMH_E_1970_001_002_44",
            recorded_from_date="1971-01-12", recorded_to_date="1971-02-24", location=self.paris,
            ethnic_group=self.a, title="item 1", author="Mickael SHEPHERD", collector="Charles PREMIER",
            comment="comment 1")

        self.item_1.set_revision(self.david)

        self.item_2 = MediaItem(id=2, collection=self.volonte, code="CNRSMH_I_1960_001_129",
            recorded_from_date="1981-01-12", recorded_to_date="1991-02-24", location=self.france,
            ethnic_group=self.a, title="item 2", author="Rick ROLL", comment="comment 2")

        self.item_2.set_revision(self.david)

        self.item_3 = MediaItem(id=3, collection=self.nicolas, code="CNRSMH_I_1967_123_456_01",
            recorded_from_date="1968-01-12", recorded_to_date="1968-02-24", location=self.belgique,
            ethnic_group=self.b, title="item 3", author="John SMITH", collector="Paul CARLOS",
            comment="comment 3",  )

        self.item_3.set_revision(self.olivier)

        self.item_4 = MediaItem(id=4, collection=self.persepolis, code="CNRSMH_E_1970_001_002_22_33",
            recorded_from_date="1972-01-12", recorded_to_date="1972-02-24", location=self.europe,
            ethnic_group=self.a, title="item 4", alt_title="I4", author="Keanu REAVES",
            collector="Christina BARCELONA", comment="comment 4")

        self.item_4.set_revision(self.olivier)

        self.item_5 = MediaItem(id=5, collection=self.volonte,code="CNRSMH_I_1960_001_789_85_22",
            approx_duration="00:05:00", recorded_from_date="1978-01-12", recorded_to_date="1978-02-24",
            location=self.belgique, ethnic_group=self.a, title="item 5", alt_title="I5",
            author="Simon PAUL", collector="Javier BARDEM",
            comment="comment 5")

        self.item_5.set_revision(self.olivier)

        self.item_6 = MediaItem(id=6, collection=self.persepolis, code="CNRSMH_E_1970_001_002_90",
            recorded_from_date="1968-01-12", recorded_to_date="1968-02-11", location=self.france,
            ethnic_group=self.b, title="item 6", author="Paul ANDERSON",
            collector="Jim CARLSON", comment="comment 10000")

        self.item_6.set_revision(self.david)

        self.collections = MediaCollection.objects.all()
        self.items       = MediaItem.objects.all()

    def tearDown(self):
        User.objects.all().delete()
        LocationType.objects.all().delete()
        Location.objects.all().delete()
        EthnicGroup.objects.all().delete()
        MediaCollection.objects.all().delete()
        MediaItem.objects.all().delete()

    def testQuickSearchOnCollections(self):
        "Test quick_search property of MediaCollection class"
        result = self.collections.quick_search("persepolis")
        self.assertEquals(len(result), 1)
        self.assertEquals(result[0], self.persepolis)
        self.assertEquals(self.collections.quick_search("nietzsche")[0], self.volonte)
        result = self.collections.quick_search("nicolas")
        self.assertEquals(result[0], self.nicolas)

    def testQuickSearchOnItems(self):
        "Test quick_search property of MediaItem class"
        result = self.items.quick_search("item").order_by("title")
        self.assertEquals(result[0], self.item_1)
        self.assertEquals(result[1], self.item_2)
        self.assertEquals(result[2], self.item_3)
        self.assertEquals(result[3], self.item_4)
        self.assertEquals(result[4], self.item_5)
        self.assertEquals(result[5], self.item_6)

    def testWordSearch(self):
        "Test quick_search property of MediaCollection class, specificly quick_search on collection title"
        result = self.collections.quick_search("volonté puissance")
        self.assertEquals(result[0], self.volonte)
        result = self.collections.quick_search("puissance volonté")
        self.assertEquals(result[0], self.volonte)
        result = self.collections.quick_search("volonte puissance")
        self.assertEquals(result[0], self.volonte)
        result = self.collections.quick_search("puissance volonte")
        self.assertEquals(result[0], self.volonte)

    def testLocationSearch(self):
        "Test by_country and by_continent properties of MediaCollection class"
        self.assertEquals(self.collections.by_location(self.france)[0], self.persepolis)
        self.assertEquals(self.collections.by_location(self.europe)[0], self.persepolis)
        self.assertEquals(self.collections.by_location(self.belgique).order_by("title")[0], self.nicolas)
        self.assertEquals(self.collections.by_location(self.belgique).order_by("title")[1], self.volonte)

    def testRecordingYear(self):
        "Test by_recording_year property of MediaCollection class"
        self.assertEquals(self.collections.by_recording_year(1970, 1980)[0], self.persepolis)
        result = self.collections.by_recording_year(1975).order_by("title")
        self.assertEquals(result[0], self.persepolis)
        self.assertEquals(result[1], self.volonte)

    def testPublishYearOnCollection(self):
        "Test by_publish_year property of MediaCollection class"
        result=self.collections.by_publish_year(1999).order_by("title")
        self.assertEquals(result[0], self.nicolas)
        self.assertEquals(result[1], self.volonte)

    def testEthnicGroup(self):
        "Test by_ethnic_group property of MediaCollection class"
        result=self.collections.by_ethnic_group(self.a).order_by("title")
        self.assertEquals(result[0], self.persepolis)
        self.assertEquals(result[1], self.volonte)

    def testRecordingDate(self):
        "Test by_recording_date property of MediaItem class"
        result = self.items.by_recording_date("1968-01-01", "1972-12-12").order_by("title")
        self.assertEquals(result[0], self.item_1)
        self.assertEquals(result[1], self.item_3)
        self.assertEquals(result[2], self.item_4)
        self.assertEquals(result[3], self.item_6)
        result = self.items.by_recording_date("1968-02-06").order_by("title")
        self.assertEquals(result[0], self.item_3)
        self.assertEquals(result[1], self.item_6)

    def testTitle(self):
        "Test by_title property of MediaItem class"
        result = self.items.by_title("item").order_by("title")
        self.assertEquals(result[0], self.item_1)
        self.assertEquals(result[1], self.item_2)
        self.assertEquals(result[2], self.item_3)
        self.assertEquals(result[3], self.item_4)
        self.assertEquals(result[4], self.item_5)
        self.assertEquals(result[5], self.item_6)
        self.item_2.title = ''
        self.item_2.save()
        self.item_5.title = ''
        self.item_5.save()
        result = self.items.by_title("volonté").order_by("title")
        self.assertEquals(result[0], self.item_2)
        self.assertEquals(result[1], self.item_5)
        result = self.items.by_title("puissance volonté").order_by("title")
        self.assertEquals(result[0], self.item_2)
        self.assertEquals(result[1], self.item_5)

    def testPublishYearOnItem(self):
        "Test by_publish_year property of MediaItem class"
        result = self.items.by_publish_year(1999).order_by("title")
        self.assertEquals(result[0], self.item_2)
        self.assertEquals(result[1], self.item_3)
        self.assertEquals(result[2], self.item_5)

    def testWordSearchCore(self):
        "Test word_search property of CoreQuerySet class"
        self.assertEquals(self.collections.word_search("title", "volonté")[0], self.volonte)
        self.assertEquals(self.collections.word_search("code", "CNRSMH_E_1970_001_002")[0], self.persepolis)
        self.assertEquals(self.items.word_search("code", "CNRSMH_E_1970_001_002_44")[0], self.item_1)
        result = self.items.word_search("comment", "comment").order_by("title")
        self.assertEquals(result[0], self.item_1)
        self.assertEquals(result[1], self.item_2)
        self.assertEquals(result[2], self.item_3)
        self.assertEquals(result[3], self.item_4)
        self.assertEquals(result[4], self.item_5)
        self.assertEquals(result[5], self.item_6)

    def testByChangeTimeOnCollection(self):
        "Test by_change_time property of MediaCollection class"
        now = datetime.now()
        print self.collections.all()
        print MediaCollection.objects.all()
        result = self.collections.by_change_time(now - timedelta(hours=1), now).order_by("title")
        self.assertEquals(result[0], self.persepolis)

    def testByChangeTimeOnItem(self):
        "Test by_change_time property of MediaItem class"
        now = datetime.now()
        result = self.items.by_change_time(now - timedelta(hours=1), now).order_by("title")
        self.assertEquals(result[0], self.item_1)

    def testWithoutCollection(self):
        "Test without_collection property of MediaItem class"
        self.assertEquals(self.items.without_collection().count(), 0)

    def testCodeRequired(self):
        "Test that a proper failure occur when a collection code isn't provided"
        c = MediaCollection()
        try:
            c.set_revision(self.olivier)
        except RequiredFieldError, e:
            self.assertEquals(e.field.name, 'code')
        else:
Ejemplo n.º 7
0
    def setUp(self):
        "Create a test database based on objects created in Django"

        #        self.david   = User.objects.create(username="******", level="user")
        #        self.olivier = User.objects.create(username="******", level="admin")
        self.david = User.objects.create_user(username="******")
        self.olivier = User.objects.create_user(username="******")

        self.country = LocationType.objects.create(code="country",
                                                   name="Country")
        self.continent = LocationType.objects.create(code="continent",
                                                     name="Continent")
        self.city = LocationType.objects.create(code="city", name="City")

        self.paris = Location.objects.create(name="Paris",
                                             type=Location.OTHER_TYPE,
                                             complete_type=self.city)
        self.france = Location.objects.create(name="France",
                                              type=Location.COUNTRY,
                                              complete_type=self.country)
        self.europe = Location.objects.create(name="Europe",
                                              type=Location.CONTINENT,
                                              complete_type=self.continent)
        self.belgique = Location.objects.create(name="Belgique",
                                                type=Location.COUNTRY,
                                                complete_type=self.country)

        self.europe.add_child(self.france)
        self.france.add_child(self.paris)
        self.europe.add_child(self.belgique)

        self.a = EthnicGroup.objects.create(value="a")
        self.b = EthnicGroup.objects.create(value="b")
        self.c = EthnicGroup.objects.create(value="c")
        self.d = EthnicGroup.objects.create(value="d")

        self.persepolis = MediaCollection(id=1,
                                          code="CNRSMH_E_1970_001_002",
                                          reference="A1",
                                          title="persepolis",
                                          creator="Abraham LINCOLN",
                                          collector="Friedrich HEINZ",
                                          year_published=2009,
                                          is_published=True,
                                          recorded_from_year=1970,
                                          recorded_to_year=1980)

        self.persepolis.set_revision(self.david)

        self.volonte = MediaCollection(id=2,
                                       reference="A2",
                                       code="CNRSMH_I_1960_001",
                                       title="Volonté de puissance",
                                       creator="Friedrich NIETZSCHE",
                                       collector="Jean AMORA",
                                       year_published=1999,
                                       recorded_from_year=1960,
                                       recorded_to_year=2000)

        self.volonte.set_revision(self.olivier)

        self.nicolas = MediaCollection(id=3,
                                       reference="A3",
                                       code="CNRSMH_I_1967_123",
                                       title="petit nicolas",
                                       creator="Georgette McKenic",
                                       collector="Paul MAILLE",
                                       year_published=1999,
                                       recorded_from_year=1967,
                                       recorded_to_year=1968)

        self.nicolas.set_revision(self.olivier)

        self.item_1 = MediaItem(id=1,
                                collection=self.persepolis,
                                code="CNRSMH_E_1970_001_002_44",
                                recorded_from_date="1971-01-12",
                                recorded_to_date="1971-02-24",
                                location=self.paris,
                                ethnic_group=self.a,
                                title="item 1",
                                author="Mickael SHEPHERD",
                                collector="Charles PREMIER",
                                comment="comment 1")

        self.item_1.set_revision(self.david)

        self.item_2 = MediaItem(id=2,
                                collection=self.volonte,
                                code="CNRSMH_I_1960_001_129",
                                recorded_from_date="1981-01-12",
                                recorded_to_date="1991-02-24",
                                location=self.france,
                                ethnic_group=self.a,
                                title="item 2",
                                author="Rick ROLL",
                                comment="comment 2")

        self.item_2.set_revision(self.david)

        self.item_3 = MediaItem(
            id=3,
            collection=self.nicolas,
            code="CNRSMH_I_1967_123_456_01",
            recorded_from_date="1968-01-12",
            recorded_to_date="1968-02-24",
            location=self.belgique,
            ethnic_group=self.b,
            title="item 3",
            author="John SMITH",
            collector="Paul CARLOS",
            comment="comment 3",
        )

        self.item_3.set_revision(self.olivier)

        self.item_4 = MediaItem(id=4,
                                collection=self.persepolis,
                                code="CNRSMH_E_1970_001_002_22_33",
                                recorded_from_date="1972-01-12",
                                recorded_to_date="1972-02-24",
                                location=self.europe,
                                ethnic_group=self.a,
                                title="item 4",
                                alt_title="I4",
                                author="Keanu REAVES",
                                collector="Christina BARCELONA",
                                comment="comment 4")

        self.item_4.set_revision(self.olivier)

        self.item_5 = MediaItem(id=5,
                                collection=self.volonte,
                                code="CNRSMH_I_1960_001_789_85_22",
                                approx_duration="00:05:00",
                                recorded_from_date="1978-01-12",
                                recorded_to_date="1978-02-24",
                                location=self.belgique,
                                ethnic_group=self.a,
                                title="item 5",
                                alt_title="I5",
                                author="Simon PAUL",
                                collector="Javier BARDEM",
                                comment="comment 5")

        self.item_5.set_revision(self.olivier)

        self.item_6 = MediaItem(id=6,
                                collection=self.persepolis,
                                code="CNRSMH_E_1970_001_002_90",
                                recorded_from_date="1968-01-12",
                                recorded_to_date="1968-02-11",
                                location=self.france,
                                ethnic_group=self.b,
                                title="item 6",
                                author="Paul ANDERSON",
                                collector="Jim CARLSON",
                                comment="comment 10000")

        self.item_6.set_revision(self.david)

        self.collections = MediaCollection.objects.all()
        self.items = MediaItem.objects.all()
Ejemplo n.º 8
0
class CollectionItemTestCase(unittest.TestCase):
    pytestmark = pytest.mark.django_db

    def setUp(self):
        "Create a test database based on objects created in Django"

        #        self.david   = User.objects.create(username="******", level="user")
        #        self.olivier = User.objects.create(username="******", level="admin")
        self.david = User.objects.create_user(username="******")
        self.olivier = User.objects.create_user(username="******")

        self.country = LocationType.objects.create(code="country",
                                                   name="Country")
        self.continent = LocationType.objects.create(code="continent",
                                                     name="Continent")
        self.city = LocationType.objects.create(code="city", name="City")

        self.paris = Location.objects.create(name="Paris",
                                             type=Location.OTHER_TYPE,
                                             complete_type=self.city)
        self.france = Location.objects.create(name="France",
                                              type=Location.COUNTRY,
                                              complete_type=self.country)
        self.europe = Location.objects.create(name="Europe",
                                              type=Location.CONTINENT,
                                              complete_type=self.continent)
        self.belgique = Location.objects.create(name="Belgique",
                                                type=Location.COUNTRY,
                                                complete_type=self.country)

        self.europe.add_child(self.france)
        self.france.add_child(self.paris)
        self.europe.add_child(self.belgique)

        self.a = EthnicGroup.objects.create(value="a")
        self.b = EthnicGroup.objects.create(value="b")
        self.c = EthnicGroup.objects.create(value="c")
        self.d = EthnicGroup.objects.create(value="d")

        self.persepolis = MediaCollection(id=1,
                                          code="CNRSMH_E_1970_001_002",
                                          reference="A1",
                                          title="persepolis",
                                          creator="Abraham LINCOLN",
                                          collector="Friedrich HEINZ",
                                          year_published=2009,
                                          is_published=True,
                                          recorded_from_year=1970,
                                          recorded_to_year=1980)

        self.persepolis.set_revision(self.david)

        self.volonte = MediaCollection(id=2,
                                       reference="A2",
                                       code="CNRSMH_I_1960_001",
                                       title="Volonté de puissance",
                                       creator="Friedrich NIETZSCHE",
                                       collector="Jean AMORA",
                                       year_published=1999,
                                       recorded_from_year=1960,
                                       recorded_to_year=2000)

        self.volonte.set_revision(self.olivier)

        self.nicolas = MediaCollection(id=3,
                                       reference="A3",
                                       code="CNRSMH_I_1967_123",
                                       title="petit nicolas",
                                       creator="Georgette McKenic",
                                       collector="Paul MAILLE",
                                       year_published=1999,
                                       recorded_from_year=1967,
                                       recorded_to_year=1968)

        self.nicolas.set_revision(self.olivier)

        self.item_1 = MediaItem(id=1,
                                collection=self.persepolis,
                                code="CNRSMH_E_1970_001_002_44",
                                recorded_from_date="1971-01-12",
                                recorded_to_date="1971-02-24",
                                location=self.paris,
                                ethnic_group=self.a,
                                title="item 1",
                                author="Mickael SHEPHERD",
                                collector="Charles PREMIER",
                                comment="comment 1")

        self.item_1.set_revision(self.david)

        self.item_2 = MediaItem(id=2,
                                collection=self.volonte,
                                code="CNRSMH_I_1960_001_129",
                                recorded_from_date="1981-01-12",
                                recorded_to_date="1991-02-24",
                                location=self.france,
                                ethnic_group=self.a,
                                title="item 2",
                                author="Rick ROLL",
                                comment="comment 2")

        self.item_2.set_revision(self.david)

        self.item_3 = MediaItem(
            id=3,
            collection=self.nicolas,
            code="CNRSMH_I_1967_123_456_01",
            recorded_from_date="1968-01-12",
            recorded_to_date="1968-02-24",
            location=self.belgique,
            ethnic_group=self.b,
            title="item 3",
            author="John SMITH",
            collector="Paul CARLOS",
            comment="comment 3",
        )

        self.item_3.set_revision(self.olivier)

        self.item_4 = MediaItem(id=4,
                                collection=self.persepolis,
                                code="CNRSMH_E_1970_001_002_22_33",
                                recorded_from_date="1972-01-12",
                                recorded_to_date="1972-02-24",
                                location=self.europe,
                                ethnic_group=self.a,
                                title="item 4",
                                alt_title="I4",
                                author="Keanu REAVES",
                                collector="Christina BARCELONA",
                                comment="comment 4")

        self.item_4.set_revision(self.olivier)

        self.item_5 = MediaItem(id=5,
                                collection=self.volonte,
                                code="CNRSMH_I_1960_001_789_85_22",
                                approx_duration="00:05:00",
                                recorded_from_date="1978-01-12",
                                recorded_to_date="1978-02-24",
                                location=self.belgique,
                                ethnic_group=self.a,
                                title="item 5",
                                alt_title="I5",
                                author="Simon PAUL",
                                collector="Javier BARDEM",
                                comment="comment 5")

        self.item_5.set_revision(self.olivier)

        self.item_6 = MediaItem(id=6,
                                collection=self.persepolis,
                                code="CNRSMH_E_1970_001_002_90",
                                recorded_from_date="1968-01-12",
                                recorded_to_date="1968-02-11",
                                location=self.france,
                                ethnic_group=self.b,
                                title="item 6",
                                author="Paul ANDERSON",
                                collector="Jim CARLSON",
                                comment="comment 10000")

        self.item_6.set_revision(self.david)

        self.collections = MediaCollection.objects.all()
        self.items = MediaItem.objects.all()

    def tearDown(self):
        User.objects.all().delete()
        LocationType.objects.all().delete()
        Location.objects.all().delete()
        EthnicGroup.objects.all().delete()
        MediaCollection.objects.all().delete()
        MediaItem.objects.all().delete()

    def testQuickSearchOnCollections(self):
        "Test quick_search property of MediaCollection class"
        result = self.collections.quick_search("persepolis")
        self.assertEquals(len(result), 1)
        self.assertEquals(result[0], self.persepolis)
        self.assertEquals(
            self.collections.quick_search("nietzsche")[0], self.volonte)
        result = self.collections.quick_search("nicolas")
        self.assertEquals(result[0], self.nicolas)

    def testQuickSearchOnItems(self):
        "Test quick_search property of MediaItem class"
        result = self.items.quick_search("item").order_by("title")
        self.assertEquals(result[0], self.item_1)
        self.assertEquals(result[1], self.item_2)
        self.assertEquals(result[2], self.item_3)
        self.assertEquals(result[3], self.item_4)
        self.assertEquals(result[4], self.item_5)
        self.assertEquals(result[5], self.item_6)

    def testWordSearch(self):
        "Test quick_search property of MediaCollection class, specificly quick_search on collection title"
        result = self.collections.quick_search("volonté puissance")
        self.assertEquals(result[0], self.volonte)
        result = self.collections.quick_search("puissance volonté")
        self.assertEquals(result[0], self.volonte)
        result = self.collections.quick_search("volonte puissance")
        self.assertEquals(result[0], self.volonte)
        result = self.collections.quick_search("puissance volonte")
        self.assertEquals(result[0], self.volonte)

    def testLocationSearch(self):
        "Test by_country and by_continent properties of MediaCollection class"
        self.assertEquals(
            self.collections.by_location(self.france)[0], self.persepolis)
        self.assertEquals(
            self.collections.by_location(self.europe)[0], self.persepolis)
        self.assertEquals(
            self.collections.by_location(self.belgique).order_by("title")[0],
            self.nicolas)
        self.assertEquals(
            self.collections.by_location(self.belgique).order_by("title")[1],
            self.volonte)

    def testRecordingYear(self):
        "Test by_recording_year property of MediaCollection class"
        self.assertEquals(
            self.collections.by_recording_year(1970, 1980)[0], self.persepolis)
        result = self.collections.by_recording_year(1975).order_by("title")
        self.assertEquals(result[0], self.persepolis)
        self.assertEquals(result[1], self.volonte)

    def testPublishYearOnCollection(self):
        "Test by_publish_year property of MediaCollection class"
        result = self.collections.by_publish_year(1999).order_by("title")
        self.assertEquals(result[0], self.nicolas)
        self.assertEquals(result[1], self.volonte)

    def testEthnicGroup(self):
        "Test by_ethnic_group property of MediaCollection class"
        result = self.collections.by_ethnic_group(self.a).order_by("title")
        self.assertEquals(result[0], self.persepolis)
        self.assertEquals(result[1], self.volonte)

    def testRecordingDate(self):
        "Test by_recording_date property of MediaItem class"
        result = self.items.by_recording_date("1968-01-01",
                                              "1972-12-12").order_by("title")
        self.assertEquals(result[0], self.item_1)
        self.assertEquals(result[1], self.item_3)
        self.assertEquals(result[2], self.item_4)
        self.assertEquals(result[3], self.item_6)
        result = self.items.by_recording_date("1968-02-06").order_by("title")
        self.assertEquals(result[0], self.item_3)
        self.assertEquals(result[1], self.item_6)

    def testTitle(self):
        "Test by_title property of MediaItem class"
        result = self.items.by_title("item").order_by("title")
        self.assertEquals(result[0], self.item_1)
        self.assertEquals(result[1], self.item_2)
        self.assertEquals(result[2], self.item_3)
        self.assertEquals(result[3], self.item_4)
        self.assertEquals(result[4], self.item_5)
        self.assertEquals(result[5], self.item_6)
        self.item_2.title = ''
        self.item_2.save()
        self.item_5.title = ''
        self.item_5.save()
        result = self.items.by_title("volonté").order_by("title")
        self.assertEquals(result[0], self.item_2)
        self.assertEquals(result[1], self.item_5)
        result = self.items.by_title("puissance volonté").order_by("title")
        self.assertEquals(result[0], self.item_2)
        self.assertEquals(result[1], self.item_5)

    def testPublishYearOnItem(self):
        "Test by_publish_year property of MediaItem class"
        result = self.items.by_publish_year(1999).order_by("title")
        self.assertEquals(result[0], self.item_2)
        self.assertEquals(result[1], self.item_3)
        self.assertEquals(result[2], self.item_5)

    def testWordSearchCore(self):
        "Test word_search property of CoreQuerySet class"
        self.assertEquals(
            self.collections.word_search("title", "volonté")[0], self.volonte)
        self.assertEquals(
            self.collections.word_search("code", "CNRSMH_E_1970_001_002")[0],
            self.persepolis)
        self.assertEquals(
            self.items.word_search("code", "CNRSMH_E_1970_001_002_44")[0],
            self.item_1)
        result = self.items.word_search("comment", "comment").order_by("title")
        self.assertEquals(result[0], self.item_1)
        self.assertEquals(result[1], self.item_2)
        self.assertEquals(result[2], self.item_3)
        self.assertEquals(result[3], self.item_4)
        self.assertEquals(result[4], self.item_5)
        self.assertEquals(result[5], self.item_6)

    def testByChangeTimeOnCollection(self):
        "Test by_change_time property of MediaCollection class"
        now = datetime.now()
        print self.collections.all()
        print MediaCollection.objects.all()
        result = self.collections.by_change_time(now - timedelta(hours=1),
                                                 now).order_by("title")
        self.assertEquals(result[0], self.persepolis)

    def testByChangeTimeOnItem(self):
        "Test by_change_time property of MediaItem class"
        now = datetime.now()
        result = self.items.by_change_time(now - timedelta(hours=1),
                                           now).order_by("title")
        self.assertEquals(result[0], self.item_1)

    def testWithoutCollection(self):
        "Test without_collection property of MediaItem class"
        self.assertEquals(self.items.without_collection().count(), 0)

    def testCodeRequired(self):
        "Test that a proper failure occur when a collection code isn't provided"
        c = MediaCollection()
        try:
            c.set_revision(self.olivier)
        except RequiredFieldError, e:
            self.assertEquals(e.field.name, 'code')
        else:
Ejemplo n.º 9
0
    def wav_import(self):
        from telemeta.models import MediaItem, MediaCollection

        collections = []
        for collection in self.collections:
            collection_dir = self.source_dir + os.sep + collection
            collection_files = os.listdir(collection_dir)

            if not '/.' in collection_dir and self.pattern in collection_dir:
                collection_name = collection.split(os.sep)[-1]
                collections.append(collection_name)
                c = MediaCollection.objects.filter(code=collection_name)

                if not c and collection + '.csv' in collection_files:
                    msg = collection + ' collection NON présente dans la base de données, SORTIE '
                    self.logger.error(collection, msg)
                    sys.exit(msg)
                elif not c:
                    msg = 'collection NON présente dans la base de données, CREATION '
                    self.logger.info(collection, msg)
                    c = MediaCollection(code=collection_name,
                                        title=collection_name)
                    c.save()
                    c.set_revision(self.user)
                else:
                    msg = 'collection présente dans la base de données, SELECTION'
                    self.logger.info(collection, msg)

        for collection in collections:
            collection_dir = self.source_dir + os.sep + collection
            collection_name = collection
            collection_files = os.listdir(collection_dir)
            msg = '************************ ' + collection + ' ******************************'
            self.logger.info(collection, msg[:70])
            overwrite = True
            csv_file = ''
            rows = {}

            if collection + '.csv' in collection_files:
                csv_file = self.source_dir + os.sep + collection + os.sep + collection + '.csv'
                csv_data = csv.reader(open(csv_file), delimiter=';')
                for row in csv_data:
                    rows[row[1].strip()] = row[0].strip()
                msg = collection + ' import du fichier CSV de la collection'
                self.logger.info(collection, msg[:70])
            else:
                msg = collection + ' pas de fichier CSV dans la collection'
                self.logger.info(collection, msg[:70])

            c = MediaCollection.objects.filter(code=collection_name)
            if not c:
                c = MediaCollection(code=collection_name)
                c.save()
                msg = ' collection NON présente dans la BDD, CREATION '
                self.logger.info(c.code, msg)
            else:
                c = c[0]
                msg = ' id = ' + str(c.id)
                self.logger.info(c.code, msg)

            audio_files = []
            for file in collection_files:
                ext = ['WAV', 'wav']
                if file.split('.')[-1] in ext and file[0] != '.':
                    audio_files.append(file)

            audio_files.sort()
            nb_items = c.items.count()
            counter = 0

            for file in audio_files:
                code = file.split('.')[0]
                wav_file = self.source_dir + os.sep + collection + os.sep + file

                if len(audio_files) <= nb_items:
                    items = MediaItem.objects.filter(code=code)

                    old_ref = ''
                    if code in rows and not items:
                        old_ref = rows[code]
                        items = MediaItem.objects.filter(old_code=old_ref)

                    if items:
                        item = items[0]
                        msg = code + ' : ' + item.old_code + ' : Cas 1 ou 2 : id = ' + str(
                            item.id)
                        self.logger.info('item', msg)
                        item.code = code
                        item.save()
                    else:
                        item = MediaItem(code=code, collection=c)
                        msg = code + ' : ' + old_ref + ' : Cas 1 ou 2 : item NON présent dans la base de données, CREATION'
                        self.logger.info('item', msg)

                    self.write_file(item, wav_file, overwrite)

                elif nb_items == 1 and len(audio_files) > 1:
                    if counter == 0:
                        msg = code + ' : Cas 3a : item n°01 présent dans la base de données, PASSE'
                        self.logger.info('item', msg)
                    else:
                        item = MediaItem(code=code, collection=c)
                        msg = code + ' : Cas 3a : item NON présent dans la base de données, CREATION'
                        self.logger.info('item', msg)
                        self.write_file(item, wav_file, overwrite)

                elif nb_items > 1 and nb_items < len(audio_files):
                    msg = code + ' : Cas 3b : nb items < nb de fichiers audio, PAS de creation'
                    self.logger.info('item', msg)

                counter += 1

        msg = 'Liste des URLs des collections importées :'
        self.logger.info('INFO', msg)
        for collection in collections:
            msg = 'http://' + self.domain + '/archives/collections/' + collection
            self.logger.info(collection, msg)