18 |
import os |
import os |
19 |
from types import TupleType |
from types import TupleType |
20 |
from distutils.core import setup, Extension, Command |
from distutils.core import setup, Extension, Command |
21 |
from distutils.command.install import install |
from distutils.command.install import install, INSTALL_SCHEMES, subst_vars |
22 |
from distutils.command.build_py import build_py |
from distutils.command.build_py import build_py |
23 |
from distutils.command.bdist_rpm import bdist_rpm |
from distutils.command.bdist_rpm import bdist_rpm |
24 |
from distutils.file_util import write_file |
from distutils.file_util import write_file |
506 |
return modules |
return modules |
507 |
|
|
508 |
|
|
509 |
|
thubaninit_contents = """ |
510 |
|
# This module was automatically generated by Thuban's install script |
511 |
|
'''Import this module once per program (best place is probably the file |
512 |
|
that ends up as your __main__ module) to be able to import Thuban |
513 |
|
afterwards. |
514 |
|
|
515 |
|
Usage: |
516 |
|
|
517 |
|
import thubaninit |
518 |
|
import Thuban |
519 |
|
''' |
520 |
|
import sys, os |
521 |
|
sys.path.insert(0, %(thubandir)s) |
522 |
|
|
523 |
|
# Put the Lib dir into the path. The Lib dir contains some extra Python |
524 |
|
# modules |
525 |
|
import Thuban |
526 |
|
thubandir = os.path.join(Thuban.__path__[0], '..') |
527 |
|
dir = os.path.join(thubandir, "Lib") |
528 |
|
if os.path.isdir(dir): |
529 |
|
sys.path.insert(0, dir) |
530 |
|
""" |
531 |
|
|
532 |
|
|
533 |
class ThubanInstall(install): |
class ThubanInstall(install): |
534 |
|
|
545 |
"(default on posix systems and only relevant there)"), |
"(default on posix systems and only relevant there)"), |
546 |
|
|
547 |
("extra-files", None, |
("extra-files", None, |
548 |
"List of filenames or (src, dest) pairs describing " |
"List of filenames or (src, dest) pairs describing" |
549 |
" extra files to install " |
" extra files to install " |
550 |
"(can only be set from witin setup.py"), |
"(can only be set from witin setup.py"), |
551 |
|
|
552 |
|
("create-init-module", None, |
553 |
|
"Create a module in the site-packages directory that" |
554 |
|
" tweaks sys.path to let you easily import thuban" |
555 |
|
" modules from outside of thuban."), |
556 |
|
("init-module-dir", None, |
557 |
|
"Directory in which to create the init module." |
558 |
|
" Defaults to Python's site-packages directory."), |
559 |
|
|
560 |
]) |
]) |
561 |
|
|
562 |
boolean_options = install.boolean_options[:] |
boolean_options = install.boolean_options[:] |
563 |
boolean_options.append("do-symlink") |
boolean_options.append("do-symlink") |
564 |
|
boolean_options.append("create-init-module") |
565 |
|
|
566 |
def initialize_options(self): |
def initialize_options(self): |
567 |
self.do_symlink = None |
self.do_symlink = None |
568 |
self.extra_files = [] |
self.extra_files = [] |
569 |
|
self.create_init_module = 1 |
570 |
|
self.init_module_dir = None |
571 |
install.initialize_options(self) |
install.initialize_options(self) |
572 |
|
|
573 |
def finalize_options(self): |
def finalize_options(self): |
577 |
else: |
else: |
578 |
self.do_symlink = 0 |
self.do_symlink = 0 |
579 |
install.finalize_options(self) |
install.finalize_options(self) |
580 |
|
self.expand_with_pure_python_dirs(["init_module_dir"]) |
581 |
|
|
582 |
|
def expand_with_pure_python_dirs(self, attrs): |
583 |
|
"""Expand the attributes with default values of base and platbase""" |
584 |
|
# it seems that the values for "prefix" and "exec_prefix" in |
585 |
|
# self.config_vars are the corresponding values used by the |
586 |
|
# python interpreter, so we just assign these to "base" and |
587 |
|
# "platbase". |
588 |
|
config_vars = self.config_vars.copy() |
589 |
|
config_vars["base"] = self.config_vars["prefix"] |
590 |
|
config_vars["platbase"] = self.config_vars["exec_prefix"] |
591 |
|
for attr in attrs: |
592 |
|
val = getattr(self, attr) |
593 |
|
if val is not None: |
594 |
|
if os.name == 'posix': |
595 |
|
val = os.path.expanduser(val) |
596 |
|
val = subst_vars(val, config_vars) |
597 |
|
setattr(self, attr, val) |
598 |
|
|
599 |
|
def select_scheme(self, scheme): |
600 |
|
"""Extend the inherited method to set init_module_dir""" |
601 |
|
install.select_scheme(self, scheme) |
602 |
|
# only set init_module_dir if it wasn't set by the user |
603 |
|
if self.init_module_dir is None: |
604 |
|
self.init_module_dir = INSTALL_SCHEMES[scheme]['purelib'] |
605 |
|
|
606 |
|
def convert_paths(self, *args): |
607 |
|
"""Extend the inherited method so that we can remember some filename |
608 |
|
""" |
609 |
|
# remember the installation directory before its root gets |
610 |
|
# changed |
611 |
|
self.install_lib_orig = self.install_lib |
612 |
|
apply(install.convert_paths, (self,) + args) |
613 |
|
|
614 |
def run(self): |
def run(self): |
615 |
install.run(self) |
install.run(self) |
633 |
self.mkpath(bindir) |
self.mkpath(bindir) |
634 |
self.link_file(scriptfile, binfile) |
self.link_file(scriptfile, binfile) |
635 |
|
|
636 |
|
if self.create_init_module: |
637 |
|
# create the init module |
638 |
|
initfilename = self.thuban_init_filename() |
639 |
|
if self.root: |
640 |
|
initfilename = change_root(self.root, initfilename) |
641 |
|
contents = thubaninit_contents % { |
642 |
|
"thubandir": repr(self.install_lib_orig) |
643 |
|
} |
644 |
|
self.mkpath(os.path.dirname(initfilename)) |
645 |
|
self.execute(write_file, (initfilename, split(contents, "\n")), |
646 |
|
"Create %s" % initfilename) |
647 |
|
|
648 |
def link_file(self, src, dest): |
def link_file(self, src, dest): |
649 |
"""Create a symbolic link dest pointing to src. |
"""Create a symbolic link dest pointing to src. |
650 |
|
|
659 |
if not os.path.exists(dest): |
if not os.path.exists(dest): |
660 |
os.symlink(src, dest) |
os.symlink(src, dest) |
661 |
|
|
662 |
|
def thuban_init_filename(self): |
663 |
|
"""Return the filename for the init-module""" |
664 |
|
# Since we override the normal install dirs to point to our own |
665 |
|
# prefix we have to reach into installed |
666 |
|
return self.init_module_dir + "/thubaninit.py" |
667 |
|
|
668 |
def get_outputs (self): |
def get_outputs (self): |
669 |
outputs = install.get_outputs(self) |
outputs = install.get_outputs(self) |
679 |
bindir = change_root(self.root, bindir) |
bindir = change_root(self.root, bindir) |
680 |
binfile = os.path.join(bindir, "thuban") |
binfile = os.path.join(bindir, "thuban") |
681 |
outputs.append(binfile) |
outputs.append(binfile) |
682 |
|
if self.create_init_module: |
683 |
|
initfilename = self.thuban_init_filename() |
684 |
|
if self.root: |
685 |
|
initfilename = change_root(self.root, initfilename) |
686 |
|
outputs.append(initfilename) |
687 |
return outputs |
return outputs |
688 |
|
|
689 |
|
|