diff --git a/multibootusb b/multibootusb index f5e3ff0..996ac26 100644 --- a/multibootusb +++ b/multibootusb @@ -12,9 +12,13 @@ import sys import os import platform -# The following line is required for distros based on rpm so as to avoid import errors when running from installed system +# The following line is required for distros based on rpm so as to avoid import errors when running from +# installed system sys.path.append('/usr/local/lib/python3.4/dist-packages') -# print(sys.path) + +# Ensure that above issue doesn't occur on debian based distro as well +if '/usr/lib/python3/dist-packages/scripts/' not in sys.path: + sys.path.append('/usr/lib/python3/dist-packages/scripts/') # Had trouble in importing scripts directory. Had to add few lines below to ensure it works on source as well as # post install @@ -57,7 +61,7 @@ Options: -h or --help : Print this help message and exit -c or --command : Invoke command line usage. This option is required; if omitted, the GUI will be launched. - -i or --iso : Path to ISO file()s. If many ISOs are supplied, + -i or --iso : Path to ISO file(s). If many ISOs are supplied, they should be separated by ',' with no spaces in between. -t or --target : Path to target USB device partition (e.g. "/dev/sdb1"). @@ -105,7 +109,6 @@ def start_gui(): if __name__ == '__main__': - running_from() if platform.system() == 'Windows': if not admin.isUserAdmin(): admin.runAsAdmin() @@ -165,7 +168,9 @@ if gui is False: log('\nOptions \'-i\' and \'-t\' must be supplied together. See the usage below.') usage() else: + running_from() cli_install_distro() elif gui is True: + running_from() start_gui() diff --git a/org.debian.pkexec.run-multibootusb.policy b/org.debian.pkexec.run-multibootusb.policy index d8e8566..e111748 100644 --- a/org.debian.pkexec.run-multibootusb.policy +++ b/org.debian.pkexec.run-multibootusb.policy @@ -12,7 +12,7 @@ auth_admin auth_admin - /usr/local/bin/multibootusb + /usr/bin/multibootusb true diff --git a/setup.py b/setup.py index 6adf520..41d9aa6 100644 --- a/setup.py +++ b/setup.py @@ -20,6 +20,15 @@ def get_data(_dir): data.append(cfg_file) return data + +def root_files(_dir): + data = [] + for _file in os.listdir(_dir): + path = os.path.join(_dir, _file) + if not os.path.isdir(path): + data.append(path) + return data + Version = mbusb_version() print(Version) setup( @@ -55,6 +64,7 @@ setup( ('/usr/share/multibootusb/data/multibootusb', ["data/multibootusb/menu.lst"]), ('/usr/share/multibootusb/data/multibootusb', ["data/multibootusb/syslinux.cfg"]), ('/usr/share/multibootusb/data/multibootusb', ["data/multibootusb/vesamenu.c32"]), - ('/usr/share/multibootusb/data/multibootusb/grub', get_data('data/multibootusb/grub')), + ('/usr/share/multibootusb/data/multibootusb/grub/i386-pc', get_data('data/multibootusb/grub')), + ('/usr/share/multibootusb/data/multibootusb/grub', root_files('data/multibootusb/grub')), ('/usr/share/multibootusb/data/tools/syslinux', get_data('data/tools/syslinux'))] )