vcs.maemo.org Git - mussorgsky/blob - src/tracker_backend.py

   1 #!/usr/bin/env python2.5
   2 import dbus
   3 import os
   4
   5 TRACKER = 'org.freedesktop.Tracker'
   6 TRACKER_OBJ = '/org/freedesktop/Tracker/Metadata'
   7 TRACKER_SEARCH_OBJ = '/org/freedesktop/Tracker/Search'
   8
   9 RDF_NO_PROPX = """
  10 <rdfq:Condition>
  11   <rdfq:and>
  12     <rdfq:equals>
  13       <rdfq:Property name="%s" />
  14       <rdf:String></rdf:String>
  15     </rdfq:equals>
  16   </rdfq:and>
  17 </rdfq:Condition>
  18 """
  19
  20 RDF_NO_ARTIST = RDF_NO_PROPX % "Audio:Artist"
  21 RDF_NO_ALBUM = RDF_NO_PROPX % "Audio:Album"
  22 RDF_NO_TITLE = RDF_NO_PROPX % "Audio:Title"
  23
  24 class TrackerBackend:
  25
  26     def __init__ (self):
  27         print "Tracker backend up"
  28         bus = dbus.SessionBus ()
  29         self.tracker_metadata = bus.get_object (TRACKER, TRACKER_OBJ)
  30         self.iface_metadata = dbus.Interface (self.tracker_metadata,
  31                                               "org.freedesktop.Tracker.Metadata")
  32
  33         self.tracker_search = bus.get_object (TRACKER, TRACKER_SEARCH_OBJ)
  34         self.iface_search = dbus.Interface (self.tracker_search,
  35                                             "org.freedesktop.Tracker.Search")
  36
  37     def count_songs_wo_artist (self):
  38         return self.iface_metadata.GetCount ("Music", "*", RDF_NO_ARTIST)
  39
  40     def count_songs_wo_title (self):
  41         return self.iface_metadata.GetCount ("Music", "*", RDF_NO_TITLE)
  42
  43     def count_songs_wo_album (self):
  44         return self.iface_metadata.GetCount ("Music", "*", RDF_NO_ALBUM)
  45
  46     def __run_rdf_query (self, rdf_query):
  47         results = self.iface_search.Query (-1, "Music",
  48                                            ["Audio:Artist",
  49                                             "Audio:Title",
  50                                             "Audio:Album",
  51                                             "File:Mime"],
  52                                            "", [], rdf_query, False,
  53                                            [], False, 0, 32000)
  54         return results
  55
  56
  57     def get_songs_without_artist (self):
  58         """
  59         Return tuples with the following fields:
  60         (uri, "Music", artist, title, album, mimetype)
  61         """
  62         return self.__run_rdf_query (RDF_NO_ARTIST)
  63
  64     def get_songs_without_title (self):
  65         """
  66         Return tuples with the following fields:
  67         (uri, "Music", artist, title, album, mimetype)
  68         """
  69         return self.__run_rdf_query (RDF_NO_TITLE)
  70
  71     def get_songs_without_album (self):
  72         """
  73         Return tuples with the following fields:
  74         (uri, "Music", artist, title, album, mimetype)
  75         """
  76         return self.__run_rdf_query (RDF_NO_ALBUM)
  77
  78     def get_list_of_known_albums (self):
  79         return self.iface_metadata.GetUniqueValues ("Music",
  80                                                     ["Audio:Album"],
  81                                                     "", False, 0, 32000)
  82
  83     def get_list_of_known_artists (self):
  84         return self.iface_metadata.GetUniqueValues ("Music",
  85                                                     ["Audio:Artist"],
  86                                                     "", False, 0, 32000)
  87
  88 # Test
  89 if __name__ == "__main__":
  90
  91     tracker = TrackerBackend ()
  92
  93     print "Songs without artist: " + str(tracker.count_songs_wo_artist ())
  94
  95     results = tracker.get_songs_without_artist ()
  96     for r in results:
  97         print "'%s', '%s', '%s', '%s', '%s'" % (r[0], r[2], r[3], r[4], r[5])
  98
  99
 100     print "Songs without album " + str(tracker.count_songs_wo_album ())
 101     print "Songs without title " + str(tracker.count_songs_wo_title ())
 102
 103     albums = tracker.get_list_of_known_albums ()
 104     print "%d different albums" % (len (albums))
 105     for a in albums:
 106         print a[0]
 107