cola/compat.py

   1 import os
   2 try:
   3     set = set
   4 except NameError:
   5     from sets import Set as set
   6     set = set
   7
   8 try:
   9     import hashlib
  10 except ImportError:
  11     import md5
  12     class hashlib(object):
  13         @staticmethod
  14         def new(*args):
  15             return md5.new()
  16
  17         @classmethod
  18         def md5(cls, value=''):
  19             obj = md5.new()
  20             obj.update(value)
  21             return obj
  22
  23 def setenv(key, value):
  24     """Compatibility wrapper for setting environment variables
  25
  26     Why?  win32 requires putenv().  UNIX only requires os.environ.
  27
  28     """
  29     os.environ[key] = value
  30     os.putenv(key, value)
  31
  32
  33 def unsetenv(key):
  34     """Compatibility wrapper for unsetting environment variables"""
  35     try:
  36         del os.environment[key]
  37     except:
  38         pass
  39     if hasattr(os, 'unsetenv'):
  40         os.unsetenv(key)