Опубликовать скрипт установки в setuptools с правами sudo - PullRequest
0 голосов
/ 07 мая 2019

Я пытаюсь создать файл setup.py для установки скрипта Python в качестве службы.Пока что я могу установить скрипт и вручную добавить модуль в systemd и включить / запустить сервис, но я бы хотел сделать это автоматически.У меня проблема в том, что мой bash-скрипт, который создает файл модуля, требует привилегий sudo, и я не хочу запускать pip install с sudo.

import os
import subprocess
from setuptools import setup, Extension
import setuptools.command.install
from distutils.command.build import build as build_orig


class install(setuptools.command.install.install):
    def run(self):
        setuptools.command.install.install.run(self)
        current_dir_path = os.path.dirname(os.path.realpath(__file__))
        create_service_script_path = os.path.join(current_dir_path, 'bin', 'create_service.sh')
        subprocess.check_output([create_service_script_path])


setup(name='dspt',
      version=dspt.__version__,
      packages=['dspt'],
      include_package_data=True,
      package_data={"": ["data/template.xml"]},
      ext_modules=[ext],
      install_requires=['slackclient==1.3.1',
                        'numpy',
                        'numba',
                        'scipy',
                        'pytz',
                        'requests',
                        'pyjwt',
                        'pysmb',
                        'boto3'],
      cmdclass={
          'install': install
      },
      entry_points={'console_scripts': ['run_dspt = dspt.batch_processing:main']}
)

create_service.sh

#!/bin/bash
SYSTEMD_SCRIPT_DIR=$( cd  $(dirname "${BASH_SOURCE:=$0}") && pwd)
cp -f "$SYSTEMD_SCRIPT_DIR/dspt.service" /etc/systemd/system
chown root:root /etc/systemd/system/dspt.service

systemctl daemon-reload
systemctl enable dspt.service

файл модуля

[Unit]
Description=My super service
After=multi-user.target network.target

[Service]
User=myuser
Group=mygroup
EnvironmentFile=/etc/environment
Type=idle
Restart=always
RestartSec=3
ExecStart=/home/myuser/miniconda3/envs/dspt/bin/run_dspt

[Install]
WantedBy=multi-user.target

Обратите внимание, что на данный момент файл модуля предполагает, что пакет установлен в виртуальной среде conda с именем dspt.

Скрипт bash create_service.sh obvioulsy требует привилегий sudo, и я не могу понять, как это сделать.

...