← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

  • Committer: Robert Collins
  • Date: 2005-10-30 00:00:09 UTC
  • mfrom: (1185.16.134)
  • Revision ID: robertc@robertcollins.net-20051030000009-9db99a338a0dfdac
Merge from Martin.

Show diffs side-by-side

added added

removed removed

Lines of Context:
setup.py
5
5
# './setup.py --help' for more options
6
6
 
7
7
from distutils.core import setup
8
 
 
9
 
# more sophisticated setup script, based on pychecker setup.py
10
 
import sys, os
11
 
from distutils.command.build_scripts import build_scripts
 
8
from distutils.command.install_scripts import install_scripts
12
9
 
13
10
 
14
11
###############################
15
12
# Overridden distutils actions
16
13
###############################
17
14
 
18
 
class my_build_scripts(build_scripts):
19
 
    """Customized build_scripts distutils action.
20
 
 
 
15
class my_install_scripts(install_scripts):
 
16
    """ Customized install_scripts distutils action.
21
17
    Create bzr.bat for win32.
22
18
    """
23
 
 
24
19
    def run(self):
 
20
        import os
 
21
        import sys
 
22
 
 
23
        install_scripts.run(self)   # standard action
 
24
 
25
25
        if sys.platform == "win32":
26
 
            bat_path = os.path.join(self.build_dir, "bzr.bat")
27
 
            self.scripts.append(bat_path)
28
 
            self.mkpath(self.build_dir)
29
 
            scripts_dir = self.distribution.get_command_obj("install").\
30
 
                                                            install_scripts
31
 
            self.execute(func=self._create_bat,
32
 
                         args=[bat_path, scripts_dir],
33
 
                         msg="Create %s" % bat_path)
34
 
        build_scripts.run(self) # invoke "standard" action
35
 
 
36
 
    def _create_bat(self, bat_path, scripts_dir):
37
 
        """ Creates the batch file for bzr on win32.
38
 
        """
39
 
        try:
40
 
            script_path = os.path.join(scripts_dir, "bzr")
41
 
            bat_str = "@%s %s %%*\n" % (sys.executable, script_path)
42
 
            file(bat_path, "w").write(bat_str)
43
 
            print "file written"
44
 
        except Exception, e:
45
 
            print "ERROR: Unable to create %s: %s" % (bat_path, e)
46
 
            raise e
 
26
            try:
 
27
                scripts_dir = self.install_dir
 
28
                script_path = os.path.join(scripts_dir, "bzr")
 
29
                batch_str = "@%s %s %%*\n" % (sys.executable, script_path)
 
30
                batch_path = script_path + ".bat"
 
31
                f = file(batch_path, "w")
 
32
                f.write(batch_str)
 
33
                f.close()
 
34
                print "Created:", batch_path
 
35
            except Exception, e:
 
36
                print "ERROR: Unable to create %s: %s" % (batch_path, e)
47
37
 
48
38
 
49
39
########################
67
57
                'bzrlib.util.effbot.org',
68
58
                'bzrlib.util.configobj',
69
59
                ],
70
 
      scripts=['bzr'])
 
60
      scripts=['bzr'],
 
61
      cmdclass={'install_scripts': my_install_scripts},
 
62
     )