chiark / gitweb /
7b888b5455bed96505e268c9f4c66949a39cc46a
[disorder] / python / disorder.py.in
1 #
2 # Copyright (C) 2004, 2005, 2007 Richard Kettlewell
3 #
4 # This program is free software; you can redistribute it and/or modify
5 # it under the terms of the GNU General Public License as published by
6 # the Free Software Foundation; either version 2 of the License, or
7 # (at your option) any later version.
8 #
9 # This program is distributed in the hope that it will be useful, but
10 # WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12 # General Public License for more details.
13 #
14 # You should have received a copy of the GNU General Public License
15 # along with this program; if not, write to the Free Software
16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
17 # USA
18 #
19
20 """Python support for DisOrder
21
22 Provides disorder.client, a class for accessing a DisOrder server.
23
24 Example 1:
25
26   #! /usr/bin/env python
27   import disorder
28   d = disorder.client()
29   p = d.playing()
30   if p:
31     print p['track']
32
33 Example 2:
34
35   #! /usr/bin/env python
36   import disorder
37   import sys
38   d = disorder.client()
39   for path in sys.argv[1:]:
40     d.play(path)
41
42 See disorder_protocol(5) for details of the communication protocol.
43
44 NB that this code only supports servers configured to use SHA1-based
45 authentication.  If the server demands another hash then it will not be
46 possible to use this module.
47 """
48
49 import re
50 import string
51 import os
52 import pwd
53 import socket
54 import binascii
55 import sha
56 import sys
57 import locale
58
59 _configfile = "pkgconfdir/config"
60 _dbhome = "pkgstatedir"
61 _userconf = True
62
63 # various regexps we'll use
64 _ws = re.compile(r"^[ \t\n\r]+")
65 _squote = re.compile("'(([^\\\\']|\\\\[\\\\\"'n])+)'")
66 _dquote = re.compile("\"(([^\\\\\"]|\\\\[\\\\\"'n])+)\"")
67 _unquoted = re.compile("[^\"' \\t\\n\\r][^ \t\n\r]*")
68
69 _response = re.compile("([0-9]{3}) ?(.*)")
70
71 version = "_version_"
72
73 ########################################################################
74 # exception classes
75
76 class Error(Exception):
77   """Base class for DisOrder exceptions."""
78
79 class _splitError(Error):
80   # _split failed
81   def __init__(self, value):
82     self.value = value
83   def __str__(self):
84     return str(self.value)
85
86 class parseError(Error):
87   """Error parsing the configuration file."""
88   def __init__(self, path, line, details):
89     self.path = path
90     self.line = line
91     self.details = details
92   def __str__(self):
93     return "%s:%d: %s" % (self.path, self.line, self.details)
94
95 class protocolError(Error):
96   """DisOrder control protocol error.
97
98   Indicates a mismatch between the client and server's understanding of
99   the control protocol.
100   """
101   def __init__(self, who, error):
102     self.who = who
103     self.error = error
104   def __str__(self):
105     return "%s: %s" % (self.who, str(self.error))
106
107 class operationError(Error):
108   """DisOrder control protocol error response.
109
110   Indicates that an operation failed (e.g. an attempt to play a
111   nonexistent track).  The connection should still be usable.
112   """
113   def __init__(self, res, details, cmd=None):
114     self.res_ = int(res)
115     self.cmd_ = cmd
116     self.details_ = details
117   def __str__(self):
118     """Return the complete response string from the server, with the command
119     if available.
120
121     Excludes the final newline.
122     """
123     if self.cmd_ is None:
124       return "%d %s" % (self.res_, self.details_)
125     else:
126       return "%d %s [%s]" % (self.res_, self.details_, self.cmd_)
127   def response(self):
128     """Return the response code from the server."""
129     return self.res_
130   def details(self):
131     """Returns the detail string from the server."""
132     return self.details_
133
134 class communicationError(Error):
135   """DisOrder control protocol communication error.
136
137   Indicates that communication with the server went wrong, perhaps
138   because the server was restarted.  The caller could report an error to
139   the user and wait for further user instructions, or even automatically
140   retry the operation.
141   """
142   def __init__(self, who, error):
143     self.who = who
144     self.error = error
145   def __str__(self):
146     return "%s: %s" % (self.who, str(self.error))
147
148 ########################################################################
149 # DisOrder-specific text processing
150
151 def _unescape(s):
152   # Unescape the contents of a string
153   #
154   # Arguments:
155   #
156   # s -- string to unescape
157   #
158   s = re.sub("\\\\n", "\n", s)
159   s = re.sub("\\\\(.)", "\\1", s)
160   return s
161
162 def _split(s, *comments):
163   # Split a string into fields according to the usual Disorder string splitting
164   # conventions.
165   #
166   # Arguments:
167   #
168   # s        -- string to parse
169   # comments -- if present, parse comments
170   #
171   # Return values:
172   #
173   # On success, a list of fields is returned.
174   #
175   # On error, disorder.parseError is thrown.
176   #
177   fields = []
178   while s != "":
179     # discard comments
180     if comments and s[0] == '#':
181       break
182     # strip spaces
183     m = _ws.match(s)
184     if m:
185       s = s[m.end():]
186       continue
187     # pick of quoted fields of both kinds
188     m = _squote.match(s)
189     if not m:
190       m = _dquote.match(s)
191     if m:
192       fields.append(_unescape(m.group(1)))
193       s = s[m.end():]
194       continue
195     # and unquoted fields
196     m = _unquoted.match(s)
197     if m:
198       fields.append(m.group(0))
199       s = s[m.end():]
200       continue
201     # anything left must be in error
202     if s[0] == '"' or s[0] == '\'':
203       raise _splitError("invalid quoted string")
204     else:
205       raise _splitError("syntax error")
206   return fields
207
208 def _escape(s):
209   # Escape the contents of a string
210   #
211   # Arguments:
212   #
213   # s -- string to escape
214   #
215   if re.search("[\\\\\"'\n \t\r]", s) or s == '':
216     s = re.sub(r'[\\"]', r'\\\g<0>', s)
217     s = re.sub("\n", r"\\n", s)
218     return '"' + s + '"'
219   else:
220     return s
221
222 def _quote(list):
223   # Quote a list of values
224   return ' '.join(map(_escape, list))
225
226 def _sanitize(s):
227   # Return the value of s in a form suitable for writing to stderr
228   return s.encode(locale.nl_langinfo(locale.CODESET), 'replace')
229
230 def _list2dict(l):
231   # Convert a list of the form [k1, v1, k2, v2, ..., kN, vN]
232   # to a dictionary {k1:v1, k2:v2, ..., kN:vN}
233   d = {}
234   i = iter(l)
235   try:
236     while True:
237       k = i.next()
238       v = i.next()
239       d[str(k)] = v
240   except StopIteration:
241     pass
242   return d
243
244 def _queueEntry(s):
245   # parse a queue entry
246   return _list2dict(_split(s))
247
248 ########################################################################
249 # The client class
250
251 class client:
252   """DisOrder client class.
253
254   This class provides access to the DisOrder server either on this
255   machine or across the internet.
256
257   The server to connect to, and the username and password to use, are
258   determined from the configuration files as described in 'man
259   disorder_config'.
260
261   All methods will connect if necessary, as soon as you have a
262   disorder.client object you can start calling operational methods on
263   it.
264
265   However if the server is restarted then the next method called on a
266   connection will throw an exception.  This may be considered a bug.
267
268   All methods block until they complete.
269
270   Operation methods raise communicationError if the connection breaks,
271   protocolError if the response from the server is malformed, or
272   operationError if the response is valid but indicates that the
273   operation failed.
274   """
275
276   debug_proto = 0x0001
277   debug_body = 0x0002
278
279   def __init__(self, user=None, password=None):
280     """Constructor for DisOrder client class.
281
282     The constructor reads the configuration file, but does not connect
283     to the server.
284
285     If the environment variable DISORDER_PYTHON_DEBUG is set then the
286     debug flags are initialised to that value.  This can be overridden
287     with the debug() method below.
288
289     The constructor Raises parseError() if the configuration file is not
290     valid.
291     """
292     pw = pwd.getpwuid(os.getuid())
293     self.debugging = int(os.getenv("DISORDER_PYTHON_DEBUG", 0))
294     self.config = { 'collections': [],
295                     'username': pw.pw_name,
296                     'home': _dbhome }
297     self.user = user
298     self.password = password
299     home = os.getenv("HOME")
300     if not home:
301       home = pw.pw_dir
302     privconf = _configfile + "." + pw.pw_name
303     passfile = home + os.sep + ".disorder" + os.sep + "passwd"
304     if os.path.exists(_configfile):
305       self._readfile(_configfile)
306     if os.path.exists(privconf):
307       self._readfile(privconf)
308     if os.path.exists(passfile) and _userconf:
309       self._readfile(passfile)
310     self.state = 'disconnected'
311
312   def debug(self, bits):
313     """Enable or disable protocol debugging.  Debug messages are written
314     to sys.stderr.
315
316     Arguments:
317     bits -- bitmap of operations that should generate debug information
318
319     Bitmap values:
320     debug_proto -- dump control protocol messages (excluding bodies)
321     debug_body -- dump control protocol message bodies
322     """
323     self.debugging = bits
324
325   def _debug(self, bit, s):
326     # debug output
327     if self.debugging & bit:
328       sys.stderr.write(_sanitize(s))
329       sys.stderr.write("\n")
330       sys.stderr.flush()
331
332   def connect(self, cookie=None):
333     """c.connect(cookie=None)
334
335     Connect to the DisOrder server and authenticate.
336
337     Raises communicationError if connection fails and operationError if
338     authentication fails (in which case disconnection is automatic).
339
340     May be called more than once to retry connections (e.g. when the
341     server is down).  If we are already connected and authenticated,
342     this is a no-op.
343
344     Other operations automatically connect if we're not already
345     connected, so it is not strictly necessary to call this method.
346
347     If COOKIE is specified then that is used to log in instead of
348     the username/password.
349     """
350     if self.state == 'disconnected':
351       try:
352         self.state = 'connecting'
353         if 'connect' in self.config and len(self.config['connect']) > 0:
354           c = self.config['connect']
355           self.who = repr(c)            # temporarily
356           if len(c) == 1:
357             a = socket.getaddrinfo(None, c[0],
358                                    socket.AF_INET,
359                                    socket.SOCK_STREAM,
360                                    0,
361                                    0)
362           else:
363             a = socket.getaddrinfo(c[0], c[1],
364                                    socket.AF_INET,
365                                    socket.SOCK_STREAM,
366                                    0,
367                                    0)
368           a = a[0]
369           s = socket.socket(a[0], a[1], a[2]);
370           s.connect(a[4])
371           self.who = "%s" % a[3]
372         else:
373           s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM);
374           self.who = self.config['home'] + os.sep + "socket"
375           s.connect(self.who)
376         self.w = s.makefile("wb")
377         self.r = s.makefile("rb")
378         (res, challenge_and_algo) = self._simple()
379         (algo, challenge) = _split(challenge_and_algo)
380         if cookie is None:
381           if self.user is None:
382             user = self.config['username']
383           else:
384             user = self.user
385           if self.password is None:
386             password = self.config['password']
387           else:
388             password = self.password
389           # TODO support algorithms other than SHA-1
390           h = sha.sha()
391           h.update(password)
392           h.update(binascii.unhexlify(challenge))
393           self._simple("user", user, h.hexdigest())
394         else:
395           self._simple("cookie", cookie)
396         self.state = 'connected'
397       except socket.error, e:
398         self._disconnect()
399         raise communicationError(self.who, e)
400       except:
401         self._disconnect()
402         raise
403
404   def _disconnect(self):
405     # disconnect from the server, whatever state we are in
406     try:
407       del self.w
408       del self.r
409     except:
410       pass
411     self.state = 'disconnected'
412     
413   ########################################################################
414   # Operations
415
416   def play(self, track):
417     """Play a track.
418
419     Arguments:
420     track -- the path of the track to play.
421
422     Returns the ID of the new queue entry.
423
424     Note that queue IDs are unicode strings (because all track information
425     values are unicode strings).
426     """
427     res, details = self._simple("play", track)
428     return unicode(details)             # because it's unicode in queue() output
429
430   def remove(self, track):
431     """Remove a track from the queue.
432
433     Arguments:
434     track -- the path or ID of the track to remove.
435     """
436     self._simple("remove", track)
437
438   def enable(self):
439     """Enable playing."""
440     self._simple("enable")
441
442   def disable(self, *now):
443     """Disable playing.
444
445     Arguments:
446     now -- if present (with any value), the current track is stopped
447            too.
448     """
449     if now:
450       self._simple("disable", "now")
451     else:
452       self._simple("disable")
453
454   def scratch(self, *id):
455     """Scratch the currently playing track.
456
457     Arguments:
458     id -- if present, the ID of the track to scratch.
459     """
460     if id:
461       self._simple("scratch", id[0])
462     else:
463       self._simple("scratch")
464
465   def shutdown(self):
466     """Shut down the server.
467
468     Only trusted users can perform this operation.
469     """
470     self._simple("shutdown")
471
472   def reconfigure(self):
473     """Make the server reload its configuration.
474
475     Only trusted users can perform this operation.
476     """
477     self._simple("reconfigure")
478
479   def rescan(self, pattern):
480     """Rescan one or more collections.
481
482     Arguments:
483     pattern -- glob pattern matching collections to rescan.
484
485     Only trusted users can perform this operation.
486     """
487     self._simple("rescan", pattern)
488
489   def version(self):
490     """Return the server's version number."""
491     return self._simple("version")[1]
492
493   def playing(self):
494     """Return the currently playing track.
495
496     If a track is playing then it is returned as a dictionary.  See
497     disorder_protocol(5) for the meanings of the keys.  All keys are
498     plain strings but the values will be unicode strings.
499     
500     If no track is playing then None is returned."""
501     res, details = self._simple("playing")
502     if res % 10 != 9:
503       try:
504         return _queueEntry(details)
505       except _splitError, s:
506         raise protocolError(self.who, s.str())
507     else:
508       return None
509
510   def _somequeue(self, command):
511     self._simple(command)
512     try:
513       return map(lambda s: _queueEntry(s), self._body())
514     except _splitError, s:
515       raise protocolError(self.who, s.str())
516
517   def recent(self):
518     """Return a list of recently played tracks.
519
520     The return value is a list of dictionaries corresponding to
521     recently played tracks.  The oldest track comes first.
522
523     See disorder_protocol(5) for the meanings of the keys.  All keys are
524     plain strings but the values will be unicode strings."""
525     return self._somequeue("recent")
526
527   def queue(self):
528     """Return the current queue.
529
530     The return value is a list of dictionaries corresponding to
531     recently played tracks.  The next track to be played comes first.
532
533     See disorder_protocol(5) for the meanings of the keys.  All keys are
534     plain strings but the values will be unicode strings."""
535     return self._somequeue("queue")
536
537   def _somedir(self, command, dir, re):
538     if re:
539       self._simple(command, dir, re[0])
540     else:
541       self._simple(command, dir)
542     return self._body()
543
544   def directories(self, dir, *re):
545     """List subdirectories of a directory.
546
547     Arguments:
548     dir -- directory to list, or '' for the whole root.
549     re -- regexp that results must match.  Optional.
550
551     The return value is a list of the (nonempty) subdirectories of dir.
552     If dir is '' then a list of top-level directories is returned.
553
554     If a regexp is specified then the basename of each result must
555     match.  Matching is case-independent.  See pcrepattern(3).
556     """
557     return self._somedir("dirs", dir, re)
558   
559   def files(self, dir, *re):
560     """List files within a directory.
561
562     Arguments:
563     dir -- directory to list, or '' for the whole root.
564     re -- regexp that results must match.  Optional.
565
566     The return value is a list of playable files in dir.  If dir is ''
567     then a list of top-level files is returned.
568
569     If a regexp is specified then the basename of each result must
570     match.  Matching is case-independent.  See pcrepattern(3).
571     """
572     return self._somedir("files", dir, re)
573
574   def allfiles(self, dir, *re):
575     """List subdirectories and files within a directory.
576
577     Arguments:
578     dir -- directory to list, or '' for the whole root.
579     re -- regexp that results must match.  Optional.
580
581     The return value is a list of all (nonempty) subdirectories and
582     files within dir.  If dir is '' then a list of top-level files and
583     directories is returned.
584     
585     If a regexp is specified then the basename of each result must
586     match.  Matching is case-independent.  See pcrepattern(3).
587     """
588     return self._somedir("allfiles", dir, re)
589
590   def set(self, track, key, value):
591     """Set a preference value.
592
593     Arguments:
594     track -- the track to modify
595     key -- the preference name
596     value -- the new preference value
597     """
598     self._simple("set", track, key, value)
599
600   def unset(self, track, key):
601     """Unset a preference value.
602
603     Arguments:
604     track -- the track to modify
605     key -- the preference to remove
606     """
607     self._simple("set", track, key, value)
608
609   def get(self, track, key):
610     """Get a preference value.
611
612     Arguments:
613     track -- the track to query
614     key -- the preference to remove
615
616     The return value is the preference.
617     """
618     ret, details = self._simple("get", track, key)
619     if ret == 555:
620       return None
621     else:
622       return details
623
624   def prefs(self, track):
625     """Get all the preferences for a track.
626
627     Arguments:
628     track -- the track to query
629
630     The return value is a dictionary of all the track's preferences.
631     Note that even nominally numeric values remain encoded as strings.
632     """
633     self._simple("prefs", track)
634     r = {}
635     for line in self._body():
636       try:
637         kv = _split(line)
638       except _splitError, s:
639         raise protocolError(self.who, s.str())
640       if len(kv) != 2:
641         raise protocolError(self.who, "invalid prefs body line")
642       r[kv[0]] = kv[1]
643     return r
644
645   def _boolean(self, s):
646     return s[1] == 'yes'
647
648   def exists(self, track):
649     """Return true if a track exists
650
651     Arguments:
652     track -- the track to check for"""
653     return self._boolean(self._simple("exists", track))
654
655   def enabled(self):
656     """Return true if playing is enabled"""
657     return self._boolean(self._simple("enabled"))
658
659   def random_enabled(self):
660     """Return true if random play is enabled"""
661     return self._boolean(self._simple("random-enabled"))
662
663   def random_enable(self):
664     """Enable random play."""
665     self._simple("random-enable")
666
667   def random_disable(self):
668     """Disable random play."""
669     self._simple("random-disable")
670
671   def length(self, track):
672     """Return the length of a track in seconds.
673
674     Arguments:
675     track -- the track to query.
676     """
677     ret, details = self._simple("length", track)
678     return int(details)
679
680   def search(self, words):
681     """Search for tracks.
682
683     Arguments:
684     words -- the set of words to search for.
685
686     The return value is a list of track path names, all of which contain
687     all of the required words (in their path name, trackname
688     preferences, etc.)
689     """
690     self._simple("search", _quote(words))
691     return self._body()
692
693   def tags(self):
694     """List all tags
695
696     The return value is a list of all tags which apply to at least one
697     track."""
698     self._simple("tags")
699     return self._body()
700
701   def stats(self):
702     """Get server statistics.
703
704     The return value is list of statistics.
705     """
706     self._simple("stats")
707     return self._body()
708
709   def dump(self):
710     """Get all preferences.
711
712     The return value is an encoded dump of the preferences database.
713     """
714     self._simple("dump")
715     return self._body()
716
717   def set_volume(self, left, right):
718     """Set volume.
719
720     Arguments:
721     left -- volume for the left speaker.
722     right --  volume for the right speaker.
723     """
724     self._simple("volume", left, right)
725
726   def get_volume(self):
727     """Get volume.
728
729     The return value a tuple consisting of the left and right volumes.
730     """
731     ret, details = self._simple("volume")
732     return map(int,string.split(details))
733
734   def move(self, track, delta):
735     """Move a track in the queue.
736
737     Arguments:
738     track -- the name or ID of the track to move
739     delta -- the number of steps towards the head of the queue to move
740     """
741     ret, details = self._simple("move", track, str(delta))
742     return int(details)
743
744   def moveafter(self, target, tracks):
745     """Move a track in the queue
746
747     Arguments:
748     target -- target ID or None
749     tracks -- a list of IDs to move
750
751     If target is '' or is not in the queue then the tracks are moved to
752     the head of the queue.
753
754     Otherwise the tracks are moved to just after the target."""
755     if target is None:
756       target = ''
757     self._simple("moveafter", target, *tracks)
758
759   def log(self, callback):
760     """Read event log entries as they happen.
761
762     Each event log entry is handled by passing it to callback.
763
764     The callback takes two arguments, the first is the client and the
765     second the line from the event log.
766     
767     The callback should return True to continue or False to stop (don't
768     forget this, or your program will mysteriously misbehave).
769
770     It is suggested that you use the disorder.monitor class instead of
771     calling this method directly, but this is not mandatory.
772
773     See disorder_protocol(5) for the event log syntax.
774
775     Arguments:
776     callback -- function to call with log entry
777     """
778     ret, details = self._simple("log")
779     while True:
780       l = self._line()
781       self._debug(client.debug_body, "<<< %s" % l)
782       if l != '' and l[0] == '.':
783         if l == '.':
784           return
785         l = l[1:]
786       if not callback(self, l):
787         break
788     # tell the server to stop sending, eat the remains of the body,
789     # eat the response
790     self._send("version")
791     self._body()
792     self._response()
793
794   def pause(self):
795     """Pause the current track."""
796     self._simple("pause")
797
798   def resume(self):
799     """Resume after a pause."""
800     self._simple("resume")
801
802   def part(self, track, context, part):
803     """Get a track name part
804
805     Arguments:
806     track -- the track to query
807     context -- the context ('sort' or 'display')
808     part -- the desired part (usually 'artist', 'album' or 'title')
809
810     The return value is the preference 
811     """
812     ret, details = self._simple("part", track, context, part)
813     return details
814
815   def setglobal(self, key, value):
816     """Set a global preference value.
817
818     Arguments:
819     key -- the preference name
820     value -- the new preference value
821     """
822     self._simple("set-global", key, value)
823
824   def unsetglobal(self, key):
825     """Unset a global preference value.
826
827     Arguments:
828     key -- the preference to remove
829     """
830     self._simple("set-global", key, value)
831
832   def getglobal(self, key):
833     """Get a global preference value.
834
835     Arguments:
836     key -- the preference to look up
837
838     The return value is the preference 
839     """
840     ret, details = self._simple("get-global", key)
841     if ret == 555:
842       return None
843     else:
844       return details
845
846   def make_cookie(self):
847     """Create a login cookie"""
848     ret, details = self._simple("make-cookie")
849     return _split(details)[0]
850   
851   def revoke(self):
852     """Revoke a login cookie"""
853     self._simple("revoke")
854
855   def adduser(self, user, password):
856     """Create a user"""
857     self._simple("adduser", user, password)
858
859   def deluser(self, user):
860     """Delete a user"""
861     self._simple("deluser", user)
862
863   def userinfo(self, user, key):
864     """Get user information"""
865     res, details = self._simple("userinfo", user, key)
866     if res == 555:
867       return None
868     return _split(details)[0]
869
870   def edituser(self, user, key, value):
871     """Set user information"""
872     self._simple("edituser", user, key, value)
873
874   def users(self):
875     """List all users
876
877     The return value is a list of all users."""
878     self._simple("users")
879     return self._body()
880
881   ########################################################################
882   # I/O infrastructure
883
884   def _line(self):
885     # read one response line and return as some suitable string object
886     #
887     # If an I/O error occurs, disconnect from the server.
888     #
889     # XXX does readline() DTRT regarding character encodings?
890     try:
891       l = self.r.readline()
892       if not re.search("\n", l):
893         raise communicationError(self.who, "peer disconnected")
894       l = l[:-1]
895     except:
896       self._disconnect()
897       raise
898     return unicode(l, "UTF-8")
899
900   def _response(self):
901     # read a response as a (code, details) tuple
902     l = self._line()
903     self._debug(client.debug_proto, "<== %s" % l)
904     m = _response.match(l)
905     if m:
906       return int(m.group(1)), m.group(2)
907     else:
908       raise protocolError(self.who, "invalid response %s")
909
910   def _send(self, *command):
911     # Quote and send a command
912     #
913     # Returns the encoded command.
914     quoted = _quote(command)
915     self._debug(client.debug_proto, "==> %s" % quoted)
916     encoded = quoted.encode("UTF-8")
917     try:
918       self.w.write(encoded)
919       self.w.write("\n")
920       self.w.flush()
921       return encoded
922     except IOError, e:
923       # e.g. EPIPE
924       self._disconnect()
925       raise communicationError(self.who, e)
926     except:
927       self._disconnect()
928       raise
929
930   def _simple(self, *command):
931     # Issue a simple command, throw an exception on error
932     #
933     # If an I/O error occurs, disconnect from the server.
934     #
935     # On success or 'normal' errors returns response as a (code, details) tuple
936     #
937     # On error raise operationError
938     if self.state == 'disconnected':
939       self.connect()
940     if command:
941       cmd = self._send(*command)
942     else:
943       cmd = None
944     res, details = self._response()
945     if res / 100 == 2 or res == 555:
946       return res, details
947     raise operationError(res, details, cmd)
948
949   def _body(self):
950     # Fetch a dot-stuffed body
951     result = []
952     while True:
953       l = self._line()
954       self._debug(client.debug_body, "<<< %s" % l)
955       if l != '' and l[0] == '.':
956         if l == '.':
957           return result
958         l = l[1:]
959       result.append(l)
960
961   ########################################################################
962   # Configuration file parsing
963
964   def _readfile(self, path):
965     # Read a configuration file
966     #
967     # Arguments:
968     #
969     # path -- path of file to read
970
971     # handlers for various commands
972     def _collection(self, command, args):
973       if len(args) != 3:
974         return "'%s' takes three args" % command
975       self.config["collections"].append(args)
976       
977     def _unary(self, command, args):
978       if len(args) != 1:
979         return "'%s' takes only one arg" % command
980       self.config[command] = args[0]
981
982     def _include(self, command, args):
983       if len(args) != 1:
984         return "'%s' takes only one arg" % command
985       self._readfile(args[0])
986
987     def _any(self, command, args):
988       self.config[command] = args
989
990     # mapping of options to handlers
991     _options = { "collection": _collection,
992                  "username": _unary,
993                  "password": _unary,
994                  "home": _unary,
995                  "connect": _any,
996                  "include": _include }
997
998     # the parser
999     for lno, line in enumerate(file(path, "r")):
1000       try:
1001         fields = _split(line, 'comments')
1002       except _splitError, s:
1003         raise parseError(path, lno + 1, str(s))
1004       if fields:
1005         command = fields[0]
1006         # we just ignore options we don't know about, so as to cope gracefully
1007         # with version skew (and nothing to do with implementor laziness)
1008         if command in _options:
1009           e = _options[command](self, command, fields[1:])
1010           if e:
1011             self._parseError(path, lno + 1, e)
1012
1013   def _parseError(self, path, lno, s):
1014     raise parseError(path, lno, s)
1015
1016 ########################################################################
1017 # monitor class
1018
1019 class monitor:
1020   """DisOrder event log monitor class
1021
1022   Intended to be subclassed with methods corresponding to event log messages
1023   the implementor cares about over-ridden."""
1024
1025   def __init__(self, c=None):
1026     """Constructor for the monitor class
1027
1028     Can be passed a client to use.  If none is specified then one
1029     will be created specially for the purpose.
1030
1031     Arguments:
1032     c -- client"""
1033     if c == None:
1034       c = client();
1035     self.c = c
1036
1037   def run(self):
1038     """Start monitoring logs.  Continues monitoring until one of the
1039     message-specific methods returns False.  Can be called more than once
1040     (but not recursively!)"""
1041     self.c.log(self._callback)
1042
1043   def when(self):
1044     """Return the timestamp of the current (or most recent) event log entry"""
1045     return self.timestamp
1046
1047   def _callback(self, c, line):
1048     try:
1049       bits = _split(line)
1050     except:
1051       return self.invalid(line)
1052     if(len(bits) < 2):
1053       return self.invalid(line)
1054     self.timestamp = int(bits[0], 16)
1055     keyword = bits[1]
1056     bits = bits[2:]
1057     if keyword == 'completed':
1058       if len(bits) == 1:
1059         return self.completed(bits[0])
1060     elif keyword == 'failed':
1061       if len(bits) == 2:
1062         return self.failed(bits[0], bits[1])
1063     elif keyword == 'moved':
1064       if len(bits) == 3:
1065         try:
1066           n = int(bits[1])
1067         except:
1068           return self.invalid(line)
1069         return self.moved(bits[0], n, bits[2])
1070     elif keyword == 'playing':
1071       if len(bits) == 1:
1072         return self.playing(bits[0], None)
1073       elif len(bits) == 2:
1074         return self.playing(bits[0], bits[1])
1075     elif keyword == 'queue' or keyword == 'recent-added':
1076       try:
1077         q = _list2dict(bits)
1078       except:
1079         return self.invalid(line)
1080       if keyword == 'queue':
1081         return self.queue(q)
1082       if keyword == 'recent-added':
1083         return self.recent_added(q)
1084     elif keyword == 'recent-removed':
1085       if len(bits) == 1:
1086         return self.recent_removed(bits[0])
1087     elif keyword == 'removed':
1088       if len(bits) == 1:
1089         return self.removed(bits[0], None)
1090       elif len(bits) == 2:
1091         return self.removed(bits[0], bits[1])
1092     elif keyword == 'scratched':
1093       if len(bits) == 2:
1094         return self.scratched(bits[0], bits[1])
1095     return self.invalid(line)
1096
1097   def completed(self, track):
1098     """Called when a track completes.
1099
1100     Arguments:
1101     track -- track that completed"""
1102     return True
1103
1104   def failed(self, track, error):
1105     """Called when a player suffers an error.
1106
1107     Arguments:
1108     track -- track that failed
1109     error -- error indicator"""
1110     return True
1111
1112   def moved(self, id, offset, user):
1113     """Called when a track is moved in the queue.
1114
1115     Arguments:
1116     id -- queue entry ID
1117     offset -- distance moved
1118     user -- user responsible"""
1119     return True
1120
1121   def playing(self, track, user):
1122     """Called when a track starts playing.
1123
1124     Arguments:
1125     track -- track that has started
1126     user -- user that submitted track, or None"""
1127     return True
1128
1129   def queue(self, q):
1130     """Called when a track is added to the queue.
1131
1132     Arguments:
1133     q -- dictionary of new queue entry"""
1134     return True
1135
1136   def recent_added(self, q):
1137     """Called when a track is added to the recently played list
1138
1139     Arguments:
1140     q -- dictionary of new queue entry"""
1141     return True
1142
1143   def recent_removed(self, id):
1144     """Called when a track is removed from the recently played list
1145
1146     Arguments:
1147     id -- ID of removed entry (always the oldest)"""
1148     return True
1149
1150   def removed(self, id, user):
1151     """Called when a track is removed from the queue, either manually
1152     or in order to play it.
1153
1154     Arguments:
1155     id -- ID of removed entry
1156     user -- user responsible (or None if we're playing this track)"""
1157     return True
1158
1159   def scratched(self, track, user):
1160     """Called when a track is scratched
1161
1162     Arguments:
1163     track -- track that was scratched
1164     user -- user responsible"""
1165     return True
1166
1167   def invalid(self, line):
1168     """Called when an event log line cannot be interpreted
1169
1170     Arguments:
1171     line -- line that could not be understood"""
1172     return True
1173
1174 # Local Variables:
1175 # mode:python
1176 # py-indent-offset:2
1177 # comment-column:40
1178 # fill-column:72
1179 # End: