mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
127 lines
4.6 KiB
Python
127 lines
4.6 KiB
Python
4 years ago
|
# Copyright 2013 Hewlett-Packard Development Company, L.P.
|
||
|
|
# All Rights Reserved.
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
||
|
|
# not use this file except in compliance with the License. You may obtain
|
||
|
|
# a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||
|
|
# License for the specific language governing permissions and limitations
|
||
|
|
# under the License.
|
||
|
|
|
||
|
|
import os
|
||
|
|
import shlex
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from pbr import find_package
|
||
|
|
from pbr.hooks import base
|
||
|
|
|
||
|
|
|
||
|
|
def get_manpath():
|
||
|
|
manpath = 'share/man'
|
||
|
|
if os.path.exists(os.path.join(sys.prefix, 'man')):
|
||
|
|
# This works around a bug with install where it expects every node
|
||
|
|
# in the relative data directory to be an actual directory, since at
|
||
|
|
# least Debian derivatives (and probably other platforms as well)
|
||
|
|
# like to symlink Unixish /usr/local/man to /usr/local/share/man.
|
||
|
|
manpath = 'man'
|
||
|
|
return manpath
|
||
|
|
|
||
|
|
|
||
|
|
def get_man_section(section):
|
||
|
|
return os.path.join(get_manpath(), 'man%s' % section)
|
||
|
|
|
||
|
|
|
||
|
|
def unquote_path(path):
|
||
|
|
# unquote the full path, e.g: "'a/full/path'" becomes "a/full/path", also
|
||
|
|
# strip the quotes off individual path components because os.walk cannot
|
||
|
|
# handle paths like: "'i like spaces'/'another dir'", so we will pass it
|
||
|
|
# "i like spaces/another dir" instead.
|
||
|
|
|
||
|
|
if os.name == 'nt':
|
||
|
|
# shlex cannot handle paths that contain backslashes, treating those
|
||
|
|
# as escape characters.
|
||
|
|
path = path.replace("\\", "/")
|
||
|
|
return "".join(shlex.split(path)).replace("/", "\\")
|
||
|
|
|
||
|
|
return "".join(shlex.split(path))
|
||
|
|
|
||
|
|
|
||
|
|
class FilesConfig(base.BaseConfig):
|
||
|
|
|
||
|
|
section = 'files'
|
||
|
|
|
||
|
|
def __init__(self, config, name):
|
||
|
|
super(FilesConfig, self).__init__(config)
|
||
|
|
self.name = name
|
||
|
|
self.data_files = self.config.get('data_files', '')
|
||
|
|
|
||
|
|
def save(self):
|
||
|
|
self.config['data_files'] = self.data_files
|
||
|
|
super(FilesConfig, self).save()
|
||
|
|
|
||
|
|
def expand_globs(self):
|
||
|
|
finished = []
|
||
|
|
for line in self.data_files.split("\n"):
|
||
|
|
if line.rstrip().endswith('*') and '=' in line:
|
||
|
|
(target, source_glob) = line.split('=')
|
||
|
|
source_prefix = source_glob.strip()[:-1]
|
||
|
|
target = target.strip()
|
||
|
|
if not target.endswith(os.path.sep):
|
||
|
|
target += os.path.sep
|
||
|
|
unquoted_prefix = unquote_path(source_prefix)
|
||
|
|
unquoted_target = unquote_path(target)
|
||
|
|
for (dirpath, dirnames, fnames) in os.walk(unquoted_prefix):
|
||
|
|
# As source_prefix is always matched, using replace with a
|
||
|
|
# a limit of one is always going to replace the path prefix
|
||
|
|
# and not accidentally replace some text in the middle of
|
||
|
|
# the path
|
||
|
|
new_prefix = dirpath.replace(unquoted_prefix,
|
||
|
|
unquoted_target, 1)
|
||
|
|
finished.append("'%s' = " % new_prefix)
|
||
|
|
finished.extend(
|
||
|
|
[" '%s'" % os.path.join(dirpath, f) for f in fnames])
|
||
|
|
else:
|
||
|
|
finished.append(line)
|
||
|
|
|
||
|
|
self.data_files = "\n".join(finished)
|
||
|
|
|
||
|
|
def add_man_path(self, man_path):
|
||
|
|
self.data_files = "%s\n'%s' =" % (self.data_files, man_path)
|
||
|
|
|
||
|
|
def add_man_page(self, man_page):
|
||
|
|
self.data_files = "%s\n '%s'" % (self.data_files, man_page)
|
||
|
|
|
||
|
|
def get_man_sections(self):
|
||
|
|
man_sections = dict()
|
||
|
|
manpages = self.pbr_config['manpages']
|
||
|
|
for manpage in manpages.split():
|
||
|
|
section_number = manpage.strip()[-1]
|
||
|
|
section = man_sections.get(section_number, list())
|
||
|
|
section.append(manpage.strip())
|
||
|
|
man_sections[section_number] = section
|
||
|
|
return man_sections
|
||
|
|
|
||
|
|
def hook(self):
|
||
|
|
packages = self.config.get('packages', self.name).strip()
|
||
|
|
expanded = []
|
||
|
|
for pkg in packages.split("\n"):
|
||
|
|
if os.path.isdir(pkg.strip()):
|
||
|
|
expanded.append(find_package.smart_find_packages(pkg.strip()))
|
||
|
|
|
||
|
|
self.config['packages'] = "\n".join(expanded)
|
||
|
|
|
||
|
|
self.expand_globs()
|
||
|
|
|
||
|
|
if 'manpages' in self.pbr_config:
|
||
|
|
man_sections = self.get_man_sections()
|
||
|
|
for (section, pages) in man_sections.items():
|
||
|
|
manpath = get_man_section(section)
|
||
|
|
self.add_man_path(manpath)
|
||
|
|
for page in pages:
|
||
|
|
self.add_man_page(page)
|