Viewing file:  tools.py (3.8 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
# -*- coding: UTF-8 -*-
# Copyright © 2010 Piotr Ożarowski <piotr@debian.org>
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
from __future__ import with_statement
import logging
import re
from cPickle import dumps
from os import symlink
from debpython.version import getver
log = logging.getLogger(__name__)
EGGnPTH_RE = re.compile(r'(.*?)(-py\d\.\d(?:-[^.]*)?)?(\.egg-info|\.pth)$')
SHEBANG_RE = re.compile(r'^#!\s*/usr/bin/(?:env\s+)?(python(\d+\.\d+)?(?:-dbg)?).*')
def sitedir(version, package=None, gdb=False):
"""Return path to site-packages directory.
>>> sitedir((2, 5))
'/usr/lib/python2.5/site-packages/'
>>> sitedir((2, 7), 'python-foo', True)
'debian/python-foo/usr/lib/debug/usr/lib/python2.7/dist-packages/'
"""
if isinstance(version, basestring):
version = tuple(int(i) for i in version.split('.'))
if version >= (2, 6):
path = "/usr/lib/python%d.%d/dist-packages/" % version
else:
path = "/usr/lib/python%d.%d/site-packages/" % version
if gdb:
path = "/usr/lib/debug%s" % path
if package:
path = "debian/%s%s" % (package, path)
return path
def relpath(target, link):
"""Return relative path.
>>> relpath('/usr/share/python-foo/foo.py', '/usr/bin/foo', )
'../share/python-foo/foo.py'
"""
t = target.split('/')
l = link.split('/')
while l[0] == t[0]:
del l[0], t[0]
return '/'.join(['..'] * (len(l) - 1) + t)
def relative_symlink(target, link):
"""Create relative symlink."""
return symlink(relpath(target, link), link)
def shebang2pyver(fname):
"""Check file's shebang.
:rtype: tuple
:returns: pair of Python interpreter string and Python version
"""
try:
with open(fname) as fp:
data = fp.read(32)
match = SHEBANG_RE.match(data)
if not match:
return None
res = match.groups()
if res != (None, None):
if res[1]:
res = res[0], getver(res[1])
return res
except IOError:
log.error('cannot open %s', fname)
def clean_egg_name(name):
"""Remove Python version and platform name from Egg files/dirs.
>>> clean_egg_name('python_pipeline-0.1.3_py3k-py3.1.egg-info')
'python_pipeline-0.1.3_py3k.egg-info'
>>> clean_egg_name('Foo-1.2-py2.7-linux-x86_64.egg-info')
'Foo-1.2.egg-info'
"""
match = EGGnPTH_RE.match(name)
if match and match.group(2) is not None:
return ''.join(match.group(1, 3))
return name
class memoize(object):
def __init__(self, func):
self.func = func
self.cache = {}
def __call__(self, *args, **kwargs):
key = dumps((args, kwargs))
if key not in self.cache:
self.cache[key] = self.func(*args, **kwargs)
return self.cache[key]
|