132 lines
4.7 KiB
Python
132 lines
4.7 KiB
Python
from os import walk, remove
|
|
from os.path import sep, join, exists, normpath
|
|
from re import findall, sub
|
|
from distutils.core import setup
|
|
from distutils.command.install import install
|
|
from pprint import pprint
|
|
from fileinput import FileInput
|
|
from re import sub, match
|
|
|
|
from Configuration import *
|
|
|
|
class Setup:
|
|
|
|
config = Configuration()
|
|
manifest_path = "MANIFEST"
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
def remove_old_mainfest(self):
|
|
path = self.manifest_path
|
|
if exists(path):
|
|
remove(path)
|
|
|
|
def build_package_list(self):
|
|
packages = []
|
|
config = self.config.get_section("setup")
|
|
locations = [config["package-root"]] + config["additional-packages"]
|
|
for location in locations:
|
|
if exists(location):
|
|
for root, dirs, files in walk(location, followlinks=True):
|
|
packages.append(root.replace(sep, "."))
|
|
return packages
|
|
|
|
def build_data_map(self):
|
|
include = []
|
|
config = self.config.get_section("setup")
|
|
exclude = map(normpath, config["data-exclude"])
|
|
for root, dirs, files in walk("."):
|
|
dirs = self.remove_excluded(dirs, root, exclude)
|
|
files = [join(root, f) for f in self.remove_excluded(files, root,
|
|
exclude)]
|
|
if files:
|
|
include.append((normpath(join(config["installation-path"],
|
|
root)), files))
|
|
return include
|
|
|
|
def remove_excluded(self, paths, root, exclude):
|
|
removal = []
|
|
for path in paths:
|
|
if normpath(join(root, path)) in exclude:
|
|
removal.append(path)
|
|
for path in removal:
|
|
paths.remove(path)
|
|
return paths
|
|
|
|
def translate_title(self):
|
|
config = self.config.get_section("setup")
|
|
title = config["title"].replace(" ", config["whitespace-placeholder"])
|
|
return sub("[^\w-]", config["special-char-placeholder"], title)
|
|
|
|
def build_description(self):
|
|
description = ""
|
|
path = self.config.get("setup", "description-file")
|
|
if exists(path):
|
|
description = "\n%s\n%s\n%s" % (file(path).read(),
|
|
"Changelog\n=========",
|
|
self.translate_changelog())
|
|
return description
|
|
|
|
def translate_changelog(self):
|
|
translation = ""
|
|
path = self.config.get("setup", "changelog")
|
|
if exists(path):
|
|
lines = file(path).readlines()
|
|
package_name = lines[0].split()[0]
|
|
for line in lines:
|
|
line = line.strip()
|
|
if line.startswith(package_name):
|
|
version = findall("\((.*)\)", line)[0]
|
|
translation += "\n%s\n%s\n" % (version, "-" * len(version))
|
|
elif line and not line.startswith("--"):
|
|
if line.startswith("*"):
|
|
translation += line + "\n"
|
|
else:
|
|
translation += " " + line + "\n"
|
|
return translation
|
|
|
|
def setup(self, windows=[], options={}):
|
|
print "running setup..."
|
|
self.remove_old_mainfest()
|
|
config = self.config.get_section("setup")
|
|
scripts = []
|
|
if config["init-script"]:
|
|
scripts.append(config["init-script"])
|
|
setup(cmdclass={"install": insert_resources_path},
|
|
name=self.translate_title(),
|
|
packages=self.build_package_list(),
|
|
scripts=scripts,
|
|
data_files=self.build_data_map(),
|
|
requires=config["requirements"],
|
|
version=config["version"],
|
|
description=config["summary"],
|
|
classifiers=config["classifiers"],
|
|
long_description=self.build_description(),
|
|
license=config["license"],
|
|
platforms=config["platforms"],
|
|
author=config["contact-name"],
|
|
author_email=config["contact-email"],
|
|
url=config["url"],
|
|
windows=windows,
|
|
options=options)
|
|
|
|
|
|
class insert_resources_path(install):
|
|
|
|
def run(self):
|
|
install.run(self)
|
|
self.edit_game_object_file()
|
|
|
|
def edit_game_object_file(self):
|
|
config = Configuration().get_section("setup")
|
|
for path in self.get_outputs():
|
|
if path.endswith(config["main-object"]):
|
|
for line in FileInput(path, inplace=True):
|
|
pattern = "^ *{0} *=.*".\
|
|
format(config["resources-path-identifier"])
|
|
if match(pattern, line):
|
|
line = sub("=.*$", "= \"{0}\"".\
|
|
format(config["installation-path"]), line)
|
|
print line.strip("\n")
|