Как расширить distutils с помощью простого скрипта после установки?

Question

Мне нужно запустить простой скрипт после установки модулей и программ. У меня небольшие проблемы с поиском прямой документации о том, как это сделать. Похоже, мне нужно наследовать от distutils.command.install, переопределить некоторые методы и добавить этот объект в сценарий установки. Специфика немного туманная, и кажется, что для такой простой зацепки требуются большие усилия. Кто-нибудь знает простой способ сделать это?

Answer 1

Я копался в источнике distutils в течение дня, чтобы узнать достаточно об этом, чтобы создать кучу пользовательских команд. Это не красиво, но это работает.

import distutils.core
from distutils.command.install import install
...
class my_install(install):
    def run(self):
        install.run(self)
        # Custom stuff here
        # distutils.command.install actually has some nice helper methods
        # and interfaces. I strongly suggest reading the docstrings.
...
distutils.core.setup(..., cmdclass=dict(install=my_install), ...)

Answer 2

Хорошо, я понял это. Идея состоит в том, чтобы расширить одну из команд distutils и переписать метод run. Чтобы указать distutils использовать новый класс, вы можете использовать переменную cmdclass.

from distutils.core import setup
from distutils.command.install_data import install_data

class post_install(install_data):
    def run(self):
        # Call parent 
        install_data.run(self)
        # Execute commands
        print "Running"

setup(name="example",
      cmdclass={"install_data": post_install},
      ...
      )

Надеюсь, это поможет кому-то еще.

Answer 3

Я не мог заставить работать ответ Джо Вершнига и подправил его ответ аналогично расширенной документации distutils. Я придумал этот код, который отлично работает на моей машине.

from distutils import core
from distutils.command.install import install
...
class my_install(install):
    def run(self):
        install.run(self)
        # Custom stuff here
        # distutils.command.install actually has some nice helper methods
        # and interfaces. I strongly suggest reading the docstrings.
...
distutils.core.setup(..., cmdclass={'install': my_install})

Примечание: я не редактировал ответ Джо, так как не уверен, почему его ответ не работал на моей машине.

Answer 4

Я получил ошибку, когда попытался принять принятый здесь ответ (возможно, потому, что я использую Python 2.6 в данном конкретном случае, не уверен). Это произошло как для 'setup.py install', так и 'pip install':

sudo python setup.py install

завершается с ошибкой: ошибка в setup.cfg: команда 'my_install' не имеет такой опции 'single_version_externally_managed'

А ТАКЖЕ

sudo pip install . -U

более многословно, но также с ошибкой: опция --single-version-внешне управляемая не распознана

Вариация принятого ответа

Замена импорта из distutils с помощью setuptools решила проблему для меня:

from setuptools import setup
from setuptools.command.install import install