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 RDF_ANY_MISSING_METADATA = """
  25 <rdfq:Condition>
  26   <rdfq:or>
  27     <rdfq:equals>
  28       <rdfq:Property name="Audio:Artist" />
  29       <rdf:String></rdf:String>
  30     </rdfq:equals>
  31     <rdfq:equals>
  32       <rdfq:Property name="Audio:Title" />
  33       <rdf:String></rdf:String>
  34     </rdfq:equals>
  35     <rdfq:equals>
  36       <rdfq:Property name="Audio:Album" />
  37       <rdf:String></rdf:String>
  38     </rdfq:equals>
  39   </rdfq:or>
  40 </rdfq:Condition>
  41 """
  42
  43
  44 class TrackerBackend:
  45
  46     def __init__ (self):
  47         print "Tracker backend up"
  48         bus = dbus.SessionBus ()
  49         self.tracker_metadata = bus.get_object (TRACKER, TRACKER_OBJ)
  50         self.iface_metadata = dbus.Interface (self.tracker_metadata,
  51                                               "org.freedesktop.Tracker.Metadata")
  52
  53         self.tracker_search = bus.get_object (TRACKER, TRACKER_SEARCH_OBJ)
  54         self.iface_search = dbus.Interface (self.tracker_search,
  55                                             "org.freedesktop.Tracker.Search")
  56
  57     def count_songs_wo_artist (self):
  58         return self.iface_metadata.GetCount ("Music", "*", RDF_NO_ARTIST)
  59
  60     def count_songs_wo_title (self):
  61         return self.iface_metadata.GetCount ("Music", "*", RDF_NO_TITLE)
  62
  63     def count_songs_wo_album (self):
  64         return self.iface_metadata.GetCount ("Music", "*", RDF_NO_ALBUM)
  65
  66     def __run_rdf_query (self, rdf_query):
  67         results = self.iface_search.Query (-1, "Music",
  68                                            ["Audio:Artist",
  69                                             "Audio:Title",
  70                                             "Audio:Album",
  71                                             "File:Mime"],
  72                                            "", [], rdf_query, False,
  73                                            [], False, 0, 32000)
  74         return results
  75
  76     def get_all_broken_songs (self):
  77         """
  78         Return tuples with the following fields:
  79         (uri, "Music", artist, title, album, mimetype)
  80         """
  81         return self.__run_rdf_query (RDF_ANY_MISSING_METADATA)
  82
  83     def get_all_songs (self):
  84         return self.__run_rdf_query ("")
  85
  86
  87     def get_list_of_known_albums (self):
  88         return self.iface_metadata.GetUniqueValues ("Music",
  89                                                     ["Audio:Album"],
  90                                                     "", False, 0, 32000)
  91
  92     def get_list_of_known_artists (self):
  93         return self.iface_metadata.GetUniqueValues ("Music",
  94                                                     ["Audio:Artist"],
  95                                                     "", False, 0, 32000)
  96
  97     def get_all_pairs_artist_album (self):
  98         return self.iface_metadata.GetUniqueValues ("Music",
  99                                                     ["Audio:Artist", "Audio:Album"],
 100                                                     "", False, 0, 32000)
 101
 102 # Test
 103 if __name__ == "__main__":
 104
 105     tracker = TrackerBackend ()
 106
 107     print "Songs without artist: " + str(tracker.count_songs_wo_artist ())
 108
 109     results = tracker.get_songs_without_artist ()
 110     for r in results:
 111         print "'%s', '%s', '%s', '%s', '%s'" % (r[0], r[2], r[3], r[4], r[5])
 112
 113
 114     print "Songs without album " + str(tracker.count_songs_wo_album ())
 115     print "Songs without title " + str(tracker.count_songs_wo_title ())
 116
 117     albums = tracker.get_list_of_known_albums ()
 118     print "%d different albums" % (len (albums))
 119     for a in albums:
 120         print a[0]
 121
 122     print "\nAll songs:"
 123     print tracker.get_all_songs ()