mirror of
https://github.com/odriverobotics/ODrive.git
synced 2026-02-06 15:11:52 +08:00
100 lines
3.9 KiB
Python
100 lines
3.9 KiB
Python
|
|
import re
|
|
import subprocess
|
|
import os
|
|
import sys
|
|
import platform
|
|
|
|
def version_str_to_tuple(version_string):
|
|
"""
|
|
Converts a version string to a tuple of the form
|
|
(major, minor, revision, prerelease)
|
|
|
|
Example: "fw-v0.3.6-23" => (0, 3, 6, True)
|
|
|
|
If version_string does not match the pattern above, this function throws an
|
|
Exception.
|
|
"""
|
|
regex=r'.*v([0-9]+)\.([0-9]+)\.([0-9]+)(.*)'
|
|
if not re.match(regex, version_string):
|
|
raise Exception()
|
|
return (int(re.sub(regex, r"\1", version_string)),
|
|
int(re.sub(regex, r"\2", version_string)),
|
|
int(re.sub(regex, r"\3", version_string)),
|
|
(re.sub(regex, r"\4", version_string) != ""))
|
|
|
|
|
|
def get_version_from_git():
|
|
script_dir = os.path.dirname(os.path.realpath(__file__))
|
|
try:
|
|
# Determine the current git commit version
|
|
git_tag = subprocess.check_output(["git", "describe", "--always", "--tags", "--match=*fw*", "--dirty=*"],
|
|
cwd=script_dir)
|
|
git_tag = git_tag.decode(sys.stdout.encoding or 'ascii').rstrip('\n')
|
|
|
|
(major, minor, revision, is_prerelease) = version_str_to_tuple(git_tag)
|
|
|
|
# if is_prerelease:
|
|
# revision += 1
|
|
return git_tag, major, minor, revision, is_prerelease
|
|
|
|
except Exception as ex:
|
|
print(ex)
|
|
return "[unknown version]", 0, 0, 0, 1
|
|
|
|
def get_version_str(git_only=False, is_post_release=False, bump_rev=False, release_override=False):
|
|
"""
|
|
Returns the versions of the tools
|
|
If git_only is true, the version.txt file is ignored even
|
|
if it is present.
|
|
"""
|
|
script_dir = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
# Try to read the version.txt file that is generated during
|
|
# the packaging step
|
|
version_file_path = os.path.join(script_dir, 'version.txt')
|
|
if os.path.exists(version_file_path) and git_only == False:
|
|
with open(version_file_path) as version_file:
|
|
return version_file.readline().rstrip('\n')
|
|
|
|
_, major, minor, revision, unreleased = get_version_from_git()
|
|
if bump_rev:
|
|
revision += 1
|
|
version = '{}.{}.{}'.format(major, minor, revision)
|
|
if is_post_release:
|
|
version += ".post"
|
|
elif not release_override and unreleased:
|
|
version += ".dev"
|
|
return version
|
|
|
|
if __name__ == '__main__':
|
|
import argparse
|
|
parser = argparse.ArgumentParser(description='Version Dump\n')
|
|
parser.add_argument("--output", type=argparse.FileType('w'), default='-',
|
|
help="C header output file")
|
|
|
|
args = parser.parse_args()
|
|
|
|
git_name, major, minor, revision, unreleased = get_version_from_git()
|
|
print('Firmware version {}.{}.{}{} ({})'.format(
|
|
major, minor, revision, '-dev' if unreleased else '',
|
|
git_name))
|
|
#args.output.write('const unsigned char fw_version = "{}"\n'.format(git_name))
|
|
args.output.write('const unsigned char fw_version_major_ = {};\n'.format(major))
|
|
args.output.write('const unsigned char fw_version_minor_ = {};\n'.format(minor))
|
|
args.output.write('const unsigned char fw_version_revision_ = {};\n'.format(revision))
|
|
args.output.write('const unsigned char fw_version_unreleased_ = {};\n'.format(1 if unreleased else 0))
|
|
|
|
def setup_udev_rules(logger):
|
|
if platform.system() != 'Linux':
|
|
if logger: logger.error("This command only makes sense on Linux")
|
|
return
|
|
if os.getuid() != 0:
|
|
if logger: logger.warn("you should run this as root, otherwise it will probably not work")
|
|
with open('/etc/udev/rules.d/91-odrive.rules', 'w') as file:
|
|
file.write('SUBSYSTEM=="usb", ATTR{idVendor}=="1209", ATTR{idProduct}=="0d3[0-9]", MODE="0666", ENV{ID_MM_DEVICE_IGNORE}="1"\n')
|
|
file.write('SUBSYSTEM=="usb", ATTR{idVendor}=="0483", ATTR{idProduct}=="df11", MODE="0666"\n')
|
|
subprocess.check_call(["udevadm", "control", "--reload-rules"])
|
|
subprocess.check_call(["udevadm", "trigger"])
|
|
if logger: logger.info('udev rules configured successfully')
|