vcs.maemo.org Git - doneit/blob - src/toolbox.py

   1 import sys
   2 import StringIO
   3 import urllib
   4 from xml.dom import minidom
   5 import datetime
   6
   7
   8 class Optional(object):
   9         """
  10         Taglines:
  11         Even you don't have to worry about knowing when to perform None checks
  12         When the NULL object pattern just isn't good enough
  13
  14         >>> a = Optional()
  15         >>> a.is_good()
  16         False
  17         >>> a.get_nothrow("Blacksheep")
  18         'Blacksheep'
  19         >>> a.set("Lamb")
  20         >>> a.is_good()
  21         True
  22         >>> a.get_nothrow("Blacksheep"), a.get(), a()
  23         ('Lamb', 'Lamb', 'Lamb')
  24         >>> a.clear()
  25         >>> a.is_good()
  26         False
  27         >>> a.get_nothrow("Blacksheep")
  28         'Blacksheep'
  29         >>>
  30         >>> b = Optional("Lamb")
  31         >>> a.set("Lamb")
  32         >>> a.is_good()
  33         True
  34         >>> a.get_nothrow("Blacksheep"), a.get(), a()
  35         ('Lamb', 'Lamb', 'Lamb')
  36         >>> a.clear()
  37         >>> a.is_good()
  38         False
  39         >>> a.get_nothrow("Blacksheep")
  40         'Blacksheep'
  41         """
  42
  43         class NonExistent(object):
  44                 pass
  45
  46         def __init__(self, value = NonExistent):
  47                 self._value = value
  48
  49         def set(self, value):
  50                 self._value = value
  51
  52         def clear(self):
  53                 self._value = self.NonExistent
  54
  55         def is_good(self):
  56                 return self._value is not self.NonExistent
  57
  58         def get_nothrow(self, default = None):
  59                 return self._value if self.is_good() else default
  60
  61         def get(self):
  62                 if not self.is_good():
  63                         raise ReferenceError("Optional not initialized")
  64                 return self._value
  65
  66         def __call__(self):
  67                 # Implemented to imitate weakref
  68                 return self.get()
  69
  70         def __str__(self):
  71                 return str(self.get_nothrow(""))
  72
  73         def __repr__(self):
  74                 return "<Optional at %x; to %r>" % (
  75                         id(self), self.get_nothrow("Nothing")
  76                 )
  77
  78
  79 def open_anything(source, alternative=None):
  80         """URI, filename, or string --> stream
  81
  82         This function lets you define parsers that take any input source
  83         (URL, pathname to local or network file, or actual data as a string)
  84         and deal with it in a uniform manner.  Returned object is guaranteed
  85         to have all the basic stdio read methods (read, readline, readlines).
  86         Just .close() the object when you're done with it.
  87         """
  88         if hasattr(source, "read"):
  89                 return source
  90
  91         if source == '-':
  92                 return sys.stdin
  93
  94         # try to open with urllib (if source is http, ftp, or file URL)
  95         try:
  96                 return urllib.urlopen(source)
  97         except (IOError, OSError):
  98                 ##pass
  99                 print "ERROR with URL ("+source+")!\n"
 100
 101         # try to open with native open function (if source is pathname)
 102         try:
 103                 return open(source)
 104         except (IOError, OSError):
 105                 ##pass
 106                 print "ERROR with file!\n"
 107
 108         # treat source as string
 109         if alternative == None:
 110                 print 'LAST RESORT.  String is "'+source+'"\n'
 111                 return StringIO.StringIO(str(source))
 112         else:
 113                 print 'LAST RESORT.  String is "'+alternative+'"\n'
 114                 return StringIO.StringIO(str(alternative))
 115
 116
 117 def load_xml(source, alternative=None):
 118         """load XML input source, return parsed XML document
 119
 120         - a URL of a remote XML file ("http://diveintopython.org/kant.xml")
 121         - a filename of a local XML file ("~/diveintopython/common/py/kant.xml")
 122         - standard input ("-")
 123         - the actual XML document, as a string
 124         """
 125         sock = open_anything(source, alternative)
 126         try:
 127                 xmldoc = minidom.parse(sock).documentElement
 128         except (IOError, OSError):
 129                 print "ERROR with data"
 130                 sock.close()
 131                 sock = open_anything('<response method="getProjects"><project projName="ERROR!"/></response>')
 132                 xmldoc = minidom.parse(sock).documentElement
 133         sock.close()
 134         return xmldoc
 135
 136
 137 def to_fuzzy_date(targetDate, todaysDate = datetime.datetime.today()):
 138         """
 139         Conert a date/time/datetime object to a fuzzy date
 140
 141         @bug Not perfect, but good enough for now
 142         """
 143         delta = targetDate - todaysDate
 144         days = abs(delta.days)
 145         directionBy1 = "Next" if 0 < delta.days else "Last"
 146         directionByN = "Later" if 0 < delta.days else "Earlier"
 147         directionByInf = "from now" if 0 < delta.days else "ago"
 148         if 2*365 < days:
 149                 return "Forever %s" % directionByInf
 150         elif 365 < days:
 151                 return "%s year" % directionBy1
 152         elif 2*30 < days:
 153                 return "%s this year" % directionByN
 154         elif 30 < days:
 155                 return "%s month" % directionBy1
 156         elif 14 < days:
 157                 return "%s this month" % directionByN
 158         elif 7 < days:
 159                 return "%s week" % directionBy1
 160         elif 2 < days:
 161                 return "%s this week" % directionByN
 162         elif 1 < days:
 163                 return "%s day" % directionByN
 164         else:
 165                 return "Today"